网易2019校招内推编程题-瞌睡-C++实现

[编程题] 瞌睡

时间限制：1秒

空间限制：262144K

小易觉得高数课太无聊了，决定睡觉。不过他对课上的一些内容挺感兴趣，所以希望你在老师讲到有趣的部分的时候叫醒他一下。你知道了小易对一堂课每分钟知识点的感兴趣程度，并以分数量化，以及他在这堂课上每分钟是否会睡着，你可以叫醒他一次，这会使得他在接下来的k分钟内保持清醒。你需要选择一种方案最大化小易这堂课听到的知识点分值。

输入描述:

第一行 n, k (1 <= n, k <= 105) ，表示这堂课持续多少分钟，以及叫醒小易一次使他能够保持清醒的时间。
第二行 n 个数，a1, a2, ... , an(1 <= ai <= 104) 表示小易对每分钟知识点的感兴趣评分。
第三行 n 个数，t1, t2, ... , tn 表示每分钟小易是否清醒, 1表示清醒。

输出描述:

小易这堂课听到的知识点的最大兴趣值。

输入例子1:

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

输出例子1:

16

解法:

先计算出不叫醒小易的知识点分值，再用K值大小的窗口在兴趣程度数组线性滑动，记录最大值，二者相加即为小易这堂课听到的知识点分值。

#include<iostream>
#include<vector>
using namespace std;
int main()
{
    long long n, k, i;
    cin >> n >> k;
    vector<long long> score(n, 0);//每分钟兴趣评分
    vector<long long> isAwaken(n, 0);//每分钟清醒
    for (i = 0; i<n; ++i)
        cin >> score[i];
    for (i = 0; i<n; ++i)
        cin >> isAwaken[i];
    long long  baseInterest=0,maxAwakenInterest = 0, tempAwakenInterest = 0;
    for (i = 0; i < n;i++)//不叫醒小易知识点分值
    {
        if (isAwaken[i]==1)
            baseInterest += score[i];
    }
    for(i=0;i<k;++i)//先计算K段范围内，叫醒小易所听到的额外知识点分值
    {
        if (isAwaken[i]==0)
            tempAwakenInterest += score[i];
    }
    maxAwakenInterest=tempAwakenInterest;
    for(i=k;i<n;++i)
    {
        if (isAwaken[i]==0)  //当前K段内叫醒新增知识点分值
            tempAwakenInterest += score[i];
        if(isAwaken[i-k]==0)//去除在前一个k段内但不在当前k段内的知识点分值
            tempAwakenInterest -= score[i-k];
        if (tempAwakenInterest>maxAwakenInterest)//记录叫醒新增知识点分值最大值
            maxAwakenInterest = tempAwakenInterest;
    }
    cout << baseInterest + maxAwakenInterest << endl;
    return 0;
}