题目描述
给定一行n个非负整数a[1]…a[n]。现在你可以选择其中若干个数,但不能有超过k个连续的数字被选择。你的任务是使得选出的数字的和最大。
输入格式
第一行两个整数n,k
以下n行,每行一个整数表示a[i]。
输出格式
输出一个值表示答案。
一个小dp
我们设f[i]为不选i时候的最优值
然后我写了个60分暴力
结果吸了一口氧过了
(溜)
#include <cstdio>
#include <iostream>
#include <algorithm>
#define int long long
using namespace std;
int n,k;
int a[100005];
int sum[100005];
int f[100005];
inline int read(){
int x=0,f=1;
char ch=getchar();
while(ch<'0'||ch>'9'){
if(ch=='-')f=-1;ch=getchar();}
while(ch>='0'&&ch<='9'){
x=x*10+ch-'0';ch=getchar();}
return x*f;
}
main(){
n=read();k=read();
for(int i=1;i<=n;i++) a[i]=read(),sum[i]=sum[i-1]+a[i];
int q=0;
for(int i=1;i<=n+1;i++)
for(int j=max(q,i-k-1);j<i;j++)
f[i]=max(f[i],f[j]+sum[i-1]-sum[j]);
printf("%lld",f[n+1]);
return 0;
}