leetcode-數組-簡單-存在重複元素 II

題目

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

來源：力扣（LeetCode）
鏈接：https://leetcode-cn.com/problems/contains-duplicate-ii
著作權歸領釦網絡所有。商業轉載請聯繫官方授權，非商業轉載請註明出處。

示例 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

解答

方法一：

窮舉法

class Solution {
public:
    bool containsNearbyDuplicate(vector<int>& nums, int k) {
        for (int i = 0; i < nums.size(); ++i) {
            for (int j = i + 1; j < nums.size(); ++j) {
                if (nums[i] == nums[j] && ((j - i) <= k)) {
                    return true;
                }
            }
        }
        return false;
    }
};

方法二：

hashmap

class Solution {
public:
    bool containsNearbyDuplicate(vector<int>& nums, int k) {
        std::unordered_map<int, int> data_map;
        for (int i = 0; i < nums.size(); ++i) {
            if (data_map.find(nums[i]) != data_map.end()) {
                //說明這個點已經被記錄了
                if (i - data_map[nums[i]] <= k) {
                    return true;
                } else {
                    //不滿足條件就把索引更新一下
                    data_map[nums[i]] = i;
                }
            } else {
                data_map[nums[i]] = i;
            }
        }
        return false;
    }
};