leetcode 448.找到所有数组中消失的数字
题干
给定一个范围在 1 ≤ a[i] ≤ n ( n = 数组大小 ) 的 整型数组,数组中的元素一些出现了两次,另一些只出现一次。
找到所有在 [1, n] 范围之间没有出现在数组中的数字。
您能在不使用额外空间且时间复杂度为O(n)的情况下完成这个任务吗? 你可以假定返回的数组不算在额外空间内。
示例:
输入:
[4,3,2,7,8,2,3,1]
输出:
[5,6]
题解
大年初二睡大觉,很容易想到哈希表统计数字频率,但显然效率不太行。
class Solution {
public:
vector<int> findDisappearedNumbers(vector<int>& nums) {
int n = nums.size();
unordered_map<int,int> freq;
vector<int> ans;
for(auto i : nums){
freq[i]++;
}
for(int i = 1 ; i <= n ; ++i){
if(freq[i] < 1){
ans.push_back(i);
}
}
return ans;
}
};
我们可以扫描数组,当数组中间出现不连续的部分时,譬如[…,2,5,…],我们就可以将中间缺失的数字加入答案,即3,4。
可惜开头和结尾需要特判,因为有可能出现虽然数组连续但是从大于1的数开始或小于n的数结尾的情况。
class Solution {
public:
vector<int> findDisappearedNumbers(vector<int>& nums) {
int n = nums.size();
if(n == 0){
return {};
}
vector<int> ans;
sort(nums.begin(),nums.end());
for(int i = 1 ; i < n ; ++i){
if(nums[i] != nums[i-1] && nums[i] != nums[i-1] + 1){
for(int j = nums[i-1] + 1 ; j < nums[i] ; ++j){
ans.push_back(j);
}
}
}
for(int i = nums[n-1] + 1 ; i <= n ; ++i){
ans.push_back(i);
}
for(int i = 1 ; i < nums[0] ; ++i){
ans.push_back(i);
}
return ans;
}
};
由于数组里的所有数都落在[1,n]范围之内,所以我们可以利用原数组+哈希的思想来标记数字是否出现过。
遍历数组,对于遍历到的数字i,为了标记i已经出现过,我们把nums[i]加上一个大于等于n的数m。遍历完成后再遍历一次nums,如果其中元素<=m就说明这个该元素没有出现过,加入答案即可。
需要注意的是,由于我们用+m来标记数字的出现,所以为了排除标记的干扰获取nums中原有的数字(因为nums[i] >= i不总成立),我们每次进行标记时,取出的元素要对m取模。
class Solution {
public:
vector<int> findDisappearedNumbers(vector<int>& nums) {
int n = nums.size();
if(n == 0){
return {};
}
for(auto i : nums){
nums[(i - 1) % n] += n;
}
vector<int> ans;
for(int i = 0 ; i < n ; ++i){
if(nums[i] <= n){
ans.push_back(i+1);
}
}
return ans;
}
};