P6568 [NOI Online #3 提高组] 水壶
题目描述
有 n 个容量无穷大的水壶,它们从 1∼n 编号,初始时 i号水壶中装有 Ai单位的水。
你可以进行不超过 k次操作,每次操作需要选择一个满足 1≤x≤n-1的编号 x,然后把
x 号水壶中的水全部倒入 x+1 号水壶中。
最后你可以任意选择恰好一个水壶,并喝掉水壶中所有的水。现在请你求出,你最多能喝到多少单位的水。
输入格式
第一行一个正整数 n,表示水壶的个数。
第二行一个非负整数 k,表示操作次数上限。
第三行 n 个非负整数,相邻两个数用空格隔开,表示水壶的初始装水量 A1, A2…
输出格式
一行,仅一个非负整数,表示答案。
输入输出样例
输入 #1
10
5
890 965 256 419 296 987 45 676 976 742
输出 #1
3813
思路
很显然每次倒水之后,水总是单调递增的,那么肯定要把这个更多的水,倒到下一个水壶里面去。也就是说每次倒水的范围是一个区间。
因此问题就很显然地转化为了求一个长度为 k+1的最大连续子段和,即求区间段的最大前缀和。
题解
#include<bits/stdc++.h>
using namespace std;
#define N 1000005
int n,k,a[N],sum[N]={},m=0;
int main(){
cin >> n >> k;
for(int i=1;i<=n;i++){
cin >> a[i];
sum[i] = sum[i-1] + a[i];
}
for(int i=1;i+k<=n;i++){
m = max(m,sum[i+k]-sum[i-1]);//经过k次操作,其实一共喝掉k+1壶水;即算区间长为k+1的最大和
}
cout << m << endl;
}