剑指Offer 56-I 数组中数字出现的次数

题目描述

问题分析

这一题最直接的想法是使用哈希表去记录数字出现的次数，再遍历哈希表即可，但这样显然不符合题目对程序空间复杂度为O(1)的要求。
进一步，当考虑问题的一个子问题：再整形数组中只有一个数字出现一次而其他数字全出现两次（偶数次），这一子问题是可以通过将所有的数字依次异或操作即可。通过这一方法我们显然能够获得待求两个数字异或的结果，那么下一步只需要想出如何解出其中一个数字即可。

核心思路

问题的核心思想在于如何将问题约化到上已提到的子问题，将两个结果数字划分为两个不同的组。
现假设待求结果为[a,b]，那么我们通过将nums中所有数字异或可以得到a$\wedge$b（其中$\wedge$表示异或运算），考虑数字a$\wedge$b的二进制表示$x_0$$x_1$$\cdots$$x_m$，其中$x_i$=1表示数字a和b的第i位$a_i$和$b_i$是不同的，$x_i$=0表示数字a和b的第i位$a_i$和$b_i$是相同的。由于a与b不等，那么存在下标$i$使得$a_i$和$b_i$不同，那么可以根据这一下标作为分组依据：1、相同的数字在该位上的表现是相同的（与操作结果相同）；2、a和b在该位上的与操作结果是不同的。那么我们自然选出a$\wedge$b结果中从右至左第一个出现1的位，分类的标志flag表现为在该位上取1，在其他位上取0，通过数组中数与flag按位与操作的结果来分组即可划归到上一子问题。

代码表述

class Solution:
    def singleNumbers(self, nums: List[int]) -> List[int]:
        ret = functools.reduce(lambda x, y: x ^ y, nums)
        #functools.reduce按照二元运算函数按次序将可迭代对象每两个计算，将其降维到一个数
        div = 1
        while div & ret == 0:
            div <<= 1
        a, b = 0, 0
        for n in nums:
            if n & div:
                a ^= n
            else:
                b ^= n
        return [a, b]