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);
	}
}