bzoj3675 [ APIO2014 ] -- 斜率优化DP

最新推荐文章于 2019-05-18 16:01:00 发布

原创最新推荐文章于 2019-05-18 16:01:00 发布 · 364 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#优化 #dp

DP 专栏收录该内容

47 篇文章

订阅专栏

本文介绍了一种使用斜率优化的动态规划算法解决特定分割问题的方法。通过定义状态转移方程，利用凸包来维护最优解，并采用滚动数组优化空间复杂度。此算法适用于求解前缀和类型的最优化问题。

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

f i, j = m a x (f i - 1, k + s k \times (s i - s k)), k < j

$f_{i,j}=max(f_{i-1,k}+s_k\times (s_i-s_k)),k<j$
然后考虑斜率优化。

j<k $j<k$ 且

j $j$ 比

k $k$ 优的条件是：

f i - 1, j + s j \times (s i - s j) > f i - 1, k + s k \times (s i - s 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 2 j - f i - 1, j - s 2 k + f i - 1, 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 2 j - f i - 1 , j - ( s 2 k - f i - 1 , k ) s j - s k

$s_i<{s_j^2-f_{i-1,j}-(s_k^2-f_{i-1,k})\over {s_j-s_k}}$
将

s2j−fi−1,j $s_j^2-f_{i-1,j}$ 作为点

j $j$ 的

y $y$ 坐标，

sj $s_j$ 作为

x $x$ 坐标。就相当于

j $j$ 到

k $k$ 的斜率大于

si $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;
}