消失的两个数字

消失的两个数字

给定一个数组，包含从 1 到 N 所有的整数，但其中缺了两个数字。你能在 O(N) 时间内只用 O(1) 的空间找到它们吗？以任意顺序返回这两个数字均可。
输入: [1]
输出: [2,3]
输入: [2,3]
输出: [1,4]

这题其实是位运算中结合了消失的数字以及数组中数字出现的次数(一个整型数组 nums 里除两个数字之外，其他数字都出现了两次。请写程序找出这两个只出现一次的数字)这两道题

• 首先先对 1 到 N 所有的整数进行两次异或操作,那么得到的temp就是缺失的两个数的异或
• 找到这个异或temp的二进制最低位
• 根据这个最低位,将1~N分为两个部分,那么ans[0]和ans[1]分别在其中一部分中
• 根据这个最低位,再将nums数组分为两个部分,那么ans[0]和ans[1]分别是部分中缺失的数
• 将上述两部分对应异或就可以找到每部分中缺失的那个数,即为ans[0]和ans[1]

public static int[] missingTwo(int[] nums) {
        int[] ans = new int[2];
        int temp = 0;
        //首先先对 1 到 N 所有的整数进行两次异或操作,那么得到的temp就是缺失的两个数的异或
        for (int i = 0; i < nums.length; i++) {
            temp = temp ^ nums[i];
        }
        for (int i = 1; i <= nums.length + 2; i++) {
            temp = temp ^ i;
        }
        //找到这个异或temp的二进制最低位
        temp = temp & (-temp);

        //根据这个最低位,将1~N分为两个部分,那么ans[0]和ans[1]分别在其中一部分中
        for (int i = 1; i <= nums.length + 2; i++) {
            if ((i & temp) != 0) {
                ans[0] = ans[0] ^ i;
            } else {
                ans[1] = ans[1] ^ i;
            }
        }

        //根据这个最低位,再将nums数组分为两个部分,那么ans[0]和ans[1]分别是部分中缺失的数
        //将上述两部分对应异或就可以找到每部分中缺失的那个数,即为ans[0]和ans[1]
        for (int num : nums) {
            if ((num & temp) != 0) {
                ans[0] = ans[0] ^ num;
            } else {
                ans[1] = ans[1] ^ num;
            }
        }
        return ans;
    }