JAVA程序设计：表现良好的最长时间段（LeetCode：1124）

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

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

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

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

 

示例 1：

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

提示：

1 <= hours.length <= 10000
0 <= hours[i] <= 16

思路：我们在遍历的过程中求前缀和，若当前天工作小时数大于8小时则加1，否则减一，假设当前前缀和为sum，那么以当天结尾的最长表现良好时间段一定是第一次出现前缀和为sum-1的位置的下一个位置到当前位置的长度。

class Solution {
    public int longestWPI(int[] hours) {

        int ans = 0;
        int n = hours.length;
        int[] sum = new int[n + 1];
        Map<Integer, Integer> map = new HashMap<>();

        map.put(0, 0);
        for (int i = 1; i <= n; i++) {
            sum[i] = sum[i - 1] + (hours[i - 1] > 8 ? 1 : -1);
            if (sum[i] > 0) ans = i;
            else if (map.containsKey(sum[i] - 1))
                ans = Math.max(ans, i - map.get(sum[i] - 1));
            if (!map.containsKey(sum[i]))
                map.put(sum[i], i);
        }


        return ans;

    }
}