P2034 选择数字

P2034 选择数字

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

---

错误日志： longlong 的 \(inf\) 没有设为 0xfffffffffffffff

---

Solution

正难则反
正难则反
复习看到了双倍经验就回来看看
求最大值有点难， 转化为求最小去除值
设 \(dp[i]\) 为 \(i\) 为断点， 考虑到 \(i\) 处的最小去除值
发现每隔 \(K + 1\) 必有一个断点
所以 \(dp[i] = \min_{j = i - k - 1}^{i - 1}dp[j] + a[i]\)
然后单调队列优化， 答案在区间 \([n - K, n]\) 内
总值减一下即可

Code

#include<iostream>
#include<cstdio>
#include<queue>
#include<cstring>
#include<algorithm>
#include<climits>
#define LL long long
#define REP(i, x, y) for(LL i = (x);i <= (y);i++)
using namespace std;
LL RD(){
    LL out = 0,flag = 1;char c = getchar();
    while(c < '0' || c >'9'){if(c == '-')flag = -1;c = getchar();}
    while(c >= '0' && c <= '9'){out = out * 10 + c - '0';c = getchar();}
    return flag * out;
    }
const LL maxn = 200019, inf = 0xfffffffffffffff;
LL num, K;
LL a[maxn], sum;
LL dp[maxn];
struct Que{
	LL Index, val;
	}Q[maxn];
LL head = 1, tail;
void push_back(LL Index, LL val){
	while(head <= tail && Q[tail].val >= val)tail--;
	Q[++tail] = (Que){Index, val};
	}
LL get_min(LL p){
	while(head <= tail && p - Q[head].Index > K + 1)head++;
	return Q[head].val;
	}
void init(){
	num = RD(), K = RD();
	REP(i, 1, num)a[i] = RD(), sum += a[i];
	memset(dp, 127, sizeof(dp));
	push_back(0, 0);
	}
void solve(){
	REP(i, 1, num){
		dp[i] = get_min(i) + a[i];
		push_back(i, dp[i]);
		}
	LL ans = inf;
	REP(i, num - K, num)ans = min(ans, dp[i]);
	printf("%lld\n", sum - ans);
	return ;
	REP(i, 1, num)printf("%lld ", dp[i]);
	}
int main(){
	init();
	solve();
	return 0;
	}