leetcode 数组中数字出现的次数

题目

一个整型数组 nums 里除两个数字之外，其他数字都出现了两次。请写程序找出这两个只出现一次的数字。要求时间复杂度是O(n)，空间复杂度是O(1)。

示例 1：

输入：nums = [4,1,4,6]
输出：[1,6] 或 [6,1]
示例 2：

输入：nums = [1,2,10,4,1,4,3,3]
输出：[2,10] 或 [10,2]
 

限制：

2 <= nums.length <= 10000

题目

看到题目要求空间复杂度为O(1),那就不能使用HashMap的方式，只能采取遍历时进行数组的标记。很容易想到从头到尾数字异或，但是题目有两个相同的数，异或出来是两个数的异或，那怎么样才能分开这两个数字呢。这里我们采用分组异或。如果我们可以把所有数字分成两组，使得：两个只出现一次的数字在不同的组中；**相同的数字会被分到相同的组中。**就可以做到每次异或的得到不同的值。具体怎么做呢？
记这两个只出现了一次的数字为 a 和 b，那么所有数字异或的结果就等于 a 和 b 异或的结果，我们记为x。如果我们把 x 写成二进制的形式x kx k−1⋯x 2 x1x0 ，其中 xi ∈{0,1}，我们考虑一下xi=0 和xi=1 的含义是什么？它意味着如果我们把 a 和 b 写成二进制的形式，ai和 bi的关系——xi=1 表示 ai 和 bi不等，xi=0 表示ai 和 bi相等。假如我们任选一个不为 0的 xi，按照第 i位给原来的序列分组，如果该位为 0 就分到第一组，否则就分到第二组，这样就能满足以上两个条件。

代码

public int[] singleNumbers(int[] nums) {
        int ret = 0;
        for (int n : nums) {
            ret ^= n;
        }
        int div = 1;
        while ((div & ret) == 0) {
            div <<= 1;
        }
        int a = 0, b = 0;
        for (int n : nums) {
            if ((div & n) != 0) {
                a ^= n;
            } else {
                b ^= n;
            }
        }
        return new int[]{a, b};
    }

总结

遇到重复元素时想到位运算的性质，采取位运算来简化空间复杂度，同时巧妙利用位运算的特征。