【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的大小。

猜你喜欢

转载自blog.csdn.net/qq_40315080/article/details/116156140