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

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

一个整型数组 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

来源:力扣(LeetCode)

链接:力扣

著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

一解

分组异或。假设数组中除一个数字之外,其他数字都出现了两次,对全部数组进行异或即可得到该数字。

但本题有两个数字不同:

  1. 全组异或:a⊕a⊕b⊕b...⊕x⊕y = x⊕y

  2. 随意从 x⊕y 的二进制数字中,找出为 1 的位(异或运算,不同为 1 ,相同为 0),设为 m 。

  3. 利用 m 把数组分为二组,再分别进行全组异或。


class Solution:

    def singleNumbers(self, nums: List[int]) -> List[int]:

        x, y, n, m = 0, 0, 0, 1

        # 全组异或

        for num in nums:

            n ^= num

        # 找出二进制为 1 的那一位

        while n & m == 0:

            m <<= 1

        

        # 利用 m 进行分组,分别进行全组异或

        for num in nums:

            if num & m: x ^= num

            else: y ^= num

        return x, y