最大的和
题目描述:
给定一个长度为 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
解题思路
(%%%y总)
从题上可以知道,最终选定想要的区间后,可选的元素由两部分组成:初级的时候就被选定的+人为选定之后的。不妨设前者为S1,后者为S2,总的和为Res,那么有Res=S1+S2。显然,S1为定值,不会随选定区间的改变而改变,故Res的最大即让S2最大,(最精彩的部分来了)又选定区间中可能会包含初始被标记的,这些不能够再加一次,所以Res想取最大,其实就是最终选定的区间里,初始未被标记的元素值和最大。所以,最终的问题就转化为:找一个长度为k的区间,使得该区间内初始时未被标记为1的元素之和最大。
代码实现
核心部分在于,从前往后移动区间时,其实改变的只有两端的元素,意思就是,当前区间为[i,j]且区间内初始标记0的和为sum,往下移动一个之后变为[i+1,j+1],sum的改变包括两部分:
- 当i>=k+1时,若第i个位置上的元素ai初始标记为0,则sum-=a[i]
- 若第j+1个位置上的元素ai初始标记为0,则sum-=a[j+1]
下面给出代码:
#include<iostream>
#include<algorithm>
#define ll long long
using namespace std;
const int N=1e5+5;
int a[N];
bool opt[N];
int main()
{
int n,k;
ll sum=0,temps=0;
cin>>n>>k;
for(int i=1;i<=n;i++)
cin>>a[i];
for(int i=1;i<=n;i++)
{
cin>>opt[i];
if(opt[i])
sum+=a[i];
}
ll res=0;
for(int i=1;i<=n;i++)
{
if(!opt[i])
temps+=a[i];
if(i>=k+1&&!opt[i-k])
temps-=a[i-k];
res=max(res,temps);
}
res+=sum;
cout<<res;
return 0;
}