https://www.lydsy.com/JudgeOnline/problem.php?id=1010
时隔2年我终于把这题过了。。。。当时写的代码还是naive。。。
https://blog.csdn.net/liufengwei1/article/details/79232793
安利一下捧杯爷的讲课https://www.bilibili.com/video/av62726813
我按照这上面讲的写的,设s[i]=sum[i]+i,c=L+1
f[i]=f[j]+(s[i]-s[j]-c)^2
化简得f[j]+s[j]^2=2(s[i]-c)*s[j]+f[i]-(s[i]-c)^2.
y=f[j]+s[j]^2,x=s[j],k=2(s[i]-c),目的是f[i]最小,那么就是要截距最小,维护下凸包
但是有一些小细节,n=5e4,ci=1e7,这样的话类似s[i]*s[i]是会爆long long的,而bzoj不支持__int128,所以应该要写高精度。。。。
但是由于数据过水,我看网上代码好像都没写高精度,于是我交了一发就过了。。。
这题还可以体现一个决策单调性,因为s[i]的,是不断增加的,f[i]是不断增加的,斜率2*(s[i]-c)也是不断增加的,那么对于i>j,由f[j]转移到f[i]得到最优值的话,有i2>=i,j2>=j,f[j2]转移到f[i2]最优,所以只要加入新点弹出队尾,每次开始匹配弹出队首就行了。
如果没有决策单调性可能还要在下凸包上三分。。。
#include<bits/stdc++.h>
using namespace std;
const int maxl=5e4+10;
int n,L,c;
long long sum[maxl],s[maxl],f[maxl];
long long qx[maxl],qy[maxl];
inline void prework()
{
scanf("%d%d",&n,&L);
for(int i=1;i<=n;i++)
{
scanf("%lld",&sum[i]),sum[i]+=sum[i-1];
s[i]=sum[i]+i;
}
c=L+1;
}
inline bool cmp(int a,int b,long long x,long long y)
{
return (double)(y-qy[a])*(qx[a]-qx[b])<=(double)(x-qx[a])*(qy[a]-qy[b]);
}
inline long long calc(int id,long long k)
{
return qy[id]-qx[id]*k;
}
inline void mainwork()
{
int head=1,tail=0;
++tail;qx[tail]=0;qy[tail]=0;
long long x,y;
for(int i=1;i<=n;i++)
{
while(head<tail && calc(head+1,2*(s[i]-c))<=calc(head,2*(s[i]-c)))
head++;
f[i]=calc(head,2*(s[i]-c))+(s[i]-c)*(s[i]-c);
x=s[i];y=f[i]+s[i]*s[i];
while(head<tail && cmp(tail,tail-1,x,y))
tail--;
++tail;qx[tail]=x;qy[tail]=y;
}
}
inline void print()
{
printf("%lld\n",f[n]);
}
int main()
{
prework();
mainwork();
print();
return 0;
}