hdu 3507 Print Article(斜率优化dp)

http://acm.hdu.edu.cn/showproblem.php?pid=3507

思路:这位博主关于这道题的文章解释非常清楚

代码;

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int maxn=5e5+10;
int dp[maxn],sum[maxn];
int n,m;
int q[maxn];
int getf(int j,int k)
{
    return dp[j]+sum[j]*sum[j]-(dp[k]+sum[k]*sum[k]);
}
int getS(int j,int k)
{
    return sum[j]-sum[k];
}
int getDp(int i,int j)
{
    return dp[j]+m+(sum[i]-sum[j])*(sum[i]-sum[j]);
}
int main()
{

    while(~scanf("%d %d",&n,&m))
    {
        for(int i=0;i<=n;i++)
            dp[i]=0;
        for(int i=1;i<=n;i++){
            scanf("%d",&sum[i]);
            sum[i]+=sum[i-1];
        }
        int head=0,tail=0;
        q[tail++]=0;
        for(int i=1;i<=n;i++)
        {
            while(head+1<tail&&getf(q[head+1],q[head])<=2*sum[i]*getS(q[head+1],q[head]))
                head++;
            dp[i]=getDp(i,q[head]);
            while(head+1<tail&&getf(i,q[tail-1])*getS(q[tail-1],q[tail-2])<=getf(q[tail-1],q[tail-2])*getS(i,q[tail-1]))
                tail--;
            q[tail++]=i;
        }
        printf("%d\n",dp[n]);
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/imzxww/article/details/81461567