3493. 最大的和

3493. 最大的和

 

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

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

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

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

输入格式

第一行包含两个整数 nn 和 kk。

第二行包含 nn 个整数 aiai。

第三行包含一个长度为 nn 的 0101 序列，如果第 ii 个数为 11，表示 aiai 的初始状态为可选，如果第 ii 个数为 00，表示 aiai 的初始状态为不可选。

输出格式

一行一个整数，表示答案。

数据范围

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

输入样例1：

3 1
2 5 4
0 0 1

输出样例1：

9

输入样例2：

4 3
10 5 4 7
0 1 1 0

输出样例2：

19

先求出所有为b[i]==1的a[i]和，

然后一个区间内

右边加一个b[i]==0

左边减去一个 b[i]==0

求这个区间内最大值；

 

#include<bits/stdc++.h>
using namespace std;
int main()
{
	long long n,k;
	cin>>n>>k;
	long long a[100005];
	long long b[100005];
	for(int i=0;i<n;i++)
	{
		cin>>a[i];
	}
	for(int i=0;i<n;i++)
	{
		cin>>b[i];
	}
	long long sum=0;
	for(int i=0;i<n;i++)
	{
		if(b[i])
		sum+=a[i];
	}
	long long t;
	long long maxn=0;
	for(int i=0;i<n;i++)
	{
		if(b[i]==0)t+=a[i];
		if(i>=k&&b[i-k]==0)t-=a[i-k];
		maxn=max(t,maxn);
	}
	cout<<maxn+sum<<endl;
}