剑指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)
链接:力扣
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
一解
分组异或。假设数组中除一个数字之外,其他数字都出现了两次,对全部数组进行异或即可得到该数字。
但本题有两个数字不同:
-
全组异或:
a⊕a⊕b⊕b...⊕x⊕y = x⊕y
。 -
随意从 x⊕y 的二进制数字中,找出为 1 的位(异或运算,不同为 1 ,相同为 0),设为 m 。
-
利用 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