题目描述:
把长度为n的序列{ci}划分成若干个区间,每个区间的贡献为ci的和的平方加上M,求最小贡献
n<=500000
题目分析:
斜率优化模板题,题解网上很多。。这里主要说一下细节
while(head+1<tail&&gety(q[tail-1],q[tail-2])*getx(i,q[tail-1])>gety(i,q[tail-1])*getx(q[tail-1],q[tail-2])) tail--;
这个ci是可能为0的,也就是说转移的时候可能会有两个点重合(getx=gety=0),这时候的斜率判断就必须是>=(或<=) ,而不能是>(或<),必须把重合的点pop掉,才能保证下凸包,所以上面这种写法会WA
AC code:
#include<cstdio>
#define maxn 50015
#define LL long long
int n,m,q[maxn],head,tail;
LL s[maxn],f[maxn];
inline LL gety(int i,int j){return f[i]+s[i]*s[i]-f[j]-s[j]*s[j];}
inline LL getx(int i,int j){return 2*(s[i]-s[j]);}
int main()
{
while(~scanf("%d%d",&n,&m))
{
q[head=0]=0,tail=1;
for(int i=1;i<=n;i++) scanf("%lld",&s[i]),s[i]+=s[i-1];
for(int i=1;i<=n;i++)
{
while(head+1<tail&&gety(q[head+1],q[head])<=getx(q[head+1],q[head])*s[i]) head++;
f[i]=f[q[head]]+(s[i]-s[q[head]])*(s[i]-s[q[head]])+m;
while(head+1<tail&&gety(q[tail-1],q[tail-2])*getx(i,q[tail-1])>=gety(i,q[tail-1])*getx(q[tail-1],q[tail-2])) tail--;
q[tail++]=i;
}
printf("%lld\n",f[n]);
}
}