CSP/信奥赛C++刷题训练:经典前缀和例题(2):洛谷P6568:水壶

CSP/信奥赛C++刷题训练:经典前缀和例题(2)

[NOI Online #3 提高组] 水壶

题目描述

有 n n n 个容量无穷大的水壶,它们从 1 ∼ n 1\sim n 1∼n 编号,初始时 i i i 号水壶中装有 A i A_i Ai 单位的水。

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

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

输入格式

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

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

第三行 n n n 个非负整数,相邻两个数用空格隔开,表示水壶的初始装水量 A 1 A_1 A1, A 2 A_2 A2, ⋯ \cdots ⋯, A n A_n An。

输出格式

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

样例 #1

样例输入 #1

复制代码
10
5
890 965 256 419 296 987 45 676 976 742

样例输出 #1

复制代码
3813

提示

数据规模与约定
  • 对于 10 % 10\% 10% 的数据,保证 n ≤ 10 n \leq 10 n≤10。
  • 对于 30 % 30\% 30% 的数据,保证 n ≤ 100 n \leq 100 n≤100。
  • 对于 50 % 50\% 50% 的数据,保证 n ≤ 1 0 3 n \leq 10^3 n≤103。
  • 对于 70 % 70\% 70% 的数据,保证 n ≤ 1 0 5 n \leq 10^5 n≤105。
  • 对于 100 % 100\% 100% 的数据,保证 1 ≤ n ≤ 1 0 6 1\leq n\leq 10^6 1≤n≤106, 0 ≤ k ≤ n − 1 0\leq k \leq n-1 0≤k≤n−1, 0 ≤ A i ≤ 1 0 3 0\le A_i\le 10^3 0≤Ai≤103。

使用前缀和解题

cpp 复制代码
#include<bits/stdc++.h>
using namespace std;
//前缀和 
const int N=1e6+10;
int n,k,a[N],q[N],ans;//q数组存前缀和 
int main(){
	cin>>n;
	cin>>k;
	for(int i=1;i<=n;i++){
		cin>>a[i];
		q[i]=q[i-1]+a[i];//初始化前缀和 
	} 
	for(int i=1;i<=n-1;i++){//前缀和找最大区间和 
		ans=max(ans,q[i+k]-q[i-1]); 
	}
	cout<<ans;
	return 0;
}

文末彩蛋:

点击王老师青少年编程主页有更多精彩内容

相关推荐
Tiandaren4 小时前
Selenium 4 教程:自动化 WebDriver 管理与 Cookie 提取 || 用于解决chromedriver版本不匹配问题
selenium·测试工具·算法·自动化
岁忧5 小时前
(LeetCode 面试经典 150 题 ) 11. 盛最多水的容器 (贪心+双指针)
java·c++·算法·leetcode·面试·go
chao_7895 小时前
二分查找篇——搜索旋转排序数组【LeetCode】两次二分查找
开发语言·数据结构·python·算法·leetcode
秋说7 小时前
【PTA数据结构 | C语言版】一元多项式求导
c语言·数据结构·算法
Maybyy7 小时前
力扣61.旋转链表
算法·leetcode·链表
卡卡卡卡罗特9 小时前
每日mysql
数据结构·算法
chao_7899 小时前
二分查找篇——搜索旋转排序数组【LeetCode】一次二分查找
数据结构·python·算法·leetcode·二分查找
蜉蝣之翼❉10 小时前
CRT 不同会导致 fopen 地址不同
c++·mfc
aramae10 小时前
C++ -- STL -- vector
开发语言·c++·笔记·后端·visual studio
lifallen10 小时前
Paimon 原子提交实现
java·大数据·数据结构·数据库·后端·算法