注:此题A不掉,但是和网上标程对拍10+分钟没有发现错误。
今天学长讲CDQ分治优化斜率DP,然后我发现昨天学长讲的斜率DP还没学= =
大概就是有一些乘法的DP转移式子,然后假设j2要比j1更优,于是把带j的放到一边,带i的放到一边
发现一个类似斜率的表达式。而且下标是增大的,且f[i]也是增大的
所以斜率必须保证单调不减
单调队列就不用弹出末尾的值了,直接判断斜率是否大于等于上一段的斜率。
然后由于f[i]递增,队首指针一直会向右移动。
然而如果维护的是一个上凸包,且f[i]不一直向右那就有可能要用平衡树了
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#define maxl 50010
#define eps 1e-7
using namespace std;
int n,cnt,pnt;
long long L;
long long a[maxl],sum[maxl],p[maxl];
double k[maxl];
long long g[maxl],F[maxl],f[maxl];
void prework()
{
for(int i=1;i<=n;i++)
scanf("%lld",&a[i]);
for(int i=1;i<=n;i++)
{
sum[i]=sum[i-1]+a[i];
f[i]=sum[i]+i;
}
}
//f[i]=sum[i]+i g[i]=F[i]+(1+L+f[i])^2
void mainwork()
{
int j;
cnt=1;pnt=1;
p[1]=0;k[1]=0;g[0]=F[0]+pow(1+L+f[0],2);
for(int i=1;i<=n;i++)
{
while(pnt<cnt && k[pnt+1]-eps<(double)2*f[i])
pnt++;
j=p[pnt];
F[i]=F[j]+pow(f[i]-f[j]-1-L,2);
g[i]=F[i]+pow(1+L+f[i],2);
j=p[cnt];
if((double)(g[i]-g[j])>k[cnt]*(f[i]-f[j])-eps)
++cnt,p[cnt]=i,k[cnt]=(double)(g[i]-g[j])/((double)f[i]-f[j]);
}
}
void print()
{
printf("%lld\n",F[n]);
}
int main()
{
while(~scanf("%d%lld",&n,&L))
{
prework();
mainwork();
print();
}
return 0;
}