题目描述:
给定一个整数数组 a,其中1 ≤ a[i] ≤ n (n为数组长度), 其中有些元素出现两次而其他元素出现一次。
找到所有出现两次的元素。
你可以不用到任何额外空间并在O(n)时间复杂度内解决这个问题吗?
示例:
输入:
[4,3,2,7,8,2,3,1]
输出:
[2,3]
解题思路:
桶计数(简单但不符合题意)
使用int数组对nums数组中的元素进行计数,然后遍历这个数组,将出现次数为2的元素加入结果list中即可
标记法
由于题目给定nums[i]的范围是[1,n],且元素最多出现两次,所以我们可以根据nums[i]对应的值作为数组下标索引到nums[nums[i] - 1]的值,对其做标记,如果nums[i]出现两次,那么第二次出现的nums[i]对应的nums[nums[i] - 1]就已经被标记过了
如果是相同的元素,那么以他们为索引的元素值一定是同一个值,因此可以修改该值来标记是否被访问过
所以在遍历的过程中:
1、我们首先确定nums[nums[i] - 1]是否被标记,如果被标记的话,那就说明nums[i]出现了两次,将其加入结果list中;如果没有被标记则对其进行标记
2、这里的做标记就是将nums[nums[i] - 1]对应的值设为其相反数,也就是将其乘以-1(当然也可以将其加一个超过nums[i]范围的值,只要遍历的时候方便还原就可以)
3、还有重要的一点就是,在遍历的过程中,我们应该首先还原nums[i],否则当出现负数的时候对应的索引会出错,还原规则根据标记的规则而定,这里由于是取其相反数,所以我们还原就是取绝对值
详细代码:
桶计数
public List<Integer> fast_findDuplicates(int[] nums) {
int n = nums.length;
int[] counts = new int[n + 1];
List<Integer> res = new ArrayList<>();
for (int i = 0; i < n; i++) {
counts[nums[i]]++;
}
for (int i = 0; i <= n; i++) {
if (counts[i] == 2) {
res.add(i);
}
}
return res;
}
标记法
public List<Integer> findDuplicates(int[] nums) {
int n = nums.length;
//创建结果数组
List<Integer> res = new ArrayList<>();
for (int i = 0; i < n; i++) {
//首先还原nums[i],防止出现负数
int abs = (nums[i]) > 0 ? (nums[i]) : (-nums[i]);
//然后根据还原后的nums[i]判断nums[nums[i] - 1]是否被标记
if (nums[abs - 1] < 0) {
res.add(abs);
} else {
nums[abs - 1] *= -1;
}
}
return res;
}
最后
欢迎访问我的个人网站:风在哪个人博客
与你分享我的学习历程!