Contains Duplicate II

題目來源
Given an array of integers and an integer k, find out whether there are two distinct indices i and j in the array such that nums[i] = nums[j] and the absolute difference between i and j is at most k.
第一想法就是把每種情況都列一下,復(fù)雜度O(kn)。

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

不出意外TLE了。
然后又想了想,用哈希,記錄下每個(gè)數(shù)字出現(xiàn)的位置。

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

看了看答案,發(fā)現(xiàn)只要記錄k個(gè)就可以了,里面有重復(fù)的就true。

class Solution {
public:
    bool containsNearbyDuplicate(vector<int>& nums, int k) {
        unordered_set<int> s;
        int n = nums.size();
        for (int i=0; i<n; i++) {
            if (i > k)
                s.erase(nums[i-k-1]);
            if (s.find(nums[i]) != s.end())
                return true;
            s.insert(nums[i]);
        }
        return false;
    }
};
最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡書系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容