10.16T2 平方差

Description

冈伦确定了 个时间节点，每一个节点都存在影响世界线的“因素”。只不过，时间跳跃会对冈伦带来不可逆的影响，他们经过试验，将该影响数字化：
我们假定时间是一条数轴，现在的时间是原点 ，我们把所有的“因素”存在的时间点也用数轴上的数字表示，时间点越是往前，对应的值便越大，数轴上长为 的时间段称作 时间单位。
当冈伦到达“因素”存在的时间点时，可以每次都用一定的代价X将之收集在容器中，而当冈伦回到现在所在的时间点时，也可以用相等的代价X将之放入特殊的分析装置中（每次放入容器的“因素”无论多少都只花费一份代价 X）。
通过进一步的试验，他们发现：与收集“因素”的容器一起进行时间跳跃会给身体带来更大的负担，对于每1时间单位的时间跳跃，冈伦的负担为 
（当 前 容 器 中 因 素 个 数+1）^2 。
现在，你需要求得一种收集方式，使完成收集对冈伦带的影响最小。

Input

第一行两个整数 n,X ,表示“因素”的个数与代价 X。
第二行n个正整数di,表示第i远的“因素”距现在时间点的距离，不会有两个在相同时间点的因素。

Output

输出一行一个整数，表示冈伦会受到的最小影响。

Sample Input

2 50 1 10

Sample Output

205 先以10点代价走到时间点10 ； 以50点代价收集位于该时间点的“因素”； 再以36 点代价走到时间点1 ； 以50点代价收集位于该时间点的“因素”； 最后以9点代价走回现在，并以50 点代价把“因素”放置在分析器中。

Hint

数据范围及约定
40%数据，n<=5;
60%数据，n<=2000;X<=2*10^4;di<=4*10^4;
100%数据，n<=2*10^5;X<=10^9;di<=10^9;

 

 

 

 

code：

#include<iostream>
#include<cstdio>
#define N 200006
using namespace std;
long long a[N];
int main(){
    long long n,X;
    cin>>n>>X;
    for(long long i=1;i<=n;i++){
        cin>>a[i];
        a[i]+=a[i-1];
    }
    long long ans=1e18;
    for(long long k=1;k<=n;k++){
        long long num=3ll,sum=0;
        for(long long i=n;i>=1;i-=k){
            sum+=(a[i]-a[max(0ll,i-k)])*(max(num,5ll));
            num+=2;
            if(sum>=ans)break;
        }
        ans=min(ans,sum+(long long)(k+n)*X);
    }
    cout<<ans;
    return 0;
}

over