LeetCode 剑指 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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/shu-zu-zhong-shu-zi-chu-xian-de-ci-shu-lcof
 

解法：

分组异或。在此之前我做过“一个数组里只有一个数只出现一次，其余出现两次”和“一个数组里只有一个数只出现一次，其余出现三次”。本题可以转换为上面的题。本题有两个数只出现一次，那么我们只需要把这两个数分开异或就好了，那么怎么分开呢？

假设那个只出现一次的两个数为a，b。那么我们对数组的所有元素进行异或，可得到a^b，a^b的结果为a与b二进制中不一样的地方。那么我们可以找到a^b中的一个1，根据这个1的位置，对数组进行分组异或。

class Solution {

public:
    vector<int> singleNumbers(vector<int>& nums) {
        int sum=0;
        for(auto num:nums)
            sum^=num;
        auto getPos=[&](int x)
        {
            for(int i=0;i<nums.size();++i)
            {
                if(sum&1)
                    return i;
                else 
                    sum>>=1;
            }
            return 0;
        };
        int pos=getPos(sum);
        int p=0,q=0;
        for(auto num:nums)
        {
            if(num&1<<pos)
                p^=num;
            else 
                q^=num;
        }
        return {p, q};
    }
};