hdu 3507 Print Article 斜率优化DP

最新推荐文章于 2020-05-21 21:25:44 发布

TommyTT

最新推荐文章于 2020-05-21 21:25:44 发布

阅读量1.4k

点赞数 3

CC 4.0 BY-SA版权

分类专栏： dp

本文链接：https://blog.youkuaiyun.com/t1019256391/article/details/9337761

dp 专栏收录该内容

46 篇文章

订阅专栏

本文介绍了一种使用斜率优化技巧解决动态规划问题的方法。通过维护一个双端队列来保持状态转移方程中斜率的单调性，从而达到优化的目的。文章提供了完整的C++实现代码，展示了如何有效地更新DP表并求解最小值。

初学斜率优化……

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define maxn 500005
int sum[maxn];
int dp[maxn];
int a[maxn];
int q[maxn];
int y(int i)
{
    return dp[i]+sum[i]*sum[i];
}
int Gup(int j,int k)
{
    return y(j)-y(k);
}
int Gdown(int j,int k)
{
    return sum[j]-sum[k];
}
inline int ReadInt()        //开挂，刷一下排行
{
    char ch = getchar();
    int data = 0;
    while (ch < '0' || ch > '9')
    {
        ch = getchar();
    }
    do
    {
        data = data*10 + ch-'0';
        ch = getchar();
    }while (ch >= '0' && ch <= '9');
        return data;
}
int main()
{
    int n,m;
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        sum[0]=0;
        for(int i=1;i<=n;i++){
            a[i]=ReadInt();
            sum[i]=sum[i-1]+a[i];
        }
        int head=0,tail=-1;
        q[++tail]=0;
        for(int i=1;i<=n;i++)
        {
            while(head<tail&&Gup(q[head+1],q[head])<=2*sum[i]*Gdown(q[head+1],q[head])) head++;
            dp[i]=dp[q[head]]+(sum[i]-sum[q[head]])*(sum[i]-sum[q[head]])+m;
            while(head<tail&&Gup(i,q[tail])*Gdown(q[tail],q[tail-1])<=Gup(q[tail],q[tail-1])*Gdown(i,q[tail]))
            tail--;
            q[++tail]=i;
        }
        printf("%d\n",dp[n]);
    }
    return 0;
}