题目:
一个整型数组 nums 里除两个数字之外,其他数字都出现了两次。请写程序找出这两个只出现一次的数字。要求时间复杂度是O(n),空间复杂度是O(1)。
示例 1:
输入:nums = [4,1,4,6]
输出:[1,6] 或 [6,1]
思考:
主要考察位操作,如之前没有遇到过或写过比较难想出来,之前做过简单版本,即只有一个数字出现一次,求这个数字。
直观想法:采用dict,遍历一遍nums,key为nums中的num,value为count即可。之后遍历dict,将count的数值为1的num即key输出即可。时间复杂度为O(n),空间复杂度是O(n),不符合要求。
解答:
1)异或查找不同,即简单版本:只有一个数字出现一次,求这个数字。原理在与,任何数与自己异或都为0,0异或任何数都为任何数。因此,而已一直异或,最终的输出即为只出现一次的数字。
def singleNumbers(self, nums: List[int]) -> List[int]:
res =0
for num in nums:
res = res^num#此时res有两位为1
return res
2)有两个不同的数字时,同样先一直异或。最后异或的结果必然在某一位上为1(异或的本质就是找不同),这一位就是两个只出现一次数字的不同位。因此,a)先针对异或的结果寻找出为1的位(即两个数字不同的位)。b)将nums中的num与为1的位进行&操作,将数字分为两堆。在各堆分别进行异或操作,最终得出两位不同的数。
def singleNumbers(self, nums: List[int]) -> List[int]:
res =0
res1 =0
for num in nums:
res = res^num#此时res有两位为1
diff =1
while (res & diff) ==0:#寻找不同的位,如在第2位不同,则diff为1(对应二进制10)
diff <<=1
res =0
for num in nums:
if diff & num:#分别在第二位进行&操作,分为第二位为0的堆与第二位为1的堆。
res = res^num
else:
res1 = res1^num
return [res, res1]