剑指offer

面试题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]
        
上一篇下一篇

猜你喜欢

热点阅读