面试题56 - I. 数组中数字出现的次数
2020-04-01 本文已影响0人
人一己千
题目
一个整型数组 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 <= 10000
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/shu-zu-zhong-shu-zi-chu-xian-de-ci-shu-lcof
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路
首先看个简单版,数组中只出现一次的数字,其他都出现了两次。
直接把所有的数字异或起来就得到答案了,原因是哪些出现两次的数字和自己异或之后就变成了0.
这一题是升级版,出现了两个数字,全部异或下来的话,等于两个数的异或。
那么我们可以想办法把这两个数字分成两组,一组是num1和一些一对对的数字,另一组是num2和一对对的数字。怎么分开的话,用全部的数字异或的结果,得到的结果,找到里面包含1的位数,和每个数相与的话,就能进行分组。(因为这是两个不同的数异或的结果,异或就是在位上不同则为1!)
代码
class Solution:
def singleNumbers(self, nums: List[int]) -> List[int]:
result = 0
for i in nums:
result ^= i
h = 1
while result&h == 0:
h <<= 1
num1, num2 = 0,0
for num in nums:
if num&h == 0:
num1 ^= num
else:
num2 ^= num
return [num1,num2]