Offer 驾到,掘友接招!我正在参与2022春招打卡活动,点击查看活动详情。
一、题目描述:
给你一个含 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) 的情况下解决这个问题吗? 你可以假定返回的数组不算在额外空间内。
二、思路分析:
把每个数字nums[i]移动到它对应的下标为nums[i]-1的位置,最终不在数组中的数字对应的位置会被重复数字占据。然后扫描整个数组,找到数组中与其下标不能对应的数字即为结果。
第一步相当于基数排序,复杂度为O(n),第二步遍历数组也是O(n),所以总的时间复杂度为O(n),满足题目要求。
三、AC 代码:
class Solution {
public:
vector<int> findDisappearedNumbers(vector<int>& nums) {
int i=0;
while(i < nums.size()){
if(nums[i] != i+1 && nums[nums[i]-1] != nums[i])
swap(nums[i], nums[nums[i]-1]);
else
++i;
}
vector<int> result;
for(i=0; i<nums.size(); i++){
if(nums[i] != i+1)
result.push_back(i+1);
}
return result;
}
};
范文参考:
ACM金牌题解 | 不额外开辟空间 | 编程熊 - 找到所有数组中消失的数字 - 力扣(LeetCode) (leetcode-cn.com)