全排列序列的生成 非递归
已知一个排序好的序列a[]
,要生成以这个序列的全排列,一般我们会dfs,但是其实有比较简单的非递归方法
- 从右往左找第一个下标
l
,使得a[l] < a[l+1]
- 从右往左找第一个下标
r
,使得a[r] > a[l]
- 交换下标
l, r
对应的元素 - 将下标
l+1 ~ 结尾
的序列翻转
代码
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
int fac(int n)
{
int sum = 1;
while(n > 1)
{sum *= n; n--;}
return sum;
}
int main()
{
int n;
cin>>n;
vector<int> a(n);
for(int i=0; i<n; i++)
a[i] = i+1;
for(int i=0; i<fac(n); i++)
{
for(int i=0; i<n; i++)
cout<<a[i]<<" ";
cout<<endl;
int l, r;
for(l=n-2; l>=0; l--)
if(a[l]<a[l+1]) break;
for(r=n-1; r>=0; r--)
if(a[r]>a[l]) break;
swap(a[l], a[r]);
reverse(a.begin()+l+1, a.end());
}
return 0;
}