leetcode [260] 只出现一次的数字 III

---

Contact me:

Blog -> https://cugtyt.github.io/blog/index
Email -> cugtyt@qq.com
GitHub -> Cugtyt@GitHub

---

给定一个整数数组 nums，其中恰好有两个元素只出现一次，其余所有元素均出现两次。 找出只出现一次的那两个元素。

示例 :

输入: [1,2,1,3,2,5]
输出: [3,5]

注意：

结果输出的顺序并不重要，对于上面的例子， [5, 3] 也是正确答案。 你的算法应该具有线性时间复杂度。你能否仅使用常数空间复杂度来实现？

思路：所有数字异或，结果表示出现一次的两个数字异或的结果。找到这个结果第一个非0位置，把所有数字这一位为0的分一组，为1的分一组，那么两个数字就分别存在这两个组中。

class Solution:
    def singleNumber(self, nums: List[int]) -> List[int]:
        two_xor = 0
        for n in nums:
            two_xor ^= n

        pos = 0
        while (two_xor >> pos) & 1 == 0:
            pos += 1

        nums_a = [n for n in nums if n & (1 << pos) != 0]
        nums_b = [n for n in nums if n & (1 << pos) == 0]

        na, nb = 0, 0
        for n in nums_a:
            na ^= n
    
        for n in nums_b:
            nb ^= n

        return [na, nb]