网易校招题之瞌睡

题目描述

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

输入描述:

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

输出描述:

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

输入

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

输出

16
思路
清醒时刻的兴趣值加上在K中睡着的兴趣值
先统计清醒时刻的兴趣值总和,再利用滑动窗口,计算窗口内瞌睡的值
#include <bits/stdc++.h>
using namespace std;
 
int main()
{
    int n,k;
    cin>>n>>k;
    int s=0;
        vector<int>a(n,0);
        vector<int>b(n,0);
    for(int i=0;i<n;i++)
        cin>>a[i];
    for(int i=0;i<n;i++)
    {
        cin>>b[i];
        if(b[i])
        {
             s += a[i];//统计清醒时刻的兴趣值
        }           
    }
    int Max = 0;
    for(int i=0;i<=n-k;i++)
    {
        int t = 0;
        for(int j=0;j<k;j++)//定义窗口
        {
            if(b[i+j]==0)
            {
                t += a[i+j];//统计窗口内的瞌睡值
            }
        }
        Max = max(Max, t);//最大值更新
    }
    cout<<s+Max<<endl;
    return 0;
}

全部评论

相关推荐

评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客企业服务