说起排列组合,它们在古典概率论和组合数学中应用十分广泛,大家可能也比较熟悉,在高中阶段就已经接触过它们了。这篇文章从编程的角度出发,使用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 ) O(A_n^m) O(Anm)
效果展示:
全排列
全排列是排列的一种特殊形式,即 n = m 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 ) O(C_n^m) O(Cnm)
效果展示:
大家可以对比一下二者代码,理解二者不同的实现细节。
值得一提的是,从DFS的代码中,搜索出来的答案天然就是字典序,大家可以想想这是为什么(●’◡’●)。