【Leetcode】220. 存在重复元素 III

QUESTION

medium

题目描述

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

示例 1:

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

示例 2:

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

示例 3:

输入: nums = [1,5,9,1,5,9], k = 2, t = 3
输出: false

说明

---

SOLUTION

一开始我是想在 219 的基础上，加一层循环来保证 “nums [i] 和 nums [j] 的差的绝对值最大为 t” 这个条件，写了如下代码：

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

但是遇到了这个用例无法通过

[0,2147483647]
1
2147483647

在上面这种输入的情况下，这个循环 for(int j = nums[i] - t; j <= nums[i] + t; j++) 循环时间太长，导致时间超限。但是整体想法不变，加以改进，我们得到了方法一。

方法一

用有序的 map 来得到映射关系，循环的时候只判断已经形成映射关系的 key，通过 lower_bound 找到开始位置，最后使用了 long 作为存储类型防止减法产生溢出。

class Solution {
public:
    bool containsNearbyAlmostDuplicate(vector<int>& nums, int k, int t) {
        int size = nums.size();
        map<long, int> m;
        map<long, int>::iterator it;
        for(int i = 0; i < size; i++){
            it = m.lower_bound((long)nums[i] - t); //nums[i] - t 可能会溢出
            while(it != m.end()){
                if((long)it->first - t > nums[i]) break;
                if(i - it->second <= k) return true;
                it++;
            }
            m[nums[i]] = i;
        }
        return false;
    }
};

方法二

这个方法是从别的博客看来的，可能有点难理解。

这里我们使用map数据结构来解,用来记录数字和其下标之间的映射。 这里需要两个指针i和j，刚开始i和j都指向0，然后i开始向右走遍历数组，如果i和j之差大于k，且m中有nums[j]，则删除并j加一。这样保证了m中所有的数的下标之差都不大于k，然后我们用map数据结构的lower_bound()函数来找一个特定范围，就是大于或等于nums[i] - t的地方，所有小于这个阈值的数和nums[i]的差的绝对值会大于t (可自行带数检验)。然后检测后面的所有的数字，如果数的差的绝对值小于等于t，则返回true。最后遍历完整个数组返回false。代码如下：

class Solution {
public:
    bool containsNearbyAlmostDuplicate(vector<int>& nums, int k, int t) {
        map<long long, int> m;
        int j = 0;
        for (int i = 0; i < nums.size(); ++i) {
            if (i - j > k) m.erase(nums[j++]);
            auto a = m.lower_bound((long long)nums[i] - t);
            if (a != m.end() && abs(a->first - nums[i]) <= t) return true;
            m[nums[i]] = i;
        }
        return false;
    }
};