题解 非递归实现组合型枚举

Posted 2020-11-26 randolph68706

题目链接

用栈模拟dfs即可

#include<cstdio>
#include<stack>
using namespace std;
struct state {
    int pos,num,a;//第pos位，当前已有num个数字，用二进制状态压缩进a（省去数组存储）
};
stack<state> s;
int n,m;
int main() {
    scanf("%d%d",&n,&m);
    s.push((state) {
        0,0,0
    });
    while(!s.empty()) {
        state a=s.top();
        s.pop();
        if (a.num+n-a.pos<m) continue;
        if (a.num==m) {
            for (int i=0; i<n; i++)
                if ((a.a>>i)&1) printf("%d ",i+1);
            puts("");
            continue;
        }
        if (a.pos<n) {
            s.push((state) {
                a.pos+1,a.num,a.a
            });
            s.push((state) {
                a.pos+1,a.num+1,a.a|(1<<a.pos)
            });//题目是从小到大输出，所以小的要放在后面，先从栈中取出。这与dfs不太一样
        }
    }
}