算法問題(查找表:Set,Map)--存在重複元素II

LeetCode第219號題–存在重複元素II

題目如下

給定一個整數數組和一個整數 k，判斷數組中是否存在兩個不同的索引 i 和 j，使得 nums [i] = nums [j]，並且 i 和 j 的差的絕對值最大爲 k。

示例 1:

輸入: nums = [1,2,3,1], k = 3
輸出: true
示例 2:

輸入: nums = [1,0,1,1], k = 1
輸出: true
示例 3:

輸入: nums = [1,2,3,1,2,3], k = 2
輸出: false

解答
這道題我們使用Set和Map都可以，但是使用Set的空間複雜度只需要爲O(k)，而Map需要O(n)，兩者的時間複雜度都是爲O(n)

方法一：Set，使用HashSet
我們這裏也要有滑動窗口的思想去解決此問題

// 使用HashSet
    // 時間複雜度: O(n)
    // 空間複雜度: O(k)
    public boolean containsNearbyDuplicate(int[] nums, int k) {
        if (nums == null || nums.length <= 1)
            return false;

        if (k <= 0)
            return false;

        HashSet<Integer> record = new HashSet<>();
        for (int i = 0; i < nums.length; i++) {
            if (record.contains(nums[i]))
                return true;

            record.add(nums[i]);
            //我們確保數據不超過k+1，使得兩數的差的絕對值不超過 k
            if (record.size() == k + 1)
                record.remove(nums[i - k]);
        }

        return false;
    }

方法一：Map，使用HashMap

// 使用HashMap
    // 時間複雜度: O(n)
    // 空間複雜度: O(n)
    public boolean containsNearbyDuplicate(int[] nums, int k) {

        if (nums == null || nums.length <= 1)
            return false;

        if (k <= 0)
            return false;

        HashMap<Integer, Integer> record = new HashMap<>();
        record.put(nums[0], 0);
        for (int i = 1; i < nums.length; i++) {
            if (record.containsKey(nums[i]) && i - record.get(nums[i]) <= k)
                return true;
            record.put(nums[i], i);
        }

        return false;
    }

還有一道更簡單的題，是在本題中減去了一個條件，實現起來更簡單：
LeetCode第217號題–存在重複元素

還有此題的升級版
LeetCode第220號題–存在重複元素III