442. 数组中重复的数据
题目
给你一个长度为 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 中的每个元素出现 一次 或 两次
答案
class Solution {
public:
vector<int> findDuplicates(vector<int>& nums) {
int n = nums.size();
vector<int> v;
for(int i = 0;i < n;++i){
int index = abs(nums[i]) - 1;
if(nums[index] < 0){
v.push_back(index + 1);
}else{
nums[index] = -nums[index];
}
}
return v;
}
};
算法原理
算法的核心思想是利用数组中的数字范围与数组索引之间的映射关系。由于数组中的每个数字都在 [1, n] 范围内,并且数组的长度也是 n,我们可以将每个数字 nums[i] 映射到数组的索引 nums[i] - 1 上(因为数组索引从 0 开始,而数字从 1 开始)。
算法步骤
- 遍历数组:
遍历数组 nums 中的每个元素 nums[i]。 - 映射到索引:
计算 index = abs(nums[i]) - 1(在这个特定问题中,abs 函数是多余的,因为题目保证数组中没有负数,但保留它可以使代码更健壮,以防未来修改)。 - 检查并标记:
- 检查 nums[index] 的值:
如果 nums[index] 是负数,说明之前已经访问过索引 index,即数字 nums[i](或它的绝对值,如果考虑负数的情况)已经出现过至少一次。由于题目说明每个数字最多出现两次,因此这次出现意味着该数字出现了两次,应将其添加到结果数组中。 - 如果 nums[index] 不是负数(即它是正数或零,但在这个问题中,由于数字范围是 [1, n],它只可能是正数),则将 nums[index] 变为负数,以标记该索引(即该数字)已经被访问过。
- 收集结果:
在遍历过程中,每当发现一个数字第二次出现时(即 nums[index] 为负数时),就将其(以正数形式)添加到结果数组中。 - 返回结果:
遍历完成后,返回包含所有出现两次的数字的结果数组。