1. 程式人生 > >LeetCode演算法題219:存在重複元素 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

這個題和217題類似,只是多了一個條件,所以這裡用map雜湊表存下元素及其索引,然後找到相同元素時判斷是否與當前索引相差值小於k,是則返回true,如果沒有相同元素則繼續新增到表中。 C++原始碼:

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

python3原始碼:

class Solution:
    def containsNearbyDuplicate(self, nums, k):
        """
        :type nums: List[int]
        :type k: int
        :rtype: bool
        """
        m = {}
        for i in range(len(nums)):
            if nums[i]
in m and i-m[nums[i]]<=k: return True else: m[nums[i]] = i return False