递归实现组合型枚举

原文：https://bafanglvren.ink/archives/64/

93. 递归实现组合型枚举

原题：https://www.acwing.com/problem/content/description/95/

题目

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

输入格式

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

输出格式

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

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

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

数据范围

n>0
0≤m≤n
n+(n−m)≤25

输入样例：

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

解1

每次选新的数字都要求比上一个数字大

import java.io.IOException;
import java.io.InputStreamReader;

public class Main {
	//存储状态,0表示未选
	static int []info = new int[30];
	//存储数字使用状态，false表示没用过这个数字
	static boolean []used = new boolean [30];
	
	static void dfs(int n,int m,int step) {
		if(step > m) {
			for(int i = 1;i<=m;i++) {
					System.out.print(info[i] + " ");
			}
			System.out.println();
			return;
		}
		
		for(int i = 1;i<=n;i++) {
			//新放入的数字要没用过，并且比前面的相邻数字大
			if(!used[i] && i > info[step - 1]) {
				used[i] = true;
				info[step] = i;
				//
				dfs(n,m,step+1);
				//恢复
				used[i] = false;
				info[step] = 0;
			}
		}
	}
	
	public static void main(String[] args) throws NumberFormatException, IOException {
		BufferedReader cin = new BufferedReader (new InputStreamReader(System.in));

		String input[] = cin.readLine().split(" ");
		int n = Integer.valueOf(input[0]);
		int m = Integer.valueOf(input[1]);
		
		dfs(n, m, 1);
	}

}

解2

每次选新的数字从start开始选。将start作为参数一起递归

import java.io.IOException;
import java.io.InputStreamReader;

public class Main {

	static int n,m;
	static int []ways = new int[30];
	
	static void dfs(int start,int step) {
		
		if(step + n - start + 1 < m) {//剪枝
			return ;
		}
		if(step == m + 1) {
			for(int i = 1;i<=m;i++) {
				System.out.print(ways[i]+" ");
			}
			System.out.println();
			return ;
		}
		
		for(int i = start;i<=n;i++) {
			ways[step] = i;
			dfs(i + 1,step+1);
			//恢复现场
			ways[step] = 0;
		}
	}
	public static void main(String[] args) throws Exception, IOException {
		BufferedReader cin = new BufferedReader(new InputStreamReader(System.in));
		
		String [] input = cin.readLine().split(" ");
		
		n = Integer.valueOf(input[0]);
		m = Integer.valueOf(input[1]);
		
		dfs(1, 1);

	}

}