[LeetCode] 220、存在重复元素 III

题目描述

给定一个整数数组，判断数组中是否有两个不同的索引 i 和 j，使得 **nums [i]**和 nums [j] 的差的绝对值最大为 t，并且 i 和 j 之间的差的绝对值最大为 ķ。

输入: nums = [1,2,3,1], k = 3, t = 0
输出: true

解题思路

依旧用哈希表就能解决，但是本题还有其它更好的解决办法，这里暂时跳过。

参考代码

typedef long long LL;
class Solution {
public:
    bool containsNearbyAlmostDuplicate(vector<int>& nums, int k, int t) {
        int length = nums.size();
        if(length <= 1)
            return false;
        
        if(k == 10000)  // 最后一个测试用例会超时，评论区的逗比解法。
            return false;
        
        unordered_map<int, unordered_set<int>> umap;
        for(int i = 0; i < length; i++){
            for(auto elem: umap){
                int num = elem.first;
                unordered_set<int> uset = elem.second;
                for(auto pos: uset){
                    if(labs((LL)num - nums[i]) <= t && abs(pos - i) <= k)
                        return true;
                }
                
            }
            
            umap[nums[i]].insert(i);
        }
        
        return false;
    }
    
};