题目:
剑指 Offer 56 - I. 数组中数字出现的次数。
一个整型数组 nums 里除两个数字之外,其他数字都出现了两次。请写程序找出这两个只出现一次的数字。要求时间复杂度是O(n),空间复杂度是O(1)。
链接:https://leetcode-cn.com/leetbook/read/illustrate-lcof/506jfe/
示例:
输入:nums = [4,1,4,6]
输出:[1,6] 或 [6,1]
输入:nums = [1,2,10,4,1,4,3,3]
输出:[2,10] 或 [10,2]
限制:2 <= nums.length <= 10000
思路:
由于以或操作是相同置0,不同置1。因此先通过异或操作得到仅出现一次的两个数字的异或结果。在异或结果中必然存在至少一个1,利用这条性质,可以找出数组中仅出现一次的两个元素。
具体做法,先用1与两个数字的异或结果做与操作,如果相与的结果为0,则将1不断进行左移,直至找出第一个异或结果中第一个1(从右往左算)的位置,用于分块寻找仅出现一次的数字。之后,遍历数组中的元素,找出与异或结果中第一个1位置相同的元素,在这些元素中进行异或操作,而在其他元素中也进行异或操作,两块区域各能得出一个结果,将两个结果组成列表返回即可。
代码:
class Solution:
def singleNumbers(self, nums: List[int]) -> List[int]:
n = 0
for num in nums:
n ^= num
div = 1
while div & n == 0:
div <<= 1
a, b = 0, 0
for num in nums:
if div & num == 0:
a ^= num
else:
b ^= num
return [a, b]
时间复杂度O(n),空间复杂度O(1).