一维前缀和
这个优化主要是用来在O(1)时间内求出一个序列a中,a[i]+a[i+1]+……+a[j]的和。
具体原理十分简单:用sum[i]表示(a[1]+a[2]+……+a[i]),其中sum[0]=0,则(a[i]+a[i+1]+……+a[j])即等于sum[j]-sum[i-1]。
Description
你的朋友Mishka和你参加一个微积分讲座。讲座持续n分钟。讲师在第i分钟讲述ai个定理。
米什卡真的对微积分很感兴趣,尽管在演讲的所有时间都很难保持清醒。给你一个米什卡行为的数组t。如果Mishka在演讲的第一分钟睡着了,那么ti将等于0,否则它将等于1.当米什卡醒来时,他写下讲师所讲述的所有定理 。否则,他什么都不写。
你知道一些方式让Mishka直线醒过来持续k分钟。但是,您只能使用一次。你可以在1和n - k + 1之间的任何一分钟开始使用它。如果你在一分钟内使用它,那么Mishka会在几分钟内醒来,并且会写下讲师所讲的所有定理。
你的任务是计算Mishka能够记下的最大定理数,如果你只用一次技巧来唤醒他。
Input
输入的第一行包含两个整数n和k(1≤k≤n≤105) - 以分钟为单位的讲座持续时间以及可让Mishka保持清醒状态的分钟数。
输入的第二行包含n个整数a1,a2,… an(1≤ai≤104) - 讲师在第i分钟讲述的定理数。
输入的第三行包含n个整数t1,t2,… tn(0≤ti≤1) - 讲座第i分钟Mishka行为的类型。
Output
只打印一个整数 - 只用一次技巧来唤醒Mishka,他可以记下的定理的最大数量。
Sample Input
Input
6 3
1 3 5 2 5 4
1 1 0 1 0 0
Output
16
Hint
在示例情况下,最好的方法是在第三分钟开始时使用。 那么Mishka将能记下的定理数将等于16。
代码:
#include <iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#define MAX 100000+50
using namespace std;
int a[MAX];
int t[MAX];
int pre[MAX];
int main()
{
int n,k,sum=0,s=0;
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",&t[i]);
if(t[i]==1) \\t[i]为1时表示没有睡,把这些加到sum里
sum+=a[i];
}
memset(pre,0,sizeof(pre));
pre[0]=0;
for(int i=1;i<=n;i++)
{
if(t[i])
pre[i]=pre[i-1]+0;
else
pre[i]=pre[i-1]+a[i]; //找前缀和
}
for(int i=1;i<=n;i++)
s=max(s,pre[i+k-1]-pre[i-1]); //用pre[i+k-1]-pre[i-1]保证不会超出数组范围
s+=sum;
printf("%d\n",s);
return 0;
}