20180814三部曲 No.1.选修课程

题目描述

LCH假期过得意犹未尽，所以上信息课的时候不免打瞌睡。但是他不希望落下老师讲的知识点，所以希望老师讲到重要地方的时候叫醒他。

现在高老师告诉你每节课每分钟知识点的重要程度，并且以分数量化。

LCH会告诉你他的睡眠计划（每分钟是睡着还是清醒），但他实在太困了，所以希望你只叫醒他一次，叫醒后他会在接下来的K分钟保持清醒，然后恢复他的睡眠计划（按照给出计划睡着或者清醒）。

作为好朋友你当然希望他在有限时间内获得最多的知识，所以你希望选择一种叫醒方案，让他获得的知识量最大。

输入

第一行输入n和k，用空格隔开，表示这堂课的总时间和叫醒一次能使LCH保持清醒的时间。

第二行输入n个用空格隔开的数，表示这节课每分钟知识点的重要程度分值ai。

第三行输入n个用空格隔开的数，表示每分钟LCH的睡眠计划，1表示他这一分钟清醒，0表示睡眠。

输出

LCH这堂课收获的最大知识量。

样例输入

6 3
1 3 5 2 5 4
1 1 0 1 0 0

样例输出

16

提示

对于100%的数据，1<=n，k<=10^5，1<=a[i]<=10^4。

#include<iostream>
using namespace std;
long long n,k,a[1000001],s[1000001],maxn,sum;
bool b[1000001];
int main()
{
//  freopen("course.in","r",stdin);
//  freopen("course.out","w",stdout);
    cin>>n>>k;
    for(int i=1;i<=n;i++)
    {
        cin>>a[i];
    }
    for(int i=1;i<=n;i++)
    {
        cin>>b[i];//输入
    }
    for(int i=1;i<=n;i++)
    {
        if(b[i]==1)
        {
            sum+=a[i];//把b[i]==1的数都加上
            s[i]=s[i-1];//前缀和 如果b[i]==1 不变
        }
        if(b[i]==0)
        {
            s[i]=s[i-1]+a[i];//把b[i]==0的情况累加
        }
    }
    for(int i=1;i<=n-k;i++)
    {
        maxn=max(maxn,s[i+k-1]-s[i-1]);//计算从i到k最大的和（前缀和）
    }
    cout<<maxn+sum;//输出最大值
}