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

//问题: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);

    }

}

点赞