抛出问题:假如我们遇到这样一个问题,给你n个元素,需要取出k个元素,问你有哪些取法?
对于有算法基础的同学一定会想到要用回溯,这个方法肯定是可行的,但是当n很大的时候怎么办呢?
进一步优化,我们可能要想到用二进制枚举状态的方式,相信有状压dp基础的同学一眼就能看出。
不过在最近的周赛中我却发现,无脑的枚举所有状态,还是会浪费很多时间去查看一些无效的状态(二进制表示中1的数量不为k),于是就去查找更优的方法。
我们可以只枚举1的数量为k的所有状态,这样不就少遍历了很多次了嘛,具体写法如下【并将我的理解放在了注释中】:
int comb = (1 << num) - 1; //(1).取出字典序最小的1的连续区间
while (comb < (1 << n)) {
int x = comb & -comb; //(2).找到字典序最小的1的位置
int y = comb + x; //(3).将字典序最小的1的连续区间置为0,并将区间左侧第一个0置为1
int z=comb & ~y; //(4).将(1)取出的区间右移,直至区间中1的个数减少一个
comb = (z / x >> 1) | y; //(5).将(4)的结果与(3)的结果取并集
}