AcWing每日一题 3493. 最大的和

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

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

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

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

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

第二行包含 n 个整数 ai。

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

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

数据范围
对于 30% 的数据，1≤k≤n≤1000
对于 100% 的数据，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>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long LL;
const int N = 100010;
int a[N],b[N];
int m,n;
int main()
{
    cin>>n>>m;
    for (int i = 0; i < n; i ++ )
    cin>>a[i];
    for (int i = 0; i < n; i ++ )
    cin>>b[i];
    LL sum=0;
    for(int i=0;i<n;i++)
        if(b[i]) sum+=a[i];
    LL v=0,thissum=0;
    for(int i=0;i<n;i++)
    {
        if(!b[i]) thissum+=a[i];
        if(i>=m&&!b[i-m]) thissum-=a[i-m];
        v=max(v,thissum);
    }
    cout<<sum+v;
    return 0;
}