HDU 3045 Picnic Cows [DP斜率优化]

最新推荐文章于 2019-08-29 19:29:52 发布

ACTerminate

最新推荐文章于 2019-08-29 19:29:52 发布

阅读量264

点赞数

分类专栏：斜率优化 DP 文章标签：斜率优化

本文链接：https://blog.csdn.net/acterminate/article/details/79354279

版权

DP 同时被 2 个专栏收录

25 篇文章 0 订阅

订阅专栏

斜率优化

2 篇文章 0 订阅

订阅专栏

题意：给你长度为n的数列，将其中的数划分为几个部分，每个部分数的个数要大于等于T，每个部分的val是每个数减去最小值的总和，求val和的最小值。

题解：首先肯定是将数列排序，定义dp[i]表示，计算到i时答案的总和最小值，这样我们的转移是O(n^2)的。

根据题意可以列出，更优dp位置：（p>j）

dp[p]+sum[i]-sum[p]-a[p+1]*(i-p)<=dp[j]+sum[i]-sum[j]-a[j+1]*(i-j)

化简可得：

((dp[p]-sum[p]+p*a[p+1])-(dp[j]-sum[j]+j*a[j+1))/(a[p+1]-a[j+1])<=i

单调队列维护凸包即可。

AC代码：

#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
#include<queue>
using namespace std;
typedef long long ll;
const ll MAXN=400010;

ll q[MAXN];
ll sum[MAXN];
ll a[MAXN];
ll dp[MAXN];

ll getDP(ll i,ll j)
{
    return dp[j]+(sum[i]-sum[j])-a[j+1]*(i-j);
}
ll getUP(ll j,ll k)
{
    return dp[j]-sum[j]+j*a[j+1]-(dp[k]-sum[k]+k*a[k+1]);
}
ll getDOWN(ll j,ll k)
{
    return a[j+1]-a[k+1];
}

int main()
{
    ll n,t;
    ll head,tail;
    while(~scanf("%d%d",&n,&t))
    {
        dp[0]=sum[0]=0;
        a[0]=0;
        for(ll i=1;i<=n;i++)
            scanf("%lld",&a[i]);
        sort(a+1,a+n+1);
        for(ll i=1;i<=n;i++)
        	sum[i]=sum[i-1]+a[i];
        head=tail=0;
        q[tail++]=0;
        for(ll i=1;i<=n;i++)
        {
            while(head+1<tail&&getUP(q[head+1],q[head])<=i*getDOWN(q[head+1],q[head]))
            	head++;
            dp[i]=getDP(i,q[head]);
            ll j=i-t+1;
            if(j<t)continue;
            while(head+1<tail&&getUP(j,q[tail-1])*getDOWN(q[tail-1],q[tail-2])<=getUP(q[tail-1],q[tail-2])*getDOWN(j,q[tail-1]))
                   tail--;
            q[tail++]=j;
        }
        printf("%lld\n",dp[n]);
    }
    return 0;
}