【leetcode】存在重复元素 IIc++

题目描述：
给定一个整数数组和一个整数 k，判断数组中是否存在两个不同的索引 i 和 j，使得 nums [i] = nums [j]，并且 i 和 j 的差的 绝对值 至多为 k。

示例1：

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

示例2：

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

示例3：

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

代码：

#define max 100000
class Solution {
public:
    bool containsNearbyDuplicate(vector<int>& nums, int k) {
        int n = nums.size();
        if(n<2)return false;
        int a[max],b[max],flag_same=0,flag_k=0;
        for(int i=0;i<max;i++){
            a[i]=0;
            b[i]=0;
        }
        for(int i=0;i<n;i++){
            flag_same=0;
            flag_k=0;
            if(a[nums[i]+50000]==0){
                a[nums[i]+50000]=1;
                b[nums[i]+50000]=i;
            }
            else if(a[nums[i]+50000]!=0){//满足num[i]=nums[j]
                flag_same=1;
                if(i-b[nums[i]+50000]<=k){ //满足差的绝对值最大为k  
                    flag_k=1;
                    cout<<"!"<<endl;
                    return true;
                }
                else{
                    b[nums[i]+50000]=i;
                }
            }
        }
        return false;
    }
};

牺牲空间换时间，直接查找遍历会超时。

结合示例理解题意：当找到两个相等且位置差<=k的数则返回true，找不到则返回false。

把nums[i]当作数组下标，用数组a的值记录当前位置的数有无出现过，nums[i]出现过，则a[nums[i]]=1，否则a[nums[i]]=0。

把nums[i]当作数组下标，用另一个数组b记录nums[i]上一次出现位置的下标。只要离得最近的相等数的位置差满足<=k则可返回true，且查找时从前向后遍历，所以每次更新b[nums[i]]使得下一个找到的相同数与其上一个位置的差最小，判断差值与k的大小。