DFS之枚举排列组合

来源：互联网发布：网上车市数据运维编辑：程序博客网时间：2024/06/07 03:23

枚举可重复排列的模板

#include<iostream>using namespace std;int n,m,ans[15];int a[15];//待排列的数存储在此bool vis[15];void dfs(int cnt)//按字典序输出n个数选m个数的所有排列{    if(cnt==m)    {        for(int i=0;i<m;i++) cout<<ans[i]<<" ";        cout<<endl;        return ;    }    for(int i=0;i<n;i++)    {        ans[cnt]=a[i];        dfs(cnt+1);    }}int main(){    while(cin>>n>>m)    {        fill(vis,vis+15,0);        for(int i=0;i<n;i++) cin>>a[i];        dfs(0);    }    return 0;}

枚举不可重复的排列，加个标记数组即可。

#include<iostream>using namespace std;int n,m,ans[15];int a[15];//待排列的数存储在此bool vis[15];void dfs(int d,int cnt)//按字典序输出n个数选m个数的所有排列{    if(cnt==m)    {        for(int i=0;i<m;i++) cout<<ans[i]<<" ";        cout<<endl;        return ;    }    for(int i=0;i<n;i++)    {        if(!vis[i])        {            ans[cnt]=a[i];            vis[i]=1;            dfs(i+1,cnt+1);            vis[i]=0;        }    }}int main(){    while(cin>>n>>m)    {        fill(vis,vis+15,0);        for(int i=0;i<n;i++) cin>>a[i];        dfs(0,0);    }    return 0;}

枚举组合。

#include<iostream>using namespace std;int n,m,ans[15];int a[15];//待排列的数存储在此bool vis[15];void dfs(int d,int cnt)//按字典序输出n个数选m个数的所有组合{    if(cnt==m)    {        for(int i=0;i<m;i++) cout<<ans[i]<<" ";        cout<<endl;        return ;    }    for(int i=d;i<n;i++)    {        ans[cnt]=a[i];        dfs(i+1,cnt+1);    }}int main(){    while(cin>>n>>m)    {        fill(vis,vis+15,0);        for(int i=0;i<n;i++) cin>>a[i];        dfs(0,0);    }    return 0;}

0 0