0数组困难 LeetCode面试题 17.19. 消失的两个数字

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

描述

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

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

示例 1:

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

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

nums.length <= 30000

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

分析

异或解法：
a ^ b ^ a 的结果是 b
先计算出缺失的两个值的异或结果，因为各个数都不相等，所以结果一定不为0，即某一位是1。然后用这个数把数组分成两类，把1到n+2也分成2类，一直异或最终的结果就是其中一个数。
求和解法：
计算出缺失值的和，然后计算平均值，真实缺失值一定是在平均值的两侧，计算出较小的那个值。
计算思路是在1到平均值之间只有一个数缺失了，计算缺失的和与没有缺失的和，做差就是较小值。

class Solution {
    public int[] missingTwo(int[] nums) {
        int sum = 0;
        int y = 0;
        int n = nums.length;
        for (int num : nums) {
            sum += num;
            y ^= num;
        }
        sum = (n+2) * (n+2+1) / 2 - sum;
        for (int i = 1; i <= n + 2; i++) {
            y ^= i;
        }
        int index = 0;
        for (int i = 0; i < 32; i++) {
            if ((y >> i & 1) == 1) {
                index = i;
                break;
            }
        }
        int x = 0;
        for (int num : nums) {
            if ((num >> index & 1) == 1) {
                x ^= num;
            }
        }
        for (int i = 1; i <= n + 2; i++) {
            if ((i >> index & 1) == 1) {
                x ^= i;
            }
        }
        return new int[]{x,sum - x};
    }
}