leetcode 260. 只出现一次的数字 III【位运算】

给定一个整数数组 nums,其中恰好有两个元素只出现一次,其余所有元素均出现两次。 找出只出现一次的那两个元素。你可以按 任意顺序 返回答案。

进阶:你的算法应该具有线性时间复杂度。你能否仅使用常数空间复杂度来实现?

示例 1:

输入:nums = [1,2,1,3,2,5]
输出:[3,5]
解释:[5, 3] 也是有效的答案。

示例 2:

输入:nums = [-1,0]
输出:[-1,0]

示例 3:

输入:nums = [0,1]
输出:[1,0]

提示:

2 <= nums.length <= 3 * 104
-231 <= nums[i] <= 231 - 1
除两个只出现一次的整数外,nums 中的其他数字都出现两次

Related Topics 位运算 数组

思路: 位运算

两个相同数字的异或运算结果为0,将数组中的数进行异或运算,最终得到a ^ b 的值,而我们知道a ^ b的结果中二进制最高非0位,一定是由一个1和一个0运算所得。只要遍历数组,对a ^ b 的最高位分别针对1和0进行异或运算,就得到a, b的值。

code:

class Solution:
    def singleNumber(self, nums: List[int]) -> List[int]:
        n = len(nums)
        ans = 0
        # 异或预算运算得到a ^ b
        for num in nums:
            ans ^= num
        n = len(bin(ans)) - 3
        a, b = 0, 0
        for num in nums:
    		if num >> n & 1:  # 第 n + 1 位为 1
       			a ^= num
    		else:  # 第 n + 1 位为0
       			b ^= num
		return [a, b]

版权声明:本文为bmicnj原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。