落单的数 III

问题描述

http://www.lintcode.com/zh-cn/problem/single-number-iii/

给出2*n + 2个的数字，除其中两个数字之外其他每个数字均出现两次，找到这两个数字。

样例
给出 [1,2,2,3,4,4,5,3]，返回 1和5

笔记

先把数据全都异或一个遍，出现两次的数全都被抵消了。剩下不同的两个数a和b的异或。因为a与b不同，a^b肯定不为0。

找出a^b的最低位的1，这个1的出现是因为a的这一位和b的这一位不同。（异或的性质）。于是可以根据这一位是否是1把整个数组一分为二。再把两个数组分别异或个遍，就可以得到这两个数。

注意：以下写法是错的，有优先级问题，被自己坑了。。

if (i & mod == mod)

正确的写法是

if ((i & mod) == mod)

代码

class Solution {
public:
    /**
     * @param A : An integer array
     * @return : Two integers
     */
    vector<int> singleNumberIII(vector<int> &A) {
        // write your code here
        int axorb = 0;
        for (auto i : A)
        {
            axorb ^= i;
        }
        int mod = axorb & (~(axorb - 1));
        // cout << mod << endl;
        // cout << (mod & 4) << endl;
        // cout << (mod & 12) << endl;
        vector<int> arr1, arr2;
        for (auto i : A)
        {
            if ((i & mod) == mod)
                arr1.push_back(i);
            else
                arr2.push_back(i);
        }
        int a = 0;
        for (auto i : arr1)
            a ^= i;
        int b = 0;
        for (auto i : arr2)
            b ^= i;
        vector<int> res = {a, b};
        return res;
    }
};