g0601_0700.s0697_degree_of_an_array.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 g0601_0700.s0697_degree_of_an_array;

// #Easy #Array #Hash_Table

import java.util.HashMap;
import java.util.Map;

public class Solution {
    private static class Value {
        int count;
        int start;
        int end;

        public Value(int c, int s, int e) {
            count = c;
            start = s;
            end = e;
        }
    }

    public int findShortestSubArray(int[] nums) {
        int max = 1;
        Map map = new HashMap<>();
        for (int i = 0; i < nums.length; i++) {
            int j = nums[i];
            if (map.containsKey(j)) {
                Value v = map.get(j);
                v.count++;
                max = Math.max(max, v.count);
                v.end = i;
            } else {
                map.put(j, new Value(1, i, i));
            }
        }
        int min = Integer.MAX_VALUE;
        for (Map.Entry entry : map.entrySet()) {
            Value v = entry.getValue();
            if (v.count == max) {
                min = Math.min(min, v.end - v.start);
            }
        }
        return min + 1;
    }
}