题目

请点击

给你一份工作时间表 hours，上面记录着某一位员工每天的工作小时数。

我们认为当员工一天中的工作小时数大于 8 小时的时候，那么这一天就是「劳累的一天」。

所谓「表现良好的时间段」，意味在这段时间内，「劳累的天数」是严格 大于「不劳累的天数」。

请你返回「表现良好时间段」的最大长度

思路

这道题，还是有点晕，所以写下博客，整理下思路

为什么这里大于8小时要用1表示，否则用-1表示，就不多说了，还是模拟有效括号的想法。

• int型数组pre是用来存储hours的前缀和的
  
• prePos 是个map，里面存的是 第一次出现pre[i]时的下标i
• 要想找到【劳累天数】严格大于【不劳累天数】的，那么【劳累天数】-1 ≥ 【不劳累天数】。所以接下来寻找最长的天数
• 由于，有可能出现前缀和数组pre递增的情况，就是所有的天都是【劳累的】，所以需要判断，当pre[i]一直都大于0，就说明，最长的那个天数，就是一直到最后一天。

代码

class Solution {
    
    
public:
    int longestWPI(vector<int>& hours) {
    
    
        int size = hours.size();
        vector<int> pre;
        pre.push_back(0);
        int prefix_sum = 0;
        for(int i=0;i<size;i++){
    
    
            prefix_sum += hours[i] > 8 ? 1 : -1;
            pre.push_back(prefix_sum);
        }
        map<int, int> prePos;
        int ans = 0;
        for(int i=1;i<size+1;i++){
    
    
            if(!prePos.count(pre[i])) prePos[pre[i]] = i;
            if(prePos.count(pre[i]-1)){
    
    
                ans = max(ans, i-prePos[pre[i]-1]);
            }
            if(pre[i]>0){
    
    
                ans = max(ans, i);
            }
        }
        return ans;
    }
};