ACwing 93. 递归实现组合型枚举

目录

题目

思路

代码

---

题目

从 1∼n这 n个整数中随机选出 m个，输出所有可能的选择方案。

输入格式

两个整数 n,m在同一行用空格隔开。

输出格式

按照从小到大的顺序输出所有方案，每行 1 个。

首先，同一行内的数升序排列，相邻两个数用一个空格隔开。

其次，对于两个不同的行，对应下标的数一一比较，字典序较小的排在前面（例如 1 3 5 7 排在 1 3 6 8 前面）。

数据范围

n>0 ,
0≤m≤n,
n+(n−m)≤25

输入样例：

5 3

输出样例：

1 2 3 
1 2 4 
1 2 5 
1 3 4 
1 3 5 
1 4 5 
2 3 4 
2 3 5 
2 4 5 
3 4 5 

思路

数组：st记录解决方案，不需要used数组来判断数字是否被使用，因为要选择放进去的数字一定比前一个放进去的数字要大

dfs函数的参数：当前数位u和当前最小可以从哪个数字开始枚举start

剪枝：如果把后面的数字都选上也不够m个数字的话，那么肯定无解。当前正在选第u个数，那么说明已经选了u-1个数，然后从start到n里再继续选，如果已经选了的这u-1个数再加上【start，n】里的所有数都<m的话，说明肯定不够了，就可以去掉这个分支

u-1+n-start+1<m  等价于  u-start+n<m

代码

自己写的：比较繁琐，用到了一个记录已经存了多少位的变量t

#include<bits/stdc++.h>
using namespace std;
int n,m;
const int N=25;
int st[N];
//int used[N];

void dfs(int u,int t){
    if(t==m+1){
        for(int i=1;i<=m;i++)
            printf("%d ",st[i]);
        puts("");
        return;
    }
    for(int i=1;i<=n;i++){
        if(i>st[u-1]){
            //used[i]=true;
            t++;
            st[u]=i;
            dfs(u+1,t);

            ///恢复现场
            t--;
            //used[i]=false;
        }
    }
}

int main(){
    cin>>n>>m;
    dfs(1,1);
    return 0;
}

yxc代码：

#include<bits/stdc++.h>
using namespace std;
int n,m;
const int N=25;
int st[N];

void dfs(int u,int start){
    if(u+n-start<m) return;///剪枝
    if(u==m+1){
        for(int i=1;i<=m;i++)
            printf("%d ",st[i]);
        puts("");
        return;
    }
    for(int i=start;i<=n;i++){
            st[u]=i;
            dfs(u+1,i+1);
            ///恢复现场
        }
}


int main(){
    cin>>n>>m;
    dfs(1,1);
    return 0;
}