前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >一道位运算的算法题

一道位运算的算法题

作者头像
四火
发布2022-07-18 13:58:30
1910
发布2022-07-18 13:58:30
举报
文章被收录于专栏:四火的唠叨

最近遇到这样一道算法题:

Given an array of integers, every element appears three times except for one. Find that single one.

一组整数,除了一个只出现一次以外,其他每个整数都恰好出现三次,要寻找那个特殊的整数。

似曾相识

首先,它让我想起了另外一道类似的题目,如果把上面的“ 恰好三次”,改成“ 恰好两次”,寻找那个特殊的整数,又该怎么解?

那样的话,我希望找到一个方法,让两个相同的数进行运算以后,能够泯灭掉,这样所有的数进行运算,剩下的值就是那个特殊的数。恰好有这样的方法,这个方法就是“ 异或”:

代码语言:javascript
复制
public int singleNumber(int[] A) {
    int total = 0;
    for (int a : A)
        total ^= a;
    return total;
}

通用算法

“ 恰好两次” 恰好有“ 异或” 来解,现在“ 恰好两次” 变成了“ 恰好三次”,推广一点说,如果是“ 恰好 N 次”,该怎么解?

通用的算法中,用一个 HashMap 可以得到复杂度近似为 n 的解法,key 为数字本身,value 计数,到三次的时候 delete 掉这个 entry,循环完成以后整个 HashMap 中剩下的就是那个特殊的整数了。这个解法普普通通,没有叙述的必要。这个方法可以保证“ 恰好 N 次” 一样解决。这个算法很简单,就不写出来了。

另外一个思路,借由位操作,对于整数 32 位,对于每一位,整个数列的数加起来去取 3 的余数,就是那个特殊的数在该位上的值。这个方法也可以保证“ 恰好 N 次” 一样能够被解决:

代码语言:javascript
复制
public int singleNumber(int[] A) {
	int ret = 0;
	for (int i = 0; i < 32; i++) {
		int c = 0, mask = 1 << i;  // ① mask, 第 i 位为 1,其他位都为 0
		for (int j = 0; j < A.length; j++) {
		int val = (A[j] & mask);
		if ( val>0 || val<0 ) {  // ② 如果该数在这一位上为 1,计数器就加一
			c++;
			}
		}
		if (c%3 > 0)  // ③ 这一位的计数除以 3 取余数,在这里只可能为 0 或 1
			ret |= mask;
	}
	return ret;
}

关于补码

但是,我在一开始实现这个算法的时候,在上面代码中②的位置,我漏掉了 val<0 的情况,因为第一印象告诉我,一个正整数去与上一个掩码数,会得到一个正整数。但是这是错误的印象。比如在参数 A 等于 { -1, -1, -2, -1 } 的时候,漏掉 val<0 的结果等于一个荒唐的 2147483646。

这是为什么呢?

因为负数在内存中是以补码方式存放的,第一位最高位是符号位,0 表示正数,1 表示负数,仅当表示负数的时候,余下的 31 位等于那个数的数值每一位都取反,然后加 1。例如-1,这 32 位数是:

代码语言:javascript
复制
// 取反加一前:
1(符号位)000 0000 0000 0000 0000 0000 0000 0001
// 取反加一后:
1(符号位)111 1111 1111 1111 1111 1111 1111 1111

32 位整数的范围是从-2147483648 到 2147483647,为何负值比正值能表示的数多一个,就在于这个“ 加一”(表示 0 的时候符号位是 0,相当于表示 0 的时候占用了正数的表示法)。

所以,如果漏掉了上面代码中 val<0 的情况,在执行到 i=31 的循环的时候,掩码 mask 即 1<<i 是-2147483648,因为它把符号位给变成了 1,后面都是 0:

代码语言:javascript
复制
// 即
1(符号位)000 0000 0000 0000 0000 0000 0000 0000
// 如果按照“ 取反加一” 的规则,它是由它自己取反加一而来的,发生了溢出
1(符号位)000 0000 0000 0000 0000 0000 0000 0000

所以这个数也是补码表示的负数中,最特殊的一个。

那为什么上面说漏掉 val<0 之后算错的结果是 2147483646 呢?

这个实际要求解的数-2 在内存中的表示是这样的:

代码语言:javascript
复制
// 取反加一前:
1(符号位)000 0000 0000 0000 0000 0000 0000 0010
// 取反加一后:
1(符号位)111 1111 1111 1111 1111 1111 1111 1110
// 符号位错误:
0(符号位)111 1111 1111 1111 1111 1111 1111 1110

由于前面说到的,符号位错误了,变成了正数,而正数的表示法可不是补码表示,所以得出了 2147483646 这个数。

借助两个数的每一位存储信息

下面这个方法稍微有点难理解,而且很容易写错。需要两个数(one 和 accumulation),因为一个数在每一位上面无法存放超过两次同样的数出现的信息。每次循环中,需要先标记出现,然后再清零出现过三次的标志位。最终 one 留下的每一位都是无法清零的,即出现次数不是 3 的整数倍的。

代码语言:javascript
复制
public int singleNumber4(int A[])
{
	int one = 0; // 出现一次的标志位
	int accumulation = 0; // 积累标志位
	for (int i = 0; i < A.length; i++)
	{
		accumulation |= A[i] & one; // 只要第二次或者以上出现,就为 1
		one ^= A[i];  // 出现奇数次保留,偶数次抛弃
		int t = one & accumulation;  // 第三次的时候 one 和 accumulation 都保留了该位的值
		one &= ~t;  // 清零出现三次的该位的值
		accumulation &= ~t;
	}
	return one;
}

其实,这道题还有许多其他做法,既包括利用位运算的其他做法,也有那种“ 先排序,然后再寻找特殊数” 这样突破常规想法的解法(其实我觉得先排序这样的做法很好啊,虽然复杂度稍微高一些(取决于排序的时间复杂度了),但是清晰,而且通用性更好。方法虽然简单,但是我大概受到的教条式思维太严重了,这样的方法根本没有想到……)。

==================================================

【2014-11-9】偶然知道还有一种看起来更简单的方式,来自这里

代码语言:javascript
复制
public int singleNumber(int[] A) {
    int ones = 0, twos = 0;
    for(int i = 0; i < A.length; i++){
        ones = (ones ^ A[i]) & ~twos;
        twos = (twos ^ A[i]) & ~ones;
    }
    return ones;
}

欢迎讨论。

文章未经特殊标明皆为本人原创,未经许可不得用于任何商业用途,转载请保持完整性并注明来源链接 《四火的唠叨》

本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档