tjut 3045

最新推荐文章于 2016-11-04 22:02:44 发布

原创最新推荐文章于 2016-11-04 22:02:44 发布 · 152 阅读

0 ·

CC 4.0 BY-SA版权

斜率优化dp 专栏收录该内容

5 篇文章

订阅专栏

本文介绍了一个使用 C++ 实现的动态规划算法案例，该算法通过预处理数组求解特定问题。代码中运用了排序、动态规划等技术，并实现了自定义的数据结构来优化性能。

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

int q[MAXN];
long long  sum[MAXN];
long long a[MAXN];
long long dp[MAXN];

long long getDP(int i,int j)
{
    return dp[j]+(sum[i]-sum[j])-a[j+1]*(i-j);
}
long long getUP(int j,int k)//得到 yj-yk  k<j
{
    return dp[j]-sum[j]+j*a[j+1]-(dp[k]-sum[k]+k*a[k+1]);
}
long long getDOWN(int j,int k)//得到 xj-xk  k<j
{
    return a[j+1]-a[k+1];
}

int main()
{
   // freopen("in.txt","r",stdin);
  //  freopen("out.txt","w",stdout);
    int n,t;
    int head,tail;
    while(scanf("%d%d",&n,&t)==2)
    {
        dp[0]=sum[0]=0;
        a[0]=0;
        for(int i=1;i<=n;i++)
        {
            scanf("%I64d",&a[i]);
        }
        sort(a+1,a+n+1);
        for(int i=1;i<=n;i++)
          sum[i]=sum[i-1]+a[i];
        head=tail=0;
        q[tail++]=0;
        for(int 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]);
            int 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("%I64d\n",dp[n]);
    }
    return 0;
}