来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/find-all-duplicates-in-an-array
题目描述
给你一个长度为 n 的整数数组 nums ,其中 nums 的所有整数都在范围 [1, n] 内,且每个整数出现 一次 或 两次 。请你找出所有出现 两次 的整数,并以数组形式返回。
你必须设计并实现一个时间复杂度为 O(n) 且仅使用常量额外空间的算法解决此问题。
示例 1:
输入:nums = [4,3,2,7,8,2,3,1]
输出:[2,3]
示例 2:
输入:nums = [1,1,2]
输出:[1]
示例 3:
输入:nums = [1]
输出:[]
提示:
n == nums.length
1 <= n <= 105
1 <= nums[i] <= n
nums 中的每个元素出现 一次 或 两次
解题思路
如果没有空间复杂度的需求,那么计数统计是很好的办法,但是题目要求空间复杂度为O(1),那么不使用额外的空间去完成这道题,就需要在原数组中做文章了。由于数组的数据范围是1-n,如果不重复的话,那么每一个数字k将可以放入对应的k-1的位置,如果k-1位置上的不是k,那么可以判断这个数字重复。所以第一步,将数字k尽可能的放入对应的k-1位置上,第二步遍历数组寻找k-1位置上不是k的数字,加入结果。
代码展示
class Solution { public: vector<int> findDuplicates(vector<int>& nums) { vector<int> viRet; for(int i = 0; i < nums.size(); i++) { while(nums[nums[i] - 1] != nums[i]) swap(nums[nums[i] - 1], nums[i]); } for(int i = 0; i < nums.size(); i++) { if(nums[i] - 1 != i) viRet.push_back(nums[i]); } return viRet; } };
运行结果