瞌睡C++

题目描述

小易觉得高数课太无聊了，决定睡觉。不过他对课上的一些内容挺感兴趣，所以希望你在老师讲到有趣的部分的时候叫醒他一下。你知道了小易对一堂课每分钟知识点的感兴趣程度，并以分数量化，以及他在这堂课上每分钟是否会睡着，你可以叫醒他一次，这会使得他在接下来的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

输出

16

思路

总的分数=醒着的时候学习兴趣分数+拍醒时候的学习兴趣分数

代码

#include<iostream>
using namespace std;
int main(){
    int n,k;
    cin>>n>>k;
    int score[n],wake[n];
    int i,j,max=0,count=0,sum=0;
    for(i=0;i<n;i++)
    {
        cin>>score[i];
    }
    for(i=0;i<n;i++)
    {
        cin>>wake[i];
        if(wake[i]==1)
            sum+=score[i];//醒着的分数
    } 
    //cout<<"sum "<<sum<<endl;
//选择拍醒后，选择k个连续区间分数累计最高的
    for(i=0;n-i>=k;i++)
    {
        count=0;
        for(j=i;j<n&&j<i+k;j++)
        {
            if(!wake[j])
                count+=score[j];
        }
        max=max>count?max:count;
    }
    sum+=max;
    cout<<sum<<endl;
    
    return 0;
}

PS：

a. 第一次AC90%，卡在拍醒后，计算连续k个区间的分数，判断了两次，Wrong，修改后AC100%

b. AC的条件是拍醒后，只选了分数最高的，并不是选择多个符合条件的分数，比如：

13 4

1 2 3 5 6 4 5 8 5 5 5 6 3

0 0 0 0 0 1 1 1 1 0 0 0 0

但是2 3 5 6和5 5 6 3都可以拍，按照算法，拍醒的区间是 5 5 6 3，这样就没有那么难了。