nyoj19擅长排列的小明(DFS或STL)

擅长排列的小明

时间限制：1000 ms | 内存限制：65535 KB

难度：4

描述 小明十分聪明，而且十分擅长排列计算。比如给小明一个数字5，他能立刻给出1-5按字典序的全排列，如果你想为难他，在这5个数字中选出几个数字让他继续全排列，那么你就错了，他同样的很擅长。现在需要你写一个程序来验证擅长排列的小明到底对不对。

输入第一行输入整数N（1<N<10）表示多少组测试数据，

每组测试数据第一行两个整数 n m (1<n<9,0<m<=n)输出在1-n中选取m个字符进行全排列，按字典序全部输出,每种排列占一行，每组数据间不需分界。如样例样例输入

2
3 1
4 2

样例输出

1
2
3
12
13
14
21
23
24
31
32
34
41
42
43

题意：输入连个数n,m。从1-n中选出m个数求出所有的全排列，按字典序输出

思路：因为之前做过一个全排列的题用的是暴力，然后发现有stl函数直接用，所以还是先暴力dfs写了一下 1A，入门级的dfs。STL的写法就参考别人的代码了。

code：

#include <stdio.h>
#include <string.h>
int ans[10],vis[10];
int n;

void dfs(int num,int m)
{
int i;
if(num==m)
{
for(i=0;i<num;i++)
printf("%d",ans[i]);
printf("\n");
return ;
}
else
{
for(i=1;i<=n;i++)
{
if(vis[i]==0)
{
vis[i]=1;
ans[num++]=i;
dfs(num,m);
num--;                  //好无语，这个num也是要回溯的，这都忘记了哎，代码功底不行啊
vis[i]=0;
}
}
}
}

int main()
{
int t,m,i;
scanf("%d",&t);
while(t--)
{
scanf("%d%d",&n,&m);
memset(vis,0,sizeof(vis));
for(i=1;i<=n;i++)
{
vis[i]=1;
ans[0]=i;
dfs(1,m);
vis[i]=0;           //开始的时候这里忘记回溯了，orz...
}
}
return 0;
}

STL：

此题为全排列的前m个数，只需对n个数调用全排列函数next_permutation()，去除重复的输出前m个即可。

改写的全排列，这里用字符串输入，方便判断是否重复输出。

code：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<string>
using namespace std;
int n,m;

int main()
{
int t;
cin>>t;
while(t--){
cin>>n>>m;
string s,s1;
for(int i=1;i<=n;i++) s+=i+'0';
s1=s.substr(0,m);
cout<<s1<<endl;
while(next_permutation(s.begin(),s.end())){
if(s1!=s.substr(0,m))
{                                   //用来判断重复输出,否则会输出多次
s1=s.substr(0,m);
cout<<s1<<endl;
}
}
}
return 0;
}