codevs 3327 选择数字 单调队列优化DP

题目描述 Description

给定一行n个非负整数a[1]..a[n]。现在你可以选择其中若干个数，但不能有超过k个连续的数字被选择。你的任务是使得选出的数字的和最大。

输入描述 Input Description

第一行两个整数n，k

以下n行，每行一个整数表示a[i]。

输出描述 Output Description

输出一个值表示答案。

样例输入 Sample Input

5 2

1

2

3

4

5

样例输出 Sample Output

12

数据范围及提示 Data Size & Hint

对于20%的数据，n <= 10

对于另外20%的数据， k = 1

对于60%的数据，n <= 1000

对于100%的数据，1 <= n <= 100000，1 <= k <= n，

0 <= 数字大小 <= 1,000,000,000

思路

典型的Dp，单调队列优化一下，注意记录前缀和。
这里要提醒大家了，本题要用到高精度，但鉴于NOIP不强调高精度，我没有仔细的研究过这一专题，这也导致了这道题没法AC，不过读者们放心，正确性一定是有的。

代码

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
using namespace std;
const int INF=0x7fffffff;
const int N=100000+5;
int n,k,head,tail=1;
long long dp[N],q[N],a[N],pre[N],d[N];
int init(int x)
{
    d[x]=dp[x-1]-pre[x];
    if (x>=k+1&&d[x-k-1]==q[head]) head++;
    while(tail>head&&d[x]&&d[x]>q[tail-1]) tail--;
    q[tail++]=d[x];
}
int main() 
{
    scanf("%d%d",&n,&k);
    for (int i=1;i<=n;i++)  
    {
        scanf("%d",&a[i]);
        pre[i]=pre[i-1]+a[i];
    }
    d[0]=0;
    for (int i=1;i<=n;i++) 
    {
        init(i);
        dp[i]=q[head]+pre[i]; 
    }
    printf("%d\n",dp[n]);
    return 0;
}