找到所有数组中消失的数字
题目
给一个含 n 个整数的数组 nums ,其中 nums[i] 在区间 [1, n] 内。找出所有在 [1, n] 范围内但没有出现在 nums 中的数字,并以数组的形式返回结果。
示例
输入:nums = [4,3,2,7,8,2,3,1]
输出:[5,6]
题解
由题知,数组元素取值范围在 [1, n] 内,可以利用这一特点原地修改数组,将这一范围之外的数字作为是否存在的标志。
因此,遍历数组 nums 时,每遇到一个数 x ,将 nums[x-1]+=n ,遍历结束后所有数(不重复的数)均大于 n ,若存在 nums[i]<=n ,则说明 nums[i] 为重复元素,且 i+1 未在数组中出现。
注意,遍历数组时,其中的数可能已经 +n ,需要先做取模操作。
代码
class Solution {
public:
vector<int> findDisappearedNumbers(vector<int>& nums) {
int n=nums.size();
for(auto& num:nums){
nums[(num-1)%n]+=n;
}
vector<int> ret;
for(int i=0;i<n;i++){
if(nums[i]<=n){
ret.push_back(i+1);
}
}
return ret;
}
};