網易2019校招筆試題-瞌睡

分析：由於小易清醒的時間是連續的，所以整個搜索空間爲O(n)，根本不需要貪心或者動態規劃就能搞定。 小易這堂課獲得的興趣值分爲兩部分：本來就清醒時獲得的興趣值，被叫醒的k分鐘獲得的興趣值。因爲第一部分是固定的，所以當第二部分最大時，整體取最大值。
思路：第一部分的興趣值爲 fixed_gain，第二部分，如果在第i分鐘叫醒小易，額外獲得的興趣值爲gain[i]，gain[i]和gain[i-1]的關係如下：

// 只有當 t[i-1] 爲 0 時，纔是由於叫醒他而額外獲得的興趣值。
if( i+k-1 <=n )  
  gain[i] = gain[i-1] - a[i-1]*(1-t[i-1]) + a[i+k-1]*(1-t[i+k-1]);
else
  gain[i] = gain[i-1] - a[i-1]*(1-t[i-1]);

代碼：（37ms）

// 分析：由於是連續的，遍歷一次就OK
#include <iostream>
using namespace std;

int main()
{
    int n,k; // n 個數，k表示清醒時間
    int a[100002] ={0}; // 興趣度
    int t[100002] ={0}; // 是否清醒
    int gain[10002] ={0}; // gain[i]:第 i 分鐘叫醒小易，獲得的收益

    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;i++)
        scanf("%d",&a[i]);
    int fixed_gain = 0;     // 本來清醒時候的收益，這個是固定的
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&t[i]);
        if( t[i] == 1 )
            fixed_gain += a[i];
    }

    /************* 簡單地遍歷一次即可 **************/
    for(int i=1;i<=n;i++)
    {
        if(i==1)
        {
            for(int j=i;j<=i+k-1 && j<=n; j++)
                if( t[j] == 0 )
                    gain[i] += a[j];
        }
        else{
            if( i+k-1 <=n )
                gain[i] = gain[i-1] - a[i-1]*(1-t[i-1]) + a[i+k-1]*(1-t[i+k-1]);
            else
                gain[i] = gain[i-1] - a[i-1]*(1-t[i-1]);
        }

    }
    // 找最大的收益
    int max_gain =0;
    for(int i=1;i<=n-k+1 && i <=n;i++)
        max_gain = max(max_gain, gain[i]);
    cout << max_gain + fixed_gain <<endl;

    return 0;
}