在网上看到这么一道题:Given an array of integers, every element appears three times except for one. Find that single one.
这道题可以有多种的解法,比如遍历设标志位来统计,比如用hashmap的key存值,value存出现的次数。但是看到一种很好的算法,使用位运算来解决。
我们先来解决一个简单的问题,把题目中的three times改成two times 怎么解决呢。首先可以有这样的想法,如果有方法能让相同的两个数相遇并“解决掉对方”,这样剩下的一个数就是唯一的那个needed one。正如你所想,这样的一个方法不就是异或么。于是对于two times问题 可以用下面的方法:
package TestOX;
public class TestOXR {
public int twotimes(int[] A) {
int total = 0;
for (int a : A){
total ^= a;
}
return total;
}
public static void main(String arg[]){
<span style="white-space:pre"> </span>int[] testarray = {1,2,1,4,7,3,4,2,3};
System.out.println(new TestOXR().twotimes(testarray));
}
}
最后的输出就是:7。
那么涉及到three times问题是否也可以参考two times的呢?这里我们需要明白一个问题,所谓的异或其实和我们知道的模2加是一样的。也就是mod(a+b,2) == XOR(a,b)的。那么对于three times的问题我们也可以针对每一位进行模运算,像前面想的方法一样,让这三个相同数的相同的位相遇并“解决掉对方”。那么对于three times问题就不是模2加了。所以如下代码所示:
public int threetimes(int[] A) {
int result = 0;
for (int i = 0; i < 32; i++) {
int c = 0, mask = 1 << i; // mask, 第i位为1,其他位都为0。用来指示在int的哪一位进行取余数运算。
for (int j = 0; j < A.length; j++) {
int val = (A[j] & mask);
if ( val>0 || val<0 ) { // 如果该数在这一位上为1,计数器就加一。表示那个唯一数的那一位为1。
c++;
}
}
if (c%3 > 0) // 这一位的计数除以3取余数,在这里只可能为0或1
result |= mask;
}
return result;
}
public static void main(String arg[]){
<span style="white-space:pre"> </span>
<span style="white-space:pre"> </span>int[] testthree = {1,5,2,3,2,1,5,3,6,5,3,2,1};
<span style="white-space:pre"> </span>System.out.println(new testextend().threetimes(testthree));
<span style="white-space:pre"> </span>
<span style="white-space:pre"> </span>}
最后输出:6。
此算法的时间复杂度为O(n) n为待求数组的大小。