题目链接:https://loj.ac/problem/10177
题目大意:
方法一:
sum[i]表示前缀和,dp[i]表示选第i个数时的最优方案。
dp[i] = max{dp[j]+sum[i]-sum[j+1]} j∈[i-k-1,i-1)//枚举上一次不选的位置
那么思路就很明确了,对于dp[j]-sum[j+1],用一个位置递
增,值递减的单调队列维护。
方法二:dp[i]表示不选第i个数,选出所有数的最大值。那么sum[n]-min(dp[j], n>= j>=n-k);就是答案。
dp[i]=min(dp[j], n>=j>=n-k)+a[i];
//方法一
#include <bits/stdc++.h>
#define LL long long
using namespace std;
int a[2000005];
LL sum[2000005]={0}, f[2000005]={0};
struct node{
LL w, i;
}q[2000005];
int main(){
int n, k;
scanf("%d%d", &n, &k);
for(int i=1; i<=n; i++){
scanf("%d", &a[i]);
sum[i]=sum[i-1]+a[i];
}
int L=0, R=0;
q[0]=node{-a[1], 0};
for(int i=1; i<=k; i++){
f[i]=sum[i];
while(L<=R&&f[i]-sum[i+1]>=q[R].w){
R--;
}
q[++R]=node{f[i]-sum[i+1], i};
}
for(int i=k+1; i<=n; i++){
while(L<=R&&q[L].i<i-k-1){
L++;
}
f[i]=q[L].w+sum[i];
while(f[i]-sum[i+1]>=q[R].w){
R--;
}
q[++R]=node{f[i]-sum[i+1], i};
}
printf("%lld\n", f[n]);
return 0;
}
//把loj#10180 烽火传递改一改就AC了
#include <bits/stdc++.h>
#define LL long long
using namespace std;
LL a[2000005], dp[2000005];
struct node{
LL w, i;
}q[2000005];
int main()
{
LL n, m, sum=0;
memset(dp, 0, sizeof(dp));
scanf("%lld%lld", &n, &m);
m++;
for(LL i=1; i<=n; i++){
scanf("%lld", &a[i]);
sum+=a[i];
}
LL L=1, R=0;
//预处理
for(LL i=1; i<=m; i++){
dp[i]=a[i];
while(L<=R&&q[R].w>=dp[i]){//单调性
R--;
}
q[++R]=node{dp[i], i};
}
//滑动窗口
for(LL i=m+1; i<=n; i++){
while(L<=R&&q[L].i<(i-m)){//范围外
L++;
}
dp[i]=q[L].w+a[i];
while(L<=R&&q[R].w>=dp[i]){//单调性
R--;
}
q[++R]=node{dp[i], i};
}
LL ans=(1ll*1<<60);
for(LL i=n-m+1; i<=n; i++){
ans=min(ans, dp[i]);
}
printf("%lld\n", sum-ans);
return 0;
}