【算法笔记第8.1节-DFS 】问题 B: 【递归入门】组合的输出

题目描述

排列与组合是常用的数学方法，其中组合就是从n个元素中抽出r个元素(不分顺序且r < ＝ n)，我们可以简单地将n个元素理解为自然数1，2，…，n，从中任取r个数。
现要求你不用递归的方法输出所有组合。
例如n ＝ 5 ，r ＝ 3 ，所有组合为：
1 2 3
1 2 4
1 2 5
1 3 4
1 3 5
1 4 5
2 3 4
2 3 5
2 4 5
3 4 5 

输入

一行两个自然数n、r ( 1 < n < 21，1 < ＝ r < ＝ n )。

输出

所有的组合，每一个组合占一行且其中的元素按由小到大的顺序排列，所有的组合也按字典顺序。

#include<stdio.h>
#include<algorithm>
#include<vector>
using namespace std;
int n, a[100], k;
int ans[100];
int vis[200];
void dfs(int index)
{
    if(index == k)
    {
        for(int i=0; i<k; i++)
        {
            printf("%d", ans[i]);
            if(i==k-1)
                printf("\n");
            else
                printf(" ");
        }
    }
    for(int i = ans[index-1]; i<=n; i++)
    {
        if(vis[i]==0)
        {
            ans[index] = i;
            vis[i] = 1;
            dfs(index+1);
            vis[i] = 0;
        }
    }
    return;
}
int main()
{
   while(scanf("%d %d",&n, &k)!=EOF)
   {

       for(int i=0; i<n; i++)
       {
         vis[i] = 0;
       }
       for(int i=1; i<=n; i++)
       {
           ans[0] = i;
           vis[i] = 1;
           dfs(1);
       }
   }
    return 0;
}