P6568 [NOI Online #3 提高组] 水壶

题目描述

有 n个容量无穷大的水壶，它们从 1∼n1∼n 编号，初始时 ii号水壶中装有 Ai单位的水。

你可以进行不超过 kk 次操作，每次操作需要选择一个满足 1≤x≤n-1 的编号 x，然后把 xx 号水壶中的水全部倒入 x+1 号水壶中。

最后你可以任意选择恰好一个水壶，并喝掉水壶中所有的水。现在请你求出，你最多能喝到多少单位的水。

输入格式

第一行一个正整数 n，表示水壶的个数。

第二行一个非负整数 k，表示操作次数上限。

第三行 n个非负整数，相邻两个数用空格隔开，表示水壶的初始装水量 A1​, A2, ⋯⋯, An。

输出格式

一行，仅一个非负整数，表示答案。

输入输出样例

输入 #1

10
5
890 965 256 419 296 987 45 676 976 742

输出 #1

3813

说明/提示

数据规模与约定

• 对于 10%的数据，保证 n≤10。
• 对于 30%的数据，保证 n≤100。
• 对于 50% 的数据，保证 n≤103。
• 对于 70% 的数据，保证 n≤105。
• 对于 100%的数据，保证 1≤n≤106，0≤k≤n−1，0≤Ai≤103。

分析

很显然每次倒水之后，水总是单调递增的，那么肯定要把这个更多的水，倒到下一个水壶里面去。也就是说每次倒水的范围是一个区间。

因此问题就很显然地转化为了求一个长度为 k+1k+1 的最大连续子段和，就可以很简单地水过去了。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <cctype>
#include <queue>
#include <vector>

using namespace std;

inline int read()
{
	int x=0,f=1;char ch=getchar();
	while (!isdigit(ch)){if (ch=='-') f=-1;ch=getchar();}
	while (isdigit(ch)){x=x*10+ch-48;ch=getchar();}
	return x*f;
}

int n,k,a[1000050];

long long sum,ans;

int main()
{
	freopen("kettle.in","r",stdin);
	freopen("kettle.out","w",stdout);
	n=read(),k=read();
	for (int i=1;i<=n;i++)
		a[i]=read();
	int l=1,r=k+1;
	for (int i=l;i<=r;i++)
		sum+=a[i];
	while (r<=n)
	{
		ans=max(ans,sum);
		l++;
		r++;
		sum-=a[l-1];
		sum+=a[r];
	}
	cout << ans << endl;
	return 0;
}

这个方法对新手来说比较复杂请看下一种：

• 题意简化： 给定一个数列与区间长度，求数列中最大的区间和。

• 为什么我们可以这样想呢？

我们可以贪心地想一下，为了喝掉最多的水，肯定是要把可以倒的水全部倒到自己的杯子里。而且题目说了每杯水只能倒到后一个杯子里面，所以就是一直连续倒到后一个杯子，自己喝最后一杯就是最多的。（说了一大堆还不要是求区间加）

• 暴力 O(n2)O(n2)

我们就直接暴力枚举左端点，左端点加上 kk 就得到了右端点，然后再对于这个区间求和。（这么看上去貌似是 O(nk)O(nk) 的，但是鉴于 k≤n−1k≤n−1 就夸张点写成 O(n2)O(n2) 吧）

但是我们一看数据 ， n≤106n≤106 ，肯定会爆 qwqqwq

• 前缀和优化 O(n)O(n)

看到求区间和，我们很容易可以想到可以 O(1)O(1) 搞出区间和的好东西：前缀和！！！

我们还是枚举左端点，然后算出右端点 ，再用前缀和求区间加，记录一下区间加的最大值就好啦！

其实这个复杂度还有一个不小的常数 ，不过还是能 ACAC 的就不管了

• 代码如下：

#include<cstdio>
#include<algorithm>
using namespace std;

int q[1000005],maxx=0;//不用开 long long ，根据题目最大前缀应该是 int 能存下的
int a[1000005];

int main()
{
	//freopen("kettle.in","r",stdin);
	//freopen("kettle.out","w",stdout);
	int n,k;
	scanf("%d%d",&n,&k);
	for(int i=1;i<=n;i++)//输入数据，初始化前缀和
	{
		scanf("%d",&a[i]);
		q[i]=q[i-1]+a[i];
	}
	for(int i=1;i<=n-k;i++)//前缀和找最大区间加
	{
		maxx=max(maxx,q[i+k]-q[i-1]);
	}
	printf("%d",maxx);//输出
	//fclose(stdin);fclose(stdout);
	return 0;
 } 

 蒟蒻作者发文，大佬勿喷，多多指教 

点个赞呗