看到本题以后,分析了一下,发现是DP,然后看了下数据规模会超时。后来看了hzw的题解才发现是需要斜率优化的,方程是看懂了但是证明还是有点迷迷糊糊。最后,判断函数是对着hzw的标程打的,剩下自己手撸掉了,程序并不长,但思想很秒。
#include <cstdio>
#include <iostream>
#include <cstdlib>
#include <algorithm>
using namespace std ;
int n,l,C;
int c[50010],q[50010];
long long sum[50010],f[50010];
void init (){
cin >>n >>l;
int i;
for (i=1;i<=n;i++){
scanf ("%d",&c[i]);
sum[i]=sum[i-1]+c[i]+1;
}
C=l+1;
}
double pd(int j,int k){
return (f[k]-f[j]+(sum[k]+C)*(sum[k]+C)-(sum[j]+C)*(sum[j]+C))/(2.0*(sum[k]-sum[j]));
}
int main (){
init ();
int i,l,r,t;
l=r=1;
q[r]=0;
for (i=1;i<=n;i++){
while (l<r&&pd (q[l],q[l+1])<=sum[i]) l++;
t=q[l];
f[i]=f[t]+(sum[i]-sum[t]-C)*(sum[i]-sum[t]-C);
while(l<r&&pd (q[r],i)<pd (q[r-1],q[r])) r--;
q[++r]=i;
}
printf ("%lld\n",f[n]);
return 0;
}