计大于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;
}
};