leetcode 219

219. 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.

题意：给定一个数组和一个整数k，找出数组中是否存在两个距离不超过k的相同的数。存在返回true；反之，返回false.

思路：我们首先想到的是将数组中的每个元素与它周围距离在k以内的元素对比，这样算法的时间复杂度为O（n²），算法效率太差。如果用set存放每次需要对比的数，那么每次查找的次数从k减少为logk，这样会大大提高效率。

代码如下：

class Solution {
public:
    bool containsNearbyDuplicate(vector<int>& nums, int k) {
        set<int> s;
        if(k <= 0)
        {
            return false;
        }
        if(k >= nums.size())
        {
            k = nums.size() - 1;
        }
        for( int i = 0; i < nums.size(); 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;
    }
};