DFS练习 【递归入门】组合的输出

题目描述

排列与组合是常用的数学方法，其中组合就是从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 )。

输出

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

【代码】：

import java.util.Scanner;

public class Main {
	static int n, r, sum = 0;
	static int p[];
	static boolean vis[];

	static void dfs(int index) {
		if (index == r) {
			print();
			return;
		}
		for (int i = 1; i <= n; i++) {
			if (vis[i] == false && (index == 0 || p[index - 1] < i)) {
				p[index] = i;
				sum++;
				vis[i] = true;
				dfs(index + 1);
				vis[i] = false;
				sum--;
			}
		}

	}

	static void print() {
		// TODO Auto-generated method stub
		for (int i = 0; i < r; i++) {
			System.out.print(p[i]);
			if (i != r - 1)
				System.out.print(" ");
			else
				System.out.println();
		}
	}

	public static void main(String args[]) {
		Scanner sc = new Scanner(System.in);
		n = sc.nextInt();
		r = sc.nextInt();
		p = new int[n + 1];
		vis = new boolean[n + 1];
		dfs(0);
	}
}