问题类型

有些问题的所有的可能状态数等于 2^n (或者更多是 2^n - 1 ) 时,我们可以通过枚举遍历所有的可能状态,用二进制位运算来筛选符合我们条件的状态。

以下主要通过Go语言来演示,并附加参考的C++代码。这两种语言都很好的支持整数的二进制位运算与逻辑运算。

二进制:是计算技术中广泛采用的一种数制。二进制数据是用0和1两个数码来表示的数。它的基数为2,进位规则是“逢二进一”,借位规则是“借一当二”。

子集:是一个数学概念:如果集合A的任意一个元素都是集合B的元素,那么集合A称为集合B的子集。

  • 含有0个元素的子集有 C_n^0 个,
  • 含有1个元素的子集有 C_n^1 个,
  • 含有2个元素的子集有 C_n^2 个,
  • .........
  • 含有N个元素的子集有 C_n^n 个

由二项式系数的性质可得: C_n^0 + c_n^1 + C_n^2 + \dots + C_n^n = 2^n

二进制左移位运算: 1<<n = 2^n

1<<0=1(0);
1<<1=2(10);
1<<2=4(100);
1<<3=8(1000);
1<<4=16(10000);
...
1<<7=128(10000000);
...

整体框架

Golang

C++

具体含义(Go)

第一层循环:如果该问题的通过排列组合分析所有枚举情况有 2^n 种(如果不存在全0的情况,即 2^n - 1 种时,i 初始化为 1),则是这样的最外层循环枚举:

i 不仅实现了所有可能的循环次数,它的二进制表示还枚举了所有可能的状态。

第二层循环:当进入某个具体的状态时(进入第一层循环后),我们需要通过遍历每一位整数的二进制做逻辑与运算来搞清楚此状态具体情况。

假设有5个字母:A,B,C,D,E. 要从中至少选出一个,请问有哪些种选法?

举出其中一个可能状态,并用二进制表示:

用二进制表示一状态

此时的外循环变量 i = 1\times 2^4 + 1\times 2^3 + 1\times 2^0 = 25 此时列举 j = 0 的情况,1<<j 是在第 j 位为1其他均为0的二进制

二进制逻辑与运算

如果 i & (1<<j) >0 就说明循环变量 i 的二进制第 j 位数为1,其含义是这个选择状态下字母 E 被选中。 同样道理遍历整个循环变量 i 的二进制每一位,来搞清楚它具体指什么情况。即 i & (1<j) 表达式的值仅由我们想要知道的 i 二进制下的第 j 位决定。 要注意Go语言整数型int不可直接转为布尔值,所以 i & (1<<j) >0 来表达。

具体案例分析

和为T(二进制枚举模板)

问题描述

从一个大小为nums的数组中选取一些元素,使得它们的和等于给定的值target。每个元素限选一次,不能一个都不选。

样例输出

样例输出

思路

这个问题很明显所有可能为 2^n - 1 个,因为不能一个都不拿。所以就直接用 i 的二进制还模拟所有可能的选择情况。 当选项中有满足合为 target 的情况时输出并计数。

Golang

C++ 仅供参考解决思路