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

地址：

力扣https://leetcode-cn.com/problems/find-all-numbers-disappeared-in-an-array/

题目：

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/find-all-numbers-disappeared-in-an-array
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路：

题目可以得知，如果没有缺失数字，那么数组元素值 -1 正好与数组下标逐一对应

暴力求法当然可以，不过考虑 n 值很大的情况不可取

联想到 钥匙与盒子 的故事，一把钥匙开一个盒子

战争情况为了保证任意一个钥匙都能打开任意盒子，放钥匙就错开放

譬如：钥匙 1 可以打开盒子1，盒子 1 里面放的可能是 钥匙 3，拿到钥匙 3 就可以开盒子 3

           盒子3 放钥匙8，然后就开盒子8，如此，就能开所有盒子

归纳就是 钥匙与盒子 匹配，如果有缺失的钥匙，那么自然就没有对应打开的盒子

所以，我们可以用这样的思路来破解这道题。有对应开的钥匙，盒子标注可以被打开

考虑题意不分配额外空间，那么我们就只能将盒子的值变为 负数或者 [1,n] 以外的值表示有对应钥匙。最后再遍历看看那些盒子没有被标注过，也就对应缺失的钥匙，即缺失的数字

示例：

第一次：从 idx =0 开始

 

val = 4, val - 1 = 3 （减 1 是为了对应数组下标。因为 1<= val <=n ）

找到 [3] 的盒子，然后将其标注为有对应 钥匙 可以打开

第二次：idx = 1

val = 3， val - 1 = 2

找到 [2] 的盒子，然后将其标注为有对应 钥匙 可以打开

 

同样的道理，我们可以标注所有的盒子

 

最后只有 [4][5] 盒子没有对应钥匙，那么自然缺失数字就是 5, 6 

方法一、钥匙与盒子

/**
 * Note: The returned array must be malloced, assume caller calls free().
 */
int* findDisappearedNumbers(int* nums, int numsSize, int* returnSize){
    int i, j=0;
    int key;
    int cnt=0;
    
    for(i=0; i<numsSize; i++)
    {
        //printf("num[%d]=[%d]\n", i, nums[i]);
        if(nums[i] < 0)         // other box has key to open it
            key = (0 - nums[i]) - 1;
        else
            key = nums[i] - 1;

        if(nums[key] > 0)
            nums[key] = 0 - nums[key]; // mark box has key to open

        //printf("key=%d..num[%d]=[%d]\n", key, key, nums[key]);
    }
    
    for(i=0; i<numsSize; i++)
    {
        if(nums[i] > 0)
            cnt++;
    }

    int *ret = (int *)malloc(sizeof(int) * cnt);
    
    for(i=0,j=0; i<numsSize; i++)
    {
        if(nums[i] > 0)
            ret[j++] = i + 1;
    }

    *returnSize = cnt;
    return ret;
}

查看更多刷题笔记