leetcode 第220题
题目链接:https://leetcode-cn.com/problems/contains-duplicate-iii/
思路:最简单的思路就是暴力,在长度为k的固定窗口中寻找符合条件的元素,但是此算法的复杂度过高。所以采用滑动窗口,选择一个有序的数据结构且支持二分查找,找到符合条件的元素,在cpp中选用set.
附上代码,代码是力扣官方题解。有点懒得写了。
class Solution {
public:
bool containsNearbyAlmostDuplicate(vector<int>& nums, int k, int t) {
set<long> st;
for (int i = 0; i < nums.size(); i++) {
auto lb = st.lower_bound((long)nums[i] - t);
if (lb != st.end() && *lb <= (long)nums[i] + t) return 1;
st.insert(nums[i]);
if (i >= k) st.erase(nums[i - k]);
}
return 0;
}
};
我最开始不是这样写的!我使用了unordered_map记录元素的个数。因为我认为如果当前窗口中出现该元素被删除后会导致错误,但是很明显是不会发生的。发现如果此前一个窗口中出现两个相同的元素,那么他一定满足条件,因为他们差的绝对值是0.