解题思路:第一次做斜率优化dp,参考了博客http://www.cnblogs.com/kuangbin/archive/2012/08/26/2657650.html。使原本O(n^2)的复杂度变成O(n)。看完博客之后,看了一下代码发现代码中(yj-yk)/(xj-xk) <= sum[i],只要碰到一个不符合这个条件就用j来算dp[i]了。其实是这样的,因为符合上面那个不等式,代表j在以后都会比k要优,所以可以把k去掉,但是一旦在j的时候不满足这个等式,那么维护时斜率本来就是递增的,这意味着后面的点j'和k的斜率只会越来越大,更加满足不了要求,还是k是最优的。这意味着k比它前面的要优,比它后面的点也要优。顾就是答案。
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define N 500009
using namespace std;
int sum[N],stk[N],dp[N];
int head,tail,m;
int getdp(int i,int j)
{
return dp[j]+(sum[i]-sum[j])*(sum[i]-sum[j])+m;
}
int getup(int j,int i)
{
return dp[j]+sum[j]*sum[j]-dp[i]-sum[i]*sum[i];
}
int getdown(int j,int i)
{
return 2*(sum[j]-sum[i]);
}
int main()
{
//freopen("t.txt","r",stdin);
int n;
while(scanf("%d%d",&n,&m)!=EOF)
{
sum[0]=0;
for(int i=1;i<=n;i++) scanf("%d",&sum[i]);
for(int i=1;i<=n;i++)
{
sum[i]+=sum[i-1];
}
head=tail=0;
stk[tail++]=0;
for(int i=1;i<=n;i++)
{
while(head+1<tail&&getup(stk[head+1],stk[head])<=sum[i]*getdown(stk[head+1],stk[head]))//除法会涉及浮点运算,拖慢速度,所以用乘法
head++;
dp[i]=getdp(i,stk[head]);
while(head+1<tail&&getup(i,stk[tail-1])*getdown(stk[tail-1],stk[tail-2])<=getup(stk[tail-1],stk[tail-2])*getdown(i,stk[tail-1]))
tail--;
stk[tail++]=i;
}
printf("%d\n",dp[n]);
}
return 0;
}