HDU - 3507 Print Article

题目：要输出序列a[n]，每连续输出一段连续子序列的费用是连续输出的数字和的平方加上常数M，让我们求总 费用的最小值。

思路：

参考：http://www.cnblogs.com/kuangbin/archive/2012/08/26/2657650.html

设dp[i]表示输出前i个的最小费用，那么有如下的DP方程：
dp[i]= min{ dp[j]+(sum[i]-sum[j])^2 +M } 0<j<i
其中sum[i]表示数字的前i项和。

计算dp[i]时，k<j ,j点比k点优。
也就是dp[j]+(sum[i]-sum[j])^2+M <= dp[k]+(sum[i]-sum[k])^2+M;
整理得：
[(dp[j]+sum[j]*sum[j])-(dp[k]+sum[k]*sum[k])] / 2(sum[j]-sum[k]) <=sum[i].
如果令：yj=dp[j]+sum[j]*sum[j]   xj=2*sum[j]
那么就变成了斜率表达式：（yj-yk）/(xj-xk) <= sum[i];
而且不等式右边是递增的。
我们令g[k,j]=(yj-yk)/(xj-xk)
第一：如果上面的不等式成立，那就说j比k优，而且随着i的增大上述不等式一定是成立的，也就是对i以后算DP值时，j都比k优。那么k就是可以淘汰的。
第二：如果 k<j<i   而且 g[k,j]>g[j,i] 那么 j 是可以淘汰的。

所以这样相当于在维护一个下凸的图形，斜率在逐渐增大。

代码：

#pragma comment(linker, "/STACK:1024000000,1024000000")
#include<iostream>
#include<algorithm>
#include<ctime>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<string>
#include<vector>
#include<map>
#include<set>
#include<queue>
#include<stack>
#include<list>
#include<numeric>
using namespace std;
#define LL long long
#define ULL unsigned long long
#define INF 0x3f3f3f3f
#define mm(a,b) memset(a,b,sizeof(a))
#define PP puts("*********************");
template<class T> T f_abs(T a){ return a > 0 ? a : -a; }
template<class T> T gcd(T a, T b){ return b ? gcd(b, a%b) : a; }
template<class T> T lcm(T a,T b){return a/gcd(a,b)*b;}
// 0x3f3f3f3f3f3f3f3f
//0x3f3f3f3f

const int maxn=5e5+50;
int dp[maxn];
int q[maxn];
int sum[maxn];
int head,tail,n,M;
int getdp(int i,int j){
    return dp[j]+M+(sum[i]-sum[j])*(sum[i]-sum[j]);
}
int gety(int k,int j){//yj-yk
    return dp[j]+sum[j]*sum[j]-(dp[k]+sum[k]*sum[k]);
}
int getx(int k,int j){//xj-xk
    return 2*(sum[j]-sum[k]);
}
int main(){

//    freopen("D:\\input.txt","r",stdin);
//    freopen("D:\\output.txt","w",stdout);
    while(~scanf("%d%d",&n,&M)){
        for(int i=1;i<=n;i++)
            scanf("%d",&sum[i]);
        sum[0]=dp[0]=0;
        for(int i=1;i<=n;i++)
            sum[i]+=sum[i-1];
        head=0;
        tail=-1;
        q[++tail]=0;
        for(int i=1;i<=n;i++){
            while(head<tail&&gety(q[head],q[head+1])<=sum[i]*getx(q[head],q[head+1]))
                head++;
            dp[i]=getdp(i,q[head]);
            while(head<tail&&gety(q[tail-1],q[tail])*getx(q[tail],i)>=gety(q[tail],i)*getx(q[tail-1],q[tail]))
                tail--;
            q[++tail]=i;
        }
        printf("%d\n",dp[n]);
    }
    return 0;
}