其他分享
首页 > 其他分享> > 递归实现组合型枚举

递归实现组合型枚举

作者:互联网

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

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

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

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

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

数据范围
n>0n>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

#include<bits/stdc++.h>
using namespace std;

int n,m;
vector<int> num;

void dfs(int k)
{
    //如题解所述
    if(num.size() > m || num.size() + (n - k + 1) < m)
        return;
    //到达枚举边界,输出结果并结束
    if(k == n + 1)
    {
        for(int i = 0;i < num.size();++i)
            cout << num[i] << " ";

        cout << endl;
        return;
    }

    //选择这个数
    num.push_back(k);
    dfs(k+1);
    //回溯
    num.pop_back();

    //不选择这个数
    dfs(k+1);
}

int main()
{
    cin >> n >> m;

    dfs(1);

    return 0;
}

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>

using namespace std;

const int N = 30;

int n, m;
int st[N];  

void dfs(int u,int sum)
{
    if(sum+(n-u+1)<m||u>n+1)return;
    
    if (sum==m)
    {
        for (int i = 1; i <= n; i ++ )
            if (st[i] == 1)
                printf("%d ", i);
        printf("\n");
        return;
    }

    st[u] = 1;
    dfs(u + 1,sum+1);     // 第一个分支:不选
    st[u] = 0;  // 恢复现场

    st[u] = 0;
    dfs(u + 1,sum);     // 第二个分支:选
}

int main()
{
    cin >> n >>m;

    dfs(1,0);

    return 0;
}

标签:return,递归,int,dfs,枚举,num,组合型,include,size
来源: https://blog.csdn.net/qq_43738331/article/details/111482796