剑指 Offer 56 - I. 数组中数字出现的次数

题目介绍

一个整型数组 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.length <= 10000

解答

class Solution {
public:
    vector<int> singleNumbers(vector<int>& nums) {
        // 思路：利用位运算找出两个只出现一次的数字（x和y）分到两个数组中
        int x = 0, y = 0, n = 0, m = 1;

        // 1.遍历 nums 执行异或得 x^y = n
        for(int num : nums)
        {
            n ^= num;
        }

        // 2. 循环左移计算m，找到x和y二进制位第一次不同的位置
        while((n & m) == 0)
        {
            m <<= 1;
        }

        // 3.拆分nums为两个子数组
        // 4.分别遍历两个子数组执行异或
        for(int num : nums)
        {
            if(num & m) x ^= num;
            else y ^= num;
        }

        return vector<int> {x, y};
    }
};