给定一个长度为 n n n 的正整数数列 a 1 , a 2 , … , a n a_1,a_2,…,a_n a1,a2,…,an。
初始时,数列中的每个元素要么处于可选状态,要么处于不可选状态。
你可以选择一个长度恰好为 k k k 的区间 [ i , i + k − 1 ] [i,i+k−1] [i,i+k−1],使得 a i ∼ a i + k − 1 a_i∼a_{i+k−1} ai∼ai+k−1 这 k k k 个元素的状态全部变为可选。
请问,在经过此操作后,所有处于可选状态的元素之和最大是多少。
输入格式
第一行包含两个整数
n
n
n 和
k
k
k。
第二行包含 n n n 个整数 a i a_i ai。
第三行包含一个长度为 n n n 的 01 01 01 序列,如果第 i i i 个数为 1 1 1,表示 a i a_i ai 的初始状态为可选,如果第 i i i 个数为 0 0 0,表示 a i a_i ai 的初始状态为不可选。
输出格式
一行一个整数,表示答案。
数据范围
对于 30% 的数据,
1
≤
k
≤
n
≤
1000
1≤k≤n≤1000
1≤k≤n≤1000
对于 100% 的数据,
1
≤
k
≤
n
≤
1
0
5
,
1
≤
a
i
≤
1
0
5
1≤k≤n≤10^5,1≤a_i≤10^5
1≤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
#include<iostream>
using namespace std;
typedef long long LL;
const int N = 100010;
int n, k;
int a[N], st[N];
LL s[N];
int main(){
scanf("%d%d", &n, &k);
for(int i = 1; i <= n; i++) scanf("%d", &a[i]);
for(int i = 1; i <= n; i++) scanf("%d", &st[i]);
for(int i = 1; i <= n; i++){
s[i] = s[i - 1];
if(st[i] == 0) s[i] += a[i];
}
LL res = 0;
for(int i = 1; i + k - 1 <= n; i++)
res = max(res, s[i + k - 1] - s[i - 1]);
for(int i = 1; i <= n; i++)
if(st[i])
res += a[i];
printf("%lld\n", res);
return 0;
}