g1101_1200.s1124_longest_well_performing_interval.Solution Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of leetcode-in-java Show documentation

Java-based LeetCode algorithm problem solutions, regularly updated

There is a newer version: 1.37

package g1101_1200.s1124_longest_well_performing_interval;

// #Medium #Array #Hash_Table #Stack #Prefix_Sum #Monotonic_Stack
// #2022_03_09_Time_13_ms_(71.38%)_Space_54_MB_(37.10%)

import java.util.HashMap;

public class Solution {
    public int longestWPI(int[] hours) {
        int i = 0;
        HashMap map = new HashMap<>();
        int sum = 0;
        map.put(sum, -1);
        int max = Integer.MIN_VALUE;
        for (int val : hours) {
            sum += (val > 8) ? 1 : -1;
            if (!map.containsKey(sum)) {
                map.put(sum, i);
            }
            if (sum > 0) {
                max = i + 1;
            } else if (map.containsKey(sum - 1)) {
                max = Math.max(i - map.get(sum - 1), max);
            }
            i++;
        }
        if (max == Integer.MIN_VALUE) {
            max = 0;
        }
        return max;
    }
}