题目
leetcode 219. 存在重复元素 II
给定一个整数数组和一个整数 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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/contains-duplicate-ii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路
1、题目描述“两个不同的索引 i 和 j,那么存在:k != 0
注:k!=0很重要,如果在函数开始的时候没有判断边界,那么后面的set的包含判断需要加上k的一些判断。
2、想象动态过程可以发现,其实是一个“滑块”,左边逐渐减1,右边逐渐加1(这里类似当年大学学flash实现移动效果的原理,扯远了…),那么需要维护这个“滑块”,考虑到是否存在nums [i] = nums [j],存在性问题转换成包含问题,联想到Set的概念
代码
class Solution {
public boolean containsNearbyDuplicate(int[] nums, int k) {
//因为不同索引,所以k!=0
if(k<=0){
return false;
}
Set<Integer> set = new HashSet<Integer>();
for(int i=0;i <nums.length; i++){
//如果包含,则有相等
if(set.contains(nums[i])){
return true;
}
//如果size到达k,那么删除最前面的
if(set.size()==k){
//根据下标可以找到最前面的
set.remove(nums[i-k]);
}
set.add(nums[i]);
}
return false;
}
}
复杂度
时间复杂度:因为每次都是set的操作,故为O(n)
空间复杂度:数组本身加上一个set的大小,故为O(min(n,k))