携手创作,共同成长!这是我参与「掘金日新计划 · 8 月更文挑战」的第11天,点击查看活动详情
一、题目描述:
给你一个含 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 <= 10^5
- 1 <= nums[i] <= n
进阶:你能在不使用额外空间且时间复杂度为 O(n) 的情况下解决这个问题吗? 你可以假定返回的数组不算在额外空间内。
二、思路分析:
目标是将每个存在的元素i放到i-1下标处,并让不存在的元素下标(i-1)变为0;
- 倘若元素不在正确的位置(当前元素与下标不匹配即nums[i-1]!=i)则将当前元素与元素目标位置的元素交换。
- 如果目标位置元素已经是正确元素则将当前位置元素设为0;
- 如果碰到0和元素正确的位置直接跳过。
最后遍历数组,将值为0的下标加一后填入容器内。
例:
[4,3,2,7,8,2,3,1]-> (位置0元素为4,位置3(4-1)元素为7。交换4,7)
[7,3,2,4,8,2,3,1]-> (位置0元素为7,位置6元素为3,交换7,3)
[3,3,2,4,8,2,7,1]-> (位置0元素为3,位置2元素为2,交换3,2)
[2,3,3,4,8,2,7,1]-> (位置0元素为2,位置1元素为3,交换2,3)
[3,2,3,4,8,2,7,1]-> (位置0元素为3,位置3元素为3,将当前位置设为0,指针移到下一个位置)
[0,2,3,4,8,2,7,1]-> (位置1元素为2,正确向下搜索。) .
.
.
[1,2,3,4,0,0,7,8]
三、AC 代码:
class Solution {
public:
vector<int> findDisappearedNumbers(vector<int>& nums) {
vector<int> ans;
for(int i = 0;i<nums.size();++i)
{
if(nums[i] == 0 || nums[i] == i+1){
continue;
}
else if(nums[nums[i]-1] != nums[i])
{
int temp = nums[i];
nums[i] = nums[nums[i]-1];
nums[temp-1] = temp;
--i;
}
else
{
nums[i] = 0;
}
}
for(int i=0;i<nums.size();++i)
{
if(0==nums[i])
{
ans.push_back(i+1);
}
}
return ans;
}
};