LeetCode 1124. 表现良好的最长时间段 题解 C/C++

类比LeetCode 962. 最大宽度坡

计大于8小时为1，否则为0，生成前缀和数组，该前缀和数组就是962题中给定的数组，计算最大宽度，对前缀和数组做单调栈（严格单调递减）。

//前缀和+单调栈
class Solution {
public:
    int longestWPI(vector<int>& hours) {
    	int len = hours.size();
    	vector<int> prefix_sum(len+1,0);//前缀和，第一个是0

    	for(int i = 1;i<=len;++i) {
    		prefix_sum[i] = (hours[i-1] > 8 ? 1 : -1) + prefix_sum[i-1];
    	}

    	stack<int> dec_stk;//严格单调递减栈
    	for(int j = 0;j<=len;++j) {
    		if(dec_stk.empty() || prefix_sum[j] < prefix_sum[dec_stk.top()]) {
    			dec_stk.push(j);
    		}
    	}

    	//从后往前遍历 确定最长时间段
    	int ans = 0;
    	for(int k = len;k>ans;--k) {
    		while(!dec_stk.empty() && prefix_sum[k] > prefix_sum[dec_stk.top()]) {
    			ans = max(ans,k - dec_stk.top());
    			dec_stk.pop();
    		}
    	}
    	return ans;
		
    }
};