题目描述
给定一个整数数组 a,其中1 ≤ a[i] ≤ n (n为数组长度), 其中有些元素出现两次而其他元素出现一次。
找到所有出现两次的元素。
你可以不用到任何额外空间并在O(n)时间复杂度内解决这个问题吗?
示例:
输入:
[4,3,2,7,8,2,3,1]
输出:
[2,3]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/find-all-duplicates-in-an-array
原地修改法
1.本题还是448的一个思路,遍历nums的每个元素x,对应的(x-1)%n,找到对应的位置,nums[x]+=n
2.再次遍历的时候,若是x/n >=2,则push_back(i+1)
3.这里数组下标0~n-1对应的1-n的数字,值除以n对应出现的次数
vector<int> findDuplicates(vector<int>& nums) {
vector<int> ret;
int n = nums.size();
for(int i = 0; i < n; i++) {
int x = (nums[i] - 1) % n; //找对应的位置
nums[x] += n; //对应的元素的值+=n
}
//这里需要多判断一次,因为n会有临界情况
//当该位置取余等于0的时候,那么说明原来这个地方位置是n,所以x/n==3,才算是出现两次,不单单会在最后一个位置出现
for(int i = 0; i < n; i++) {
if(nums[i] % n == 0) {
if(nums[i] / n > 2) {
ret.push_back(i + 1);
}
} else if(nums[i] / n >= 2) {
ret.push_back(i + 1);
}
}
return ret;
}
原地修改法优化
1.对上面判断条件的优化
2.
4 3 2 2->4 11 6 6,显然2出现两次
4 3 2 4->4 7 6 12,显然4出现两次
2 3 2 4->2 11 7 8,显然这里2出现两次,但是4只出现过一次,但是那里原位置就是4,所以应该修改和优化出现两次的判断条件
3.若x%n==0&&x/n==3,否则x%n!=0&&x/n==2
vector<int> findDuplicates(vector<int>& nums) {
vector<int> ret;
int n = nums.size();
for(int i = 0; i < n; i++) {
int x = (nums[i] - 1) % n; //找对应的位置
nums[x] += n; //对应的元素的值+=n
}
//这里需要多判断一次,因为n会有临界情况
//当该位置取余等于0的时候,那么说明原来这个地方位置是n,所以x/n==3,才算是出现两次,不单单会在最后一个位置出现
for(int i = 0; i < n; i++) {
if(nums[i] % n != 0 && nums[i] / n == 2) { //这个条件和下面的一个判断条件也说不好,那个放在前面好些
ret.push_back(i + 1);
} else if(nums[i] % n == 0 && nums[i] / n == 3) {
ret.push_back(i + 1);
}
}
return ret;
}
int main() {
vector<int> nums = { 4, 3, 2, 7, 8, 2, 3, 1 };
vector<int> ret = findDuplicates(nums);
for(int i = 0; i < ret.size(); i++) {
cout << ret[i] << " ";
}
return 0;
}