DFS-3-递归实现组合型枚举

题目

从 1∼n这 n 个整数中随机选出 m 个，输出所有可能的选择方案。

输入格式

两个整数 n,m在同一行用空格隔开。

输出格式

按照从小到大的顺序输出所有方案，每行 1 个。

首先，同一行内的数升序排列，相邻两个数用一个空格隔开。

其次，对于两个不同的行，对应下标的数一一比较，字典序较小的排在前面（例如 1 3 5 7 排在 1 3 6 8 前面）。

输入样例

5 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 

思路

以输入样例为例，有三个位置，每个位置去选数，用一个state[]数组来存放已经选的数，用一个变量u来表示当前位置，用used[]来判断哪些数被选择了，因为后面的数都比前面的数大，所以在递归时，循环的时候从前一位数的下一位开始循环，所以定义start变量来控制循环的起点。

代码

import java.util.Scanner;

public class Main {
    static int n;
    static int m;

    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        n = sc.nextInt();
        m = sc.nextInt();
        int[] state = new int[m+1];
        boolean[] used = new boolean[n+1];
        dfs(state,used,1,1);
    }
   static void dfs(int[] state,boolean[] used,int u,int start){
       if(state.length == 1){
           System.out.println(0);
           return;
       }
       if(u > m){
          for(int i = 1; i <= m; i++){
              System.out.print(state[i]);
          }
          System.out.println();
          return;
       }
       for(int i = start; i <= n; i++){
           if(!used[i]){
               state[u] = i;
               used[i] = true;
           }
           dfs(state,used,u+1,i+1);
           state[u] = 0;
           used[i] = false;
       }
   }
}

改进

used[]数组是判重数组，在实现排列型枚举时，例如123和132，第二个位置的数是从1-n循环，第三个数也是，但是在这个题目中，选中12，第三个位置只能从3-5中循环，所以不再需要判重数组

import java.util.Scanner;

public class Main {
    static int n;
    static int m;

    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        n = sc.nextInt();
        m = sc.nextInt();
        int[] state = new int[m+1];
        dfs(state,1,1);
    }
   static void dfs(int[] state,int u,int start){
       if(state.length == 1){
           System.out.println(0);
           return;
       }
       if(u-1 + n - start + 1 < m) return;   //剪枝，在位置u时，如果把剩下的全选上也不够m个数的话。就直接可以退出了
       if(u > m){
          for(int i = 1; i <= m; i++){
              System.out.print(state[i]);
          }
          System.out.println();
          return;
       }
       for(int i = start; i <= n; i++){
           state[u] = i;
           dfs(state,u+1,i+1);
           state[u] = 0;
       }
   }
}