[剑指offer]数组中数字出现的子树I

[剑指offer]数组中数字出现的子树I

剑指offer-数组中数字出现的子树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

解题思路

• 相同的数异或为0，不同的数异或为1，0和任何数异或为它本身。
• 所以，数组中所有的数异或的结果 就是 目标两个数异或的结果。由于这两个数不同，所以结果必然不为0。
• 假如数组异或的结果为10010，那么说明这两个数从右往左数的第二位数是不同的。
• 那么可以根据这个数组中所有数的第二位为0或者1将这个数组划分为两个。
• 这样会将目标两个数分散在两个数组中，相同的数必然落在同一数组。
• 这两个数组中的数各自异或，得到的结果就是答案。

实现代码

class Solution {
public:
    vector<int> singleNumbers(vector<int>& nums) {
        int len=nums.size();
        vector<int> res;
        if(len==0)return res;
        int tmp=0;
        for(int i=0;i<len;i++)//异或结果
            tmp^=nums[i];
        int index=0;
        while((tmp&1)==0){//异或结果的第index位为1
            tmp>>=1;
            index++;
        }
        int r1=0,r2=0;
        for(int i=0;i<len;i++){
            if(((nums[i]>>index)&1)==0)//根据第index位，分别异或，获得结果
                r1^=nums[i];
            else
                r2^=nums[i];
        }
        res.push_back(r1);
        res.push_back(r2);
        return res;
    }
};