697. Degree of an Array - LeetCode

697. Degree of an Array - LeetCode

Question

697. Degree of an Array - LeetCode

Solution

理解两个概念：

数组的度：[1,2,2,3,1]这个数组，去重后的元素为[1,2,3]，每个元素在原数组中重复的次数分别是221，数组的度就是元素最大重复次数，这个数组中元素最大重复次数就是2

连续子数组：和字符串中的了串概念类似，数组元素顺序保持不变，取其中一部分，该题就是求在度相同的情况下最小连续子数组的长度

思路：数组的度是以元素为对象，每个元素都有一个度(degree)，第一次出现的坐标(start)，最后一次出现的坐标(end)，求出degree最大,end-start最小的那个情况，end-start+1就是该数组最小连续子数组的长度了。

public int findShortestSubArray(int[] nums) {
    int start = 0;
    int end = 0;
    int degree = 0;
    Map<Integer, Element> numElementMap = new HashMap<>();

    for (int i = 0; i < nums.length; i++) {
        int num = nums[i];
        Element element = numElementMap.get(num);
        if (element == null) {
            element = new Element(i, 0);
            numElementMap.put(num, element);
        }
        element.degree++;
        int elementEnd  = i;

        // degree最大 end-start最小
        boolean change = i == 0 ? true : false; // 第一个元素要初始化
        if (element.degree >= degree) {
            change = true;
            if (element.degree == degree && (end - start < elementEnd - element.start)) {
                change = false;
            }
        }
        if (change) {
            start = element.start;
            end = elementEnd;
            degree = element.degree;
        }

    }
    return end - start + 1;
}

class Element {
    int start;
    int degree;

    public Element(int start, int degree) {
        this.start = start;
        this.degree = degree;
    }
}

Reference

• https://gitee.com/okokabcd/leetcode