《Go语言圣经（中文版）》第二章 PopCount练习题

最新推荐文章于 2026-07-19 16:46:26 发布

原创

最新推荐文章于 2026-07-19 16:46:26 发布 · 716 阅读

收录于

当前文章被以下社区和专栏收录：

本文探讨了在Go语言中实现人口计数（即计算一个整数中1的位数）的不同方法，包括直接表达式计算、循环计算、位移算法及清零最低位算法。通过对比这些方法的性能，深入理解位操作的高效性和灵活性。

package popcount

// pc[i] is the population count of i.
var pc [256]byte

func init() {
	for i := range pc {
		pc[i] = pc[i/2] + byte(i&1)
	}
}

// PopCount returns the population count (number of set bits) of x.
func PopCount(x uint64) int {
	return int(pc[byte(x>>(0*8))] +
		pc[byte(x>>(1*8))] +
		pc[byte(x>>(2*8))] +
		pc[byte(x>>(3*8))] +
		pc[byte(x>>(4*8))] +
		pc[byte(x>>(5*8))] +
		pc[byte(x>>(6*8))] +
		pc[byte(x>>(7*8))])
}

// practice 2.3
// 重写PopCount函数，用一个循环代替单一的表达式。比较两个版本的性能。 （11.4节将展示如何系统地比较两个不同实现的性能。）
func PopCountWithForLoop(x uint64) int {
	var counts int
	for i := 0; i < 8; i++ {
		counts += int(pc[x>>(i*8)])
	}
	return counts
}

// practice 2.4
// 用移位算法重写PopCount函数，每次测试最右边的1bit，然后统计总数。
// 比较和查表算法的性能差异。
func PopCountWithBitMove(x uint64) int {
	var counts int
	for i := 0; i < 64; i++ {
		// & 运算，&1，只有最右边第1位是1时&1结果才为1
		if x&1 != 0 {
			counts++
		}
		//右移一位
		x = x >> 1