单调栈系列-1124.表现良好的最长时间段

难度中等175收藏分享切换为英文接收动态反馈

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

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

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

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

示例 1：

**输入：**hours = [9,9,6,0,6,6,9]
**输出：**3
**解释：**最长的表现良好时间段是 [9,9,6]。

示例 2：

**输入：**hours = [6,6,6]
**输出：**0

提示：

• 1 <= hours.length <= 104
• 0 <= hours[i] <= 16

解题思路：

public class P1124LongestWellPerformingInterval {
    public static void main(String[] args) {
        Solution solution = new P1124LongestWellPerformingInterval().new Solution();
        int[]hours = new int[]{14,16,1,11,3,5,3};
        System.out.println("Hello world " + solution.longestWPI(hours));

        hours = new int[]{6,6,9};
        System.out.println("Hello world " + solution.longestWPI(hours));
    }
    //leetcode submit region begin(Prohibit modification and deletion)
class Solution {
    public int longestWPI(int[] hours) {
        int len = hours.length;
        int[] hours2 = new int[len];
        for (int i = 0; i < len; i++) {
            if (hours[i] > 8) {
                hours2[i] = 1;
            } else {
                hours2[i] = -1;
            }
        }
        int[] prefixSum = new int[len + 1];
        for (int i = 1; i < len + 1; i++) {
            prefixSum[i] = prefixSum[i - 1] + hours2[i - 1];
        }
        Deque<Integer> stack = new LinkedList<>();
        for (int i = 0; i < len + 1; i++) {
            if (stack.isEmpty() || prefixSum[stack.peek()] > prefixSum[i]) {
                stack.push(i);
            }
        }
        int ans = 0;
        for (int j = len; j >= ans; j--) {
            while (!stack.isEmpty()) {
                int i = stack.peek();
                if (prefixSum[j] > prefixSum[i]) {
                    ans = Math.max(ans, j - i);
                    stack.pop();
                } else {
                    break;
                }
            }

        }
        return ans;

    }
}