0

这是我执行此操作的代码。它适用于字符串表示，但不是那个ArrayList<ArrayList<Integer>>。

public static void partition(int n) {
    partition(n, n, "",
            new ArrayList<ArrayList<Integer>>(),
            new ArrayList<Integer>());
}

public static void partition(int n, int max, String temp,
                             ArrayList<ArrayList<Integer>> master,
                             ArrayList<Integer> holder) {
    if (n == 0) {
        ArrayList<Integer> temp1 = new ArrayList<Integer>();
        for (int i = 0; i <= holder.size(); i++) {
            temp1.add(holder.get(0));
            holder.remove(0);
        }
        master.add(temp1);
        System.out.println(temp);
    }

    for (int i = Math.min(max, n); i >= 1; i--) {
        holder.add(i);
        partition(n - i, i, temp + " " + i, master, holder);
    }
}

我用 temp1 做有趣的事情的原因是，如果我只是将 temp 添加到 master，以前的元素会改变（master 中的所有元素都是指向同一个地方的引用）所以这是我深入的尝试复制+清除。

字符串有效。为什么不ArrayList<ArrayList<Integer>>呢？我该如何解决？

的输出ArrayList<ArrayList<Integer>>是：

[[5], [4, 1], [3, 2], [1, 1], [2, 2], [1, 1, 1], [1, 1, 1, 1]]
4

2 回答 2

3

你在你不应该做的分支内混合了你的holder数组。if尝试以下操作：

public static void partition(int n, int max, String temp,
                             ArrayList<ArrayList<Integer>> master,
                             ArrayList<Integer> holder) {
    if (n == 0) {
        ArrayList<Integer> temp1 = new ArrayList<Integer>();
        for (int i = 0; i < holder.size(); i++) {
            temp1.add(holder.get(i));
        }
        master.add(temp1);
        System.out.println(temp);
    }

    for (int i = Math.min(max, n); i >= 1; i--) {
        holder.add(i);
        partition(n - i, i, temp + " " + i, master, holder);
        holder.remove(holder.size() - 1);
    }
}
于 2011-09-22T13:31:36.797 回答
0

您必须将副本holder传递给递归的每个后续分支。

爪哇 7：

public static void main(String[] args) {
    List<List<Integer>> list = partition(5);
    for (List<Integer> comb : list)
        System.out.println(comb);
}

public static List<List<Integer>> partition(int n) {
    List<List<Integer>> list = new ArrayList<>();
    partition(n, n, "", list, new ArrayList<Integer>());
    return list;
}

public static void partition(int i, int max, String indent,
                             List<List<Integer>> master, List<Integer> holder) {
    if (i == 0) {
        master.add(holder);
        System.out.println(
                indent + "i=" + i + ",max=" + max + "    comb=" + holder);
    }

    for (int j = Math.min(max, i); j >= 1; j--) {
        ArrayList<Integer> temp = new ArrayList<>(holder);
        temp.add(j);
        System.out.println(
                indent + "i=" + i + ",j=" + j + ",max=" + max + "  temp=" + temp);
        partition(i - j, j, indent + "  ", master, temp);
    }
}

递归树的输出n=5：

i=5,j=5,max=5  temp=[5]
  i=0,max=5    comb=[5]
i=5,j=4,max=5  temp=[4]
  i=1,j=1,max=4  temp=[4, 1]
    i=0,max=1    comb=[4, 1]
i=5,j=3,max=5  temp=[3]
  i=2,j=2,max=3  temp=[3, 2]
    i=0,max=2    comb=[3, 2]
  i=2,j=1,max=3  temp=[3, 1]
    i=1,j=1,max=1  temp=[3, 1, 1]
      i=0,max=1    comb=[3, 1, 1]
i=5,j=2,max=5  temp=[2]
  i=3,j=2,max=2  temp=[2, 2]
    i=1,j=1,max=2  temp=[2, 2, 1]
      i=0,max=1    comb=[2, 2, 1]
  i=3,j=1,max=2  temp=[2, 1]
    i=2,j=1,max=1  temp=[2, 1, 1]
      i=1,j=1,max=1  temp=[2, 1, 1, 1]
        i=0,max=1    comb=[2, 1, 1, 1]
i=5,j=1,max=5  temp=[1]
  i=4,j=1,max=1  temp=[1, 1]
    i=3,j=1,max=1  temp=[1, 1, 1]
      i=2,j=1,max=1  temp=[1, 1, 1, 1]
        i=1,j=1,max=1  temp=[1, 1, 1, 1, 1]
          i=0,max=1    comb=[1, 1, 1, 1, 1]

列表的输出n=5：

[5]
[4, 1]
[3, 2]
[3, 1, 1]
[2, 2, 1]
[2, 1, 1, 1]
[1, 1, 1, 1, 1]

另请参阅：构建有效地求和为数字的排列

于 2021-05-03T12:41:19.550 回答