递归实现组合型枚举

题目描述

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

输入格式

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

输出格式

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

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

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

数据范围

$n>0$
$0\le m\le n$
$n+(n-m)\le 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 <iostream>
#include <vector>
using namespace std;

int n,m;
vector<int> result;

void dfs(int index)
{
    if(result.size() + n - index < m)
    {
        return ;
    }
    if(result.size() == m)
    {
        for(int i = 0;i < result.size();i++)
        {
            cout << result[i] << " ";
        }
        cout << endl;
        return ;
    }
    
    result.push_back(index + 1);
    dfs(index + 1);
    result.pop_back();
    dfs(index + 1); 
}

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

其二使用函数参数存储，使用位运算表示（当前位为1表示该位表示的数存在）。

#include <iostream>
using namespace std;

int n,m;

void dfs(int index,int sum,int status)
{
    if(sum + n - index < m)
    {
        return ;
    }
    if(sum == m)
    {
        for(int i = 0;i < n;i++)
        {
            if(status >> i & 1)
            {
                cout << (i + 1) << " ";
            }
        }
        cout << endl;
        return ;
    }
    
    dfs(index + 1,sum + 1,status | 1 << index);
    dfs(index + 1,sum,status);
    
}

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