448. 找到所有数组中消失的数字(鸽笼原理)
(注:2023秋招笔试中做了好多次这种类型题)
给你一个含 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)
的情况下解决这个问题吗? 你可以假定返回的数组不算在额外空间内。
按照进阶要求,鸽笼原理。
鸽笼原理,即把数组内容总是取负覆盖到内容值对应的下标处,然后每次提取内容用abs(保证原数组内容除了符号不会变化),这样必然有地方的内容值不是负数,这个地方的下标即未出现的数字 。注意内容值和下标隔着一个1
或者打表型,要浪费一个vector空间。
class Solution { public: vector<int> findDisappearedNumbers(vector<int>& nums) { int n=nums.size(); vector<int> ans; for(int i=0;i<nums.size();i++) { nums[abs(nums[i])-1]=-abs(nums[abs(nums[i])-1]); } for(int i=0;i<n;i++) { if(nums[i]>0)ans.push_back(i+1); } return ans; } };