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;
} 




你可能感兴趣的:(nyoj19擅长排列的小明(DFS或STL))