448. 找到所有数组中消失的数字

给你一个含 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) 的情况下解决这个问题吗? 你可以假定返回的数组不算在额外空间内。

这道题的原地算法有点意思。
先说使用额外空间。数字范围为[1,n],需要找到[1,n]中未出现的数字。很容易想到用额外数组记录。如果出现x,则x的未知标记为1。最后只需要找出额外数组中=0的数即可。

原地算法1。既然数字范围为[1,n],与 nums的下标相对应,那么如果一个数x存在,我们就另对应nums下标的数值加上一个大数。经过一次遍历后,对nums再一次遍历,如果发现哪个下标小于某个数,则即可找出来缺失的数。考虑到加上大数之后,数组范围越界,故须有%操作。

class Solution {
    public List<Integer> findDisappearedNumbers(int[] nums) {
        int n = nums.length;
        for(int i=0; i<n; i++){
            nums[nums[i]%(n+1)-1] += n+1;
        }
        List<Integer> disappear = new ArrayList<>();
        for(int i=0; i<n; i++){
            if(nums[i]<(n+1)){
                disappear.add(i+1);
            }
        }
        return disappear;
    }
}

原地算法2. 类比原地算法1,可以加,也可以取负。

class Solution {
    public List<Integer> findDisappearedNumbers(int[] nums) {
        for (int i=0; i<nums.length; i++){
            nums[Math.abs(nums[i])-1] = -Math.abs(nums[Math.abs(nums[i])-1]);
        }
        List<Integer> res = new ArrayList<>();
        for(int i=0; i<nums.length; i++){
            if(nums[i]>0){
                res.add(i+1);
            }
        }
        return res;
    }
}
posted @   啤酒加点醋  阅读(50)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 终于写完轮子一部分:tcp代理 了,记录一下
· 震惊!C++程序真的从main开始吗?99%的程序员都答错了
· 别再用vector<bool>了!Google高级工程师:这可能是STL最大的设计失误
· 单元测试从入门到精通
· 【硬核科普】Trae如何「偷看」你的代码?零基础破解AI编程运行原理
点击右上角即可分享
微信分享提示