AcWing每日一题 3493.最大的和（前缀和）

最大的和

原题链接

给定一个长度为 n 的正整数数列 a1,a2,…,an。

初始时，数列中的每个元素要么处于可选状态，要么处于不可选状态。

你可以选择一个长度恰好为 k 的区间 [i,i+k−1]，使得 ai∼ai+k−1 这 k 个元素的状态全部变为可选。

请问，在经过此操作后，所有处于可选状态的元素之和最大是多少。

数据范围
对于 30% 的数据，1≤k≤n≤1000
对于 100% 的数据，1≤k≤n≤105,1≤ai≤105

输入格式
第一行包含两个整数 n 和 k。

第二行包含 n 个整数 ai。

第三行包含一个长度为 n 的 01 序列，如果第 i 个数为 1，表示 ai 的初始状态为可选，如果第 i 个数为 0，表示 ai 的初始状态为不可选。

输出格式
一行一个整数，表示答案。

输入样例1：
3 1
2 5 4
0 0 1
输出样例1：
9
输入样例2：
4 3
10 5 4 7
0 1 1 0
输出样例2：
19

这次的题目比较简单，只需要维护两个前缀和数组就可以AC，一个是所有数的前缀和数组，一个只算可选数的前缀和数组，然后枚举修改的区间的起点，通过两个前缀和数组来计算当前修改后的值取最大值即可

AC代码如下：

//手机双击代码可以全屏哦，这真不是骗人的！
#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int maxn = 1e5 + 3;
int a[maxn];
int flag[maxn];
ll sum1[maxn],sum2[maxn];
int main(){
	int n, k;
	scanf("%d %d",&n, &k);
	for(int i = 1;i <= n;++ i){
		scanf("%d",&a[i]);
		sum1[i] = sum1[i-1] + a[i];// 
	}
	for(int i = 1;i <= n;++ i){
		scanf("%d",&flag[i]);
		if(flag[i] == 1)sum2[i] = sum2[i-1] + a[i];
		else sum2[i] = sum2[i-1];
	}
	ll ans = sum2[n];
	for(int i = 1;i <= n-k+1;++ i){
		ans = max(ans, sum2[n] + sum1[i+k-1] - sum1[i-1] - (sum2[i+k-1] - sum2[i-1]));
	}//sum1[i+k-1] - sum1[i-1]即修改后区间内所有数的值
	//sum2[i+k-1] - sum2[i-1]即原区间内所有可选数的值 
	printf("%lld",ans);
	return 0;
}