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