DFS实现排列组合（C++）

说起排列组合，它们在古典概率论和组合数学中应用十分广泛，大家可能也比较熟悉，在高中阶段就已经接触过它们了。这篇文章从编程的角度出发，使用DFS实现它们。

排列

排列，是指从给定个数的元素中取出指定个数的元素进行排序。

问题：给定一个数n，在[1,n]中选m个数，按字典序输出所有排列。

DFS代码：

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

int m,n;//与题目中的n,m意义一致
vector<int>ans;//存储答案
vector<bool>vis;//保证搜索过的点不再重复搜索（排列数不能重复）

void permutation(){
    if (ans.size()==m){//答案符合要求直接输出
        for (int i = 0; i < m; ++i) {
            cout<<ans[i]<<' ';
        }
        cout<<endl;
        return;//打印完后退出递归
    }
    for (int i = 1; i <= n; ++i) {
        if (!vis[i]){
            ans.push_back(i);//添加答案
            vis[i]=true;//标记访问过
            permutation();//递归搜索
            vis[i]=false;//取消标记
            ans.pop_back();//删除答案
        }
    }
}

int main(){
    cout<<"依次输入n和m:"<<endl;
    cin>>n>>m;
    vis.assign(n,false);
    permutation();
    return 0;
}

时间复杂度：$O(A_n^m)$

效果展示：

全排列

全排列是排列的一种特殊形式，即$n=m$的特殊情况。

STL大法好~~ 其实在c++的标准模板库STL中，早就有了相应的函数，它们分别是next_permutation()和prev_permutation中。前者求原数组排列的下一个排列，后者求原数组排列的上一个排列。

• 头文件：algorithm
• 返回值：bool值，表示是否有下一个或上一个排列存在
• 参数：2个，排列起始地址和终止地址（跟其他函数一样，左闭右开）

下面来看具体用法：

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

int main(){
    array<int,3>v={1,2,3};
    do {
        for (int i : v) {
            cout<<i<<' ';
        }
        cout<<endl;
    } while (next_permutation(v.begin(),v.end()));//求下一个排列
    
    cout<<endl;
    reverse(v.begin(), v.end());//数组反转，完成后是{3,2,1}
    
    do {
        for (int i : v) {
            cout<<i<<' ';
        }
        cout<<endl;
    } while (prev_permutation(v.begin(),v.end()));//求上一个排列
    return 0;
}

输出展示：

怎么样？是不是很方便！

组合

组合是指从给定个数的元素中仅仅取出指定个数的元素，不考虑顺序。

问题：给定一个数n，在[1,n]中选m个数，按字典序输出所有组合。

DFS代码：

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

int m,n;//与题目中的n,m意义一致
vector<int>ans;//存储答案

void combination(int pos=1){
    if (ans.size()==m){//答案符合要求直接输出
        for (int i = 0; i < m; ++i) {
            cout<<ans[i]<<' ';
        }
        cout<<endl;
        return;//打印完后退出递归
    }
    for (int i = pos; i <= n; ++i) {//从pos处开始遍历，是它与排列的不同之处
        //组合不再需要vis数组记录访问过程，因为这种搜索方式决定了它不可能会重复
        ans.push_back(i);//添加答案
        combination(i+1);//在下一个位置继续搜索
        ans.pop_back();//删除答案
    }
}

int main(){
    cout<<"依次输入n和m:"<<endl;
    cin>>n>>m;
    combination();
    return 0;
}

时间复杂度：$O(C_n^m)$

效果展示：

大家可以对比一下二者代码，理解二者不同的实现细节。

值得一提的是，从DFS的代码中，搜索出来的答案天然就是字典序，大家可以想想这是为什么(●’◡’●)。