leetcode面试题 17.19. 消失的两个数字

题目

给定一个数组，包含从 1 到 N 所有的整数，但其中缺了两个数字。你能在 O(N) 时间内只用 O(1) 的空间找到它们吗？

以任意顺序返回这两个数字均可。

示例 1:

输入: [1]
输出: [2,3]
示例 2:

输入: [2,3]
输出: [1,4]
提示：

nums.length <= 30000

来源：力扣（LeetCode）

思路分析

  如果不考虑 O(1) 的空间复杂度，可以利用 set 先存好 1 到 N 的整数，然后遍历数组，如果该值包含在 set 中，就从 set 中移出该值，最后 set 剩余的数就是缺失的数字。代码放在附录中，仅供参考。
  如果考虑到 O(1) 的空间复杂度，该题就比较复杂了。首先，可以考虑根据已知条件可以得到什么，我们知道所有数为 1 到 N 的整数，然后已知了 nums 数组，那么显然可以求得这两个缺失数的和。据此·，可以得到 mid = towSum / 2， 此时 mid 的值一定是大于等于一个数，小于等于另一个数。即在 （1, mid）中存在这个缺失的数，此时就可以很容易求出这个缺失的数。自然的，另一个数也可以求得。

代码

class Solution {
    public int[] missingTwo(int[] nums) {
        int N = nums.length + 2;
        int sum = 0;
        for (int i : nums) {
            sum += i;
        }
        // 求两数之和
        int twoSum = N * (N + 1) / 2 - sum;
        int mid = twoSum / 2;
        int beforeSum = 0;
        // 求出比 mid 小的数
        for (int i : nums) {
            if (i <= mid) {
                beforeSum += i;
            }
        }
        int one = mid * (mid + 1) / 2 - beforeSum;
        int two = twoSum - one;
        return new int[]{one, two};
    }
}

附

class Solution {
    public int[] missingTwo(int[] nums) {
        int N = nums.length + 2;
        Set<Integer> set = new HashSet<>();
        for (int i = 1; i <= N; i++) {
            set.add(i);
        }
        for (int i = 0; i < nums.length; i++) {
            if (set.contains(nums[i])) {
                set.remove(nums[i]);
            }
        }
        List<Integer> ans = new ArrayList<>();
        for (int i : set) {
            ans.add(i);
        }
        int[] res = new int[ans.size()];
        for (int i = 0; i < ans.size(); i++) {
            res[i] = ans.get(i);
            // System.out.println(res[i]);
        }
        return res;
        // return ans.toArray()
    }
}