算法--组合重复问题递归

//问题:AABBBC 中取3个,所有取法
public class Main {
	// data:不动,限制条件
	// x:取法
	// k:当前考虑的位置(字母)
	// goal:距离目标的剩余名额
	public static void f(int data[], int[] x, int k, int goal) {
		if (k == x.length) { // 为什么不是x.length-1呢:因为x.length-1是设置最后一种的取法,当k==x.length(越界)才把最后一个设置完
			if (goal == 0) { // 当goal==0,才把目标取完
				show(x);
				return;
			}
		}
		for (int i = 0; i < Math.min(data[k], goal); i++) { // i表示当前考虑的位置(字母)要取得个数
			x[k] = i; // 对取法数组进行赋值,设置各种取法
			f(data, x, k + 1, goal - i); // 递归
		}
		x[k] = 0; // 回溯:把前面赋好值的取法数组进行归0处理
	}

	public static void show(int[] x) {
		for (int i = 0; i < x.length; i++) { // i表示考虑的位置(字母)
			for (int k = 0; k < x[i]; k++) { // k表示当前考虑的字母取得个数
				System.out.println((char) ('A' + i)); // (char)('A'+i):表示当前考虑的字母
			}
		}
		System.out.println();
	}

	public static void main(String[] args) {
		int data[] = { 2, 3, 1 }; // 每个元素的最大个数
		int x[] = new int[data.length]; // 每个元素取几个
		f(data, x, 0, 3);

	}

}

点赞