LeetCode 260. 只出现一次的数字 III:异或

【LetMeFly】260.只出现一次的数字 III

力扣题目链接:https://leetcode.cn/problems/single-number-iii/

给你一个整数数组 nums,其中恰好有两个元素只出现一次,其余所有元素均出现两次。 找出只出现一次的那两个元素。你可以按 任意顺序 返回答案。

你必须设计并实现线性时间复杂度的算法且仅使用常量额外空间来解决此问题。

 

示例 1:

输入:nums = [1,2,1,3,2,5]
输出:[3,5]
解释:[5, 3] 也是有效的答案。

示例 2:

输入:nums = [-1,0]
输出:[-1,0]

示例 3:

输入:nums = [0,1]
输出:[1,0]

 

提示:

  • 2 <= nums.length <= 3 * 104
  • -231 <= nums[i] <= 231 - 1
  • 除两个只出现一次的整数外,nums 中的其他数字都出现两次

方法一:位运算(异或)

这道题的本质思路是:将所有的数分成两组,只出现了一次的数分别分到两组中,其余数根据“与单独的数的相似程度”分到这两个组中。这个过程保证了相等的两个数会被分到同一组中。

依据什么将只出现了一次的两个数分到两组中呢?我们只需要将所有的数异或,异或的结果就是“只出现一次的两个数”的异或结果。这两个数不相等,因此这个异或结果一定不为零。

异或结果中,为0的位代表两数这一位也相等,为1的位代表两数的这一位不同。那么,我们就可以根据这个异或结果的“最低一个不为0的位”为依据,将所有的数分为两组。这样,不相同的两个数一定会被分到不同的组中。

这样,对于单个组,只有一个只出现了一次的数字 和 出现了两次的数字,按照136.只出现一次的数字的方法分别提取出这两个数了。

关于如何求得一个数二进制下第一个不为0的位,可以依据lowbit的原理。

  • 时间复杂度 O ( l e n ( n u m s ) ) O(len(nums)) O(len(nums))
  • 空间复杂度 O ( 1 ) O(1) O(1)

AC代码

C++
class Solution {
public:
    vector<int> singleNumber(vector<int>& nums) {
        unsigned int temp = 0;
        for (int t : nums) {
            temp ^= t;
        }
        int mask = temp & (-temp);
        vector<int> ans(2);
        for (int t : nums) {
            ans[(t & mask) != 0] ^= t;
        }
        return ans;
    }
};
Python
# from typing import List

class Solution:
    def singleNumber(self, nums: List[int]) -> List[int]:
        temp = 0
        for t in nums:
            temp ^= t
        mask = temp & (-temp)
        ans = [0, 0]
        for t in nums:
            ans[(t & mask) != 0] ^= t
        return ans

同步发文于CSDN,原创不易,转载经作者同意后请附上原文链接哦~
Tisfy:https://letmefly.blog.csdn.net/article/details/133872707

你可能感兴趣的:(题解,#,力扣LeetCode,leetcode,算法,题解,位运算,异或)