斜率优化DP

Division HDU - 3480 （斜率优化 || 四边形优化）

链接：http://acm.hdu.edu.cn/showproblem.php?pid=3480

题意：给定 n 个数，计算将它们分为 k 个子集的最小花费。花费为：每个子集最大值和最小值之差的平方和

思路：设dp[i][j]表示前 i 个数，分成 j 个子集的最小花费

• 可以得到转移方程为：$dp[i][j]={\sum }_{k=1}^{j-1}dp[k][j-1]+(a[i]-a[k+1])\times (a[i]-a[k+1])$
• 这是一个 $O(n^3)$ 的复杂度。因此进行斜率优化
• $dp[i][j]+2a[i]\times a[k+1]=dp[k][j-1]+a[i]\times a[i]+a[k+1]\times a[k+1]$
• 设 $k=2a[i]，x=a[k+1],y=dp[k][j-1]+a[k+1]\times a[k+1],b=dp[i][j]-a[i]\times a[i]$
• 然后就维护一个下凸包，$slope(q[h],q[h+1])<=k$ 时，这个 $q[h]$ 不是最优的， h++。当 $slope(q[t-1],q[t])>=slope(q[t-1],i)$ 时，此时的 $q[t]$ 也不是最优的，t–。
• 坑点：直接使用slope计算斜率会超时，x 和 y 出现负数时也容易出错，t从1开始表示第一个决策点为 $P_0$

#include <cstdio>
#include <algorithm>
#include <cstring>
#define ll long long
using namespace std;
const int maxn=10010,maxm=5010,inf=0x3f3f3f3f;

int tt,n,m,a[maxn];
int q[maxn],h,t;
int dp[maxn][maxm];

inline int Y(int k,int j){return dp[k][j-1]+a[k+1]*a[k+1];}
inline int X(int k){return a[k+1];}
inline int getX(int k1,int k2){return X(k2)-X(k1);}
inline int getY(int k1,int k2,int j){return Y(k2,j)-Y(k1,j);}
inline long double slope(int k1,int k2,int j)
{
    int y=Y(k2,j)-Y(k1,j);
    int x=X(k2)-X(k1);
    return (long double)y/x;
}

int main()
{
    scanf("%d",&tt);
    int Case=0;
    while(tt--)
    {
        scanf("%d%d",&n,&m);
        for(int i=0;i<=n;++i)
            for(int j=0;j<=m;++j)
                dp[i][j]=inf;
        for(int i=1;i<=n;++i) scanf("%d",&a[i]);
        sort(a+1,a+1+n);
        for(int i=1;i<=n;++i) dp[i][1]=(a[i]-a[1])*(a[i]-a[1]);
        for(int j=2;j<=m;++j)
        {
            h=1,t=1;
            for(int i=1;i<=n;++i)
            {
                while(h<t)
                {
                    int k1=q[h],k2=q[h+1];
                    int x=getX(k1,k2);
                    int y=getY(k1,k2,j);
                    if(y<=2*x*a[i]) ++h;
                    else break;
                }
                int k=q[h];
                dp[i][j]=dp[k][j-1]+(a[i]-a[k+1])*(a[i]-a[k+1]);
                while(h<t)
                {
                    int k1=q[t-1],k2=q[t];
                    int x1=getX(k1,k2);
                    int y1=getY(k1,k2,j);
                    int k3=q[t-1],k4=i;
                    int x2=getX(k3,k4);
                    int y2=getY(k3,k4,j);
                    if(y1*x2>=y2*x1) --t;
                    else break;
                }
                q[++t]=i;
            }
        }
        printf("Case %d: %d\n",++Case,dp[n][m]);
    }
    return 0;
}

超时代码

#include <cstdio>
#include <algorithm>
#include <cstring>
#define ll long long
using namespace std;
const int maxn=10010,maxm=5010,inf=0x3f3f3f3f;

int tt,n,m,a[maxn];
int q[maxn],h,t;
int dp[maxn][maxm];

inline int Y(int k,int j){return dp[k][j-1]+a[k+1]*a[k+1];}
inline int X(int k){return a[k+1];}
inline long double slope(int k1,int k2,int j)
{
    int y=Y(k2,j)-Y(k1,j);
    int x=X(k2)-X(k1);
    return (long double)y/x;
}

int main()
{
    scanf("%d",&tt);
    int Case=0;
    while(tt--)
    {
        scanf("%d%d",&n,&m);
        for(int i=0;i<=n;++i)
            for(int j=0;j<=m;++j)
                dp[i][j]=inf;
        for(int i=1;i<=n;++i) scanf("%d",&a[i]);
        sort(a+1,a+1+n);
        for(int i=1;i<=n;++i) dp[i][1]=(a[i]-a[1])*(a[i]-a[1]);
        for(int j=2;j<=m;++j)
        {
            h=1,t=1;
            for(int i=1;i<=n;++i)
            {
                while(h<t&&slope(q[h],q[h+1],j)<=2*a[i]) h++;
                int k=q[h];
                dp[i][j]=dp[k][j-1]+(a[i]-a[k+1])*(a[i]-a[k+1]);
                while(h<t&&slope(q[t-1],q[t],j)>=slope(q[t-1],i,j)) t--;
                q[++t]=i;
            }
        }
        printf("Case %d: %d\n",++Case,dp[n][m]);
    }
    return 0;
}

P3195 [HNOI2008]玩具装箱 （斜率优化）

链接：https://www.luogu.com.cn/problem/P3195

#include <cstdio>
#include <algorithm>
#include <cstring>
#define ll long long
using namespace std;
const int maxn=5e4+10,inf=1e9+7;

ll n,l,dp[maxn],s[maxn];
int q[maxn],h=1,t=0;
ll Y(ll j){return dp[j]+(s[j]+l)*(s[j]+l);}
ll X(ll j){return s[j];}
long double slope(ll i,ll j)
{
    ll y=Y(i)-Y(j),x=X(i)-X(j);
    return (long double)y/x;
}
int main()
{
    scanf("%lld%lld",&n,&l);
    ++l;
    for(int i=1;i<=n;++i)
        scanf("%lld",&s[i]),s[i]+=s[i-1]+1;
    q[++t]=0;
    for(int i=1;i<=n;++i)
    {
        while(h<t&&slope(q[h],q[h+1])<=2*s[i]) h++;
        int j=q[h];
        dp[i]=dp[j]+(s[i]-s[j]-l)*(s[i]-s[j]-l);
        while(h<t&&slope(q[t-1],q[t])>=slope(q[t-1],i)) --t;
        q[++t]=i;
    }
    printf("%lld\n",dp[n]);
    return 0;
}