题目描述:
给定一个整数数组和一个整数 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的大小。