题目描述:
给你一个含 n 个整数的数组 nums ,其中 nums[i] 在区间 [1, n] 内。请你找出所有在 [1, n] 范围内但没有出现在 nums 中的数字,并以数组的形式返回结果。
示例1:
输入:nums = [4,3,2,7,8,2,3,1]
输出:[5,6]
示例2:
输入:nums = [1,1]
输出:[2]
提示:
n == nums.length
1 <= n <= 105
1 <= nums[i] <= n
进阶:你能在不使用额外空间且时间复杂度为 O(n) 的情况下解决这个问题吗? 你可以假定返回的数组不算在额外空间内。
代码:
class Solution {
public:
vector<int> findDisappearedNumbers(vector<int>& nums) {
int n = nums.size();
int a[n+1];
memset(a,0,sizeof(a));
vector<int> v;
for(int i=0;i<n;i++){
a[nums[i]]++;
}
for(int i=1;i<=n;i++){
if(a[i]==0){
v.push_back(i);
}
}
return v;
}
};
进阶代码:
class Solution {
public:
vector<int> findDisappearedNumbers(vector<int>& nums) {
int n = nums.size();
vector<int> v;
for(int i=0;i<n;i++){
cout<<nums[i]<<endl;
nums[(nums[i]-1)%n]+=n;
}
for(int i=0;i<n;i++){
if(nums[i]<=n)v.push_back(i+1);
}
return v;
}
};
原地修改。
因为方法一中使用了1个n大小的数组记录元素出现的次数,而nums数组本身恰好大小为n,所以不使用额外空间,直接在nums本身上修改。
遍历nums,对于每个数字,改变其为下标所对应的元素的值,将该值➕n,最后遍历一遍nums,若元素小于等于n,则说明该元素下标在nums数组中没有出现,将该值记录。
注意将元素值x作为下标使用时,要x-1 mod n,-1是因为数组下标本是从0开始,mod n 防止数组下标越界。