题目描述
一个整型数组 n u m s nums nums 里除两个数字之外,其他数字都出现了两次。请写程序找出这两个只出现一次的数字。要求时间复杂度是 O ( n ) O(n) O(n),空间复杂度是 O ( 1 ) O(1) O(1)。
示例 2:
输入:nums = [1,2,10,4,1,4,3,3]
输出:[2,10] 或 [10,2]
解法:分组异或
如果除了一个数字以外,其他数字都出现了两次,对全员进行异或操作,则成对出现的数字的所有位会两两抵消为
0
0
0,最终得到的结果就是那个出现了一次的数字。
那么这一方法如何扩展到找出两个出现一次的数字呢?
如果可以把所有数字分成两组,使得:
- 两个只出现一次的数字在不同的组中;
- 相同的数字会被分到相同的组中。
那么对两个组分别进行异或操作,即可得到答案的两个数字。这是解决这个问题的关键。
记这两个只出现了一次的数字为
a
a
a 和
b
b
b,那么所有数字异或的结果就等于
a
a
a 和
b
b
b 异或的结果,记为
x
x
x。如果把
x
x
x写成二进制的形式
x
k
x
k
−
1
⋯
x
2
x
1
x
0
x_k x_{k - 1} \cdots x_2 x_1 x_0
xkxk−1⋯x2x1x0,其中
x
i
∈
{
0
,
1
}
x_i \in \{ 0, 1 \}
xi∈{0,1},则
x
i
=
1
x_i=1
xi=1 表示
a
i
a_i
ai 和
b
i
b_i
bi 不等,
x
i
=
0
x_i=0
xi=0 表示
a
i
a_i
ai 和
b
i
b_i
bi 相等。假如任选一个不为
0
0
0 的
x
i
x_i
xi,按照第
i
i
i 位给原来的序列分组,如果该位为
0
0
0 就分到第一组,否则就分到第二组,这样就能满足以上两个条件,因为:
- 首先,两个相同的数字的对应位都是相同的,所以一个被分到了某一组,另一个必然被分到这一组,所以满足了条件 2 2 2。
- 这个方法在 x i = 1 x_i = 1 xi=1 的时候 a a a 和 b b b 不被分在同一组,因为 x i = 1 x_i = 1 xi=1 表示 a i a_i ai 和 b i b_i bi 不等,根据这个方法的定义「如果该位为 0 0 0 就分到第一组,否则就分到第二组」可以知道它们被分进了两组,所以满足了条件 1 1 1。
在实际操作的过程中,在得到序列的异或和 x x x 之后,对于这个「位」是可以任取的,只要它满足 x i = 1 x_i = 1 xi=1。但是为了方便,这里的代码选取的是「不为 0 0 0 的最低位」,当然也可以选择其他不为 0 0 0 的位置。
算法流程:
- 先对所有数字进行一次异或,得到两个出现一次的数字的异或值。
- 在异或结果中找到任意为 1 1 1 的位。
- 根据这一位对所有的数字进行分组。
- 在每个组内进行异或操作,得到两个数字。
复杂度分析
- 时间复杂度: O ( n ) O(n) O(n),只需要遍历数组两次。
- 空间复杂度: O ( 1 ) O(1) O(1),只需要常数的空间存放若干变量。
C++代码实现
class Solution {
public:
vector<int> singleNumbers(vector<int>& nums) {
int ret = 0;
for (int n : nums)
ret ^= n;
int div = 1;
while ((div & ret) == 0)
div <<= 1;
int a = 0, b = 0;
for (int n : nums)
if (div & n)
a ^= n;
else
b ^= n;
return vector<int>{a, b};
}
};