【网易】瞌睡

题目描述
小易觉得高数课太无聊了，决定睡觉。不过他对课上的一些内容挺感兴趣，所以希望你在老师讲到有趣的部分的时候叫醒他一下。你知道了小易对一堂课每分钟知识点的感兴趣程度，并以分数量化，以及他在这堂课上每分钟是否会睡着，你可以叫醒他一次，这会使得他在接下来的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
思路：首先在输入ti的同时先计算出ti1的兴趣和，然后再遍历一遍兴趣数组a，暴力计算其中每k个ti为0的兴趣和，记录下最大值，加上之前的ti1的兴趣和即可
后来优化一下，我们记录一下ti0的坐标，在计算k个区间内ti0的兴趣和时会省去很多时间
未优化240ms左右，优化后140ms左右

#include <iostream>
#include <algorithm>
using namespace std;
int main()
{
    int n,k;
    cin>>n>>k;
    int* a=new int [n+1];
    int* t=new int [n+1];
    vector<int> record;
    long long ans=0;
    for(int i=1;i<=n;++i)
        cin>>a[i];
    for(int i=1;i<=n;++i)
    {
        cin>>t[i];
        if(t[i])
            ans+=a[i];
        else
            record.push_back(i);
    }
    if(n<=k)
    {
        int sum=0;
        for(int i=1;i<=n;++i)
            sum+=a[i];
        cout<<sum<<endl;
    }
    else
    {
        long long maxTemp=0;
        for(int i=0;i<record.size();++i)
        {
            long long temp=0;
            int end=(record[i]+k-1>n)?n:record[i]+k-1;
            for(int j=record[i];j<=end;++j)
            {
                if(!t[j]) temp+=a[j];
            }
            maxTemp=max(maxTemp,temp);
        }
        ans+=maxTemp;
        delete[] a;
        delete[] t;
        cout<<ans<<endl;
    }
    return 0;
}