bzoj3675 [ APIO2014 ] -- 斜率优化DP

令$f_{i,j}$表示前 $j$ 个数分割 $i$ 次的答案，$s_i$表示前缀和。可以得到$DP$方程：

$$f_{i,j}=max(f_{i-1,k}+s_k\times (s_i-s_k)),k<j$$
然后考虑斜率优化。
$j<k$且 $j$ 比 $k$ 优的条件是：
$$f_{i-1,j}+s_j\times(s_i-s_j)>f_{i-1,k}+s_k\times(s_i-s_k)$$
$$s_i\times(s_j-s_k)>s_j^2-f_{i-1,j}-s_k^2+f_{i-1,k}$$
$$s_i<{s_j^2-f_{i-1,j}-(s_k^2-f_{i-1,k})\over {s_j-s_k}}$$
将 $s_j^2-f_{i-1,j}$作为点 $j$ 的 $y$ 坐标， $s_j$ 作为 $x$ 坐标。就相当于 $j$ 到 $k$ 的斜率大于 $s_i$ ，用凸包维护即可。由于空间不够，还要用滚动数组优化。

代码

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
#define N 100010
#define ll long long
inline char nc(){
    static char buf[100000],*p1=buf,*p2=buf;
    return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;
}
inline void Read(ll& x){
    char c=nc();
    for(;c<'0'||c>'9';c=nc());
    for(x=0;c>='0'&&c<='9';x=(x<<3)+(x<<1)+c-48,c=nc());
}
struct Node{
    ll x,y;
    Node(ll x=0,ll y=0):x(x),y(y){}
    Node operator - (Node a){
        return Node(x-a.x,y-a.y);
    }
}ch[N],T;
ll f[2][N],s[N];
int i,j,k,n,m,p,t,w[N];
bool b;
inline bool Check(Node a,Node b){
    return (double)a.x*b.y>=(double)a.y*b.x;
}
int main(){
    scanf("%d%d",&n,&m);
    for(i=1;i<=n;i++)Read(s[i]),s[i]+=s[i-1];
    for(i=1;i<=m;i++,b^=1){
        k=t=0;
        for(j=1;j<=n;j++){
            while(k>t)k--;
            while(k<t&&ch[k+1].y-ch[k].y<s[j]*(ch[k+1].x-ch[k].x))k++;
            f[b][j]=f[b^1][w[k]]+s[w[k]]*(s[j]-s[w[k]]);
            T.x=s[j];T.y=s[j]*s[j]-f[b^1][j];
            while(t&&Check(T-ch[t-1],ch[t]-ch[t-1]))t--;
            ch[++t]=T;w[t]=j;
        }
    }
    printf("%lld\n",f[b^1][n]);
    return 0;
}