HDU3045(Picnic Cows 斜率优化dp入门题)

题目

题意： 有n个奶牛分别有对应的兴趣值，现在对奶牛分组，每组成员不少于t，在每组中所有的成员兴趣值要减少到该组的最小值，问总共最少需要减少的兴趣值是多少。

先对所有成员的兴趣值从小到大排序。记录兴趣的前缀和sum[i]
dp[i]=min(dp[j]+sum[i]-sum[j]-a[j+1]*(i-j)) (j>=0&&j<i-t+1)
dp[j]-sum[j]+ a[j+1] * j = i*a[j+1] +dp[i]-sum[i]
以a[j+1]做为横坐标 dp[j]-sum[j]+ a[j+1] * j 为纵坐标建立好多决策点 j
画斜率为i的平行线交这些点其中一点使得该线的纵截距最小 这个点就是最优决策点j
每次要查询的平行线斜率为 i i单调递增
每个决策点的横坐标 a[j+1] 由于排序是单调递增
纵坐标dp[j]>dp[k] a[j+1] * j-sum[j]>a[k+1] * k-sum[k] 显然这几个式子都成立也递增
注意每个dp[i]更新完毕后放入队列的并不是决策 i 而是决策 i-t+1;因为这样才可以保证更新dp[i+1]时 队列里放的决策都满足题目条件每一组个数>=t的条件

#include<cstdio>
#include<algorithm>
typedef long long ll;
using namespace std;
const int N=4e5+5;
ll a[N],sum[N],dp[N];
int q[N];
inline ll getX(int i,int j){
    return a[j+1]-a[i+1];
}
inline ll getY(int i,int j){
    return (dp[j]-sum[j]+a[j+1]*j)-(dp[i]-sum[i]+a[i+1]*i);
}
inline void getDP(int i,int j){
    dp[i]=dp[j]+sum[i]-sum[j]-a[j+1]*(i-j);
}
int main(){
    int n,t;
    while(~scanf("%d%d",&n,&t)){
        for(int i=1;i<=n;++i) scanf("%lld",&a[i]);
        sort(a+1,a+n+1);
        for(int i=1;i<=n;++i) sum[i]=sum[i-1]+a[i];
        int head=1,tail=1;q[1]=0;
        for(int i=1;i<=n;++i){
            while(head<tail&&getY(q[head],q[head+1])<=i*getX(q[head],q[head+1])) ++head;
            getDP(i,q[head]);
            int j=i-t+1;if(j<t) continue;//放入单调队列的是决策j 保证修改dp[i]时单调队列里的决策<i-t+1;
            while(head<tail&&getY(q[tail-1],q[tail])*getX(q[tail],j)>=getY(q[tail],j)*getX(q[tail-1],q[tail])) --tail;
            q[++tail]=j;
        }
        printf("%lld\n",dp[n]);
    }
}