资质|算术能力5 |问题2

介绍

本题需要具备算术能力5级的程序员进行解答。题目描述如下：

有一个整数数组，其中有两个数字只出现了一次，其他数字均出现了两次。请找出这两个只出现了一次的数字。

例如，输入数组为[1, 2, 3, 4, 1, 2]，输出为[3, 4]。

解题思路

可以使用异或操作进行解答。首先将所有数字进行异或，得到的结果就是两个只出现了一次的数字的异或结果。然后找到异或结果中二进制表示中的一个不为0的位数，以此为标志将数组中的数字分为两个子数组。分别对两个子数组进行异或操作，得到的结果就是两个只出现了一次的数字。

代码实现

def find_two_single_numbers(nums: List[int]) -> List[int]:
    # 获取两个数字的异或结果
    xor = 0
    for num in nums:
        xor ^= num

    # 找到异或结果的一个不为0的位数
    index = 0
    while xor & 1 == 0:
        xor >>= 1
        index += 1

    # 将数组中的数字按照index位置是否为1分为两个子数组
    a, b = 0, 0
    mask = 1 << index
    for num in nums:
        if num & mask:
            a ^= num
        else:
            b ^= num

    return [a, b]

返回的代码片段按markdown标明。