leetcode 日经贴，Cpp code -Contains Duplicate III

原创于 2015-06-03 14:20:40 发布 · 395 阅读

0 ·

CC 4.0 BY-SA版权

算法同时被 3 个专栏收录

183 篇文章

订阅专栏

leetcode

173 篇文章

订阅专栏

Cpp

152 篇文章

订阅专栏

本文介绍了一种名为ContainsDuplicateIII的算法实现，该算法通过使用pair与set来判断一个整数数组中是否存在两个不同的索引i和j，使得|nums[i] - nums[j]| ≤ t并且|i - j| ≤ k。此算法适用于数据去重和数据清洗等场景。

Contains Duplicate III

class Solution {
public:
    long long abs(long long k) {
        return k>0? k:-k;
    }
    bool containsNearbyAlmostDuplicate(vector<int>& nums, int k, int t) {
        if (k <= 0) {
            return false;
        }
        set<pair<int, int> > st;
        set<pair<int, int> >::iterator itr;
        
        for (int i = 0; i < nums.size(); ++i) {
            if (i > 0) {
                itr = st.lower_bound(make_pair(nums[i], i));
                if (itr != st.end() && abs((long long)nums[i] - itr->first) <= t){
                    return true;
                }
                if (itr != st.begin()) {
                    --itr;
                    if (abs((long long)nums[i] - itr->first) <= t){
                        return true;
                    }
                }
            }
            if (i >= k) {
                st.erase(make_pair(nums[i - k], i - k));
            }
            st.insert(make_pair(nums[i], i));
        }
        return false;
    }
};