Question

我真的陷入了这个问题。

我希望生成所有组合，以便有替换的有序采样（我认为这就是它的名称，请参阅我的示例），结果以数组（2D数组）的形式输出。

例如

public static int[][] combinations(int n, int k)

在输入n = 3，k = 2时会给出：

[[0,0]，[0,1]，[0,2]，[1,0]，[1,1]，[1,2]，[2,0]，[2,1 ]，[2,2]]

我真的不确定如何有效地做到这一点。任何指针都表示赞赏！

Answer 1

这里我们有一个包含n元素的集合，我们希望从集合中抽取k个样本，以便排序很重要并允许重复。

这里是代码：

public static void main(String[] args) {
    int n = 4;
    int k = 2;
    int[][] result = combinations(n, k);
    System.out.println(Arrays.deepToString(result));
    //this will print:
    //[[0,0],[0,1],[0,2],[0,3],[1,0],[1,1],[1,2],[1,3],[2,0],[2,1],[2,2],[2,3],[3,0],[3,1],[3,2],[3,3]]
}

public static int[][] combinations(int n, int k) {
    int[] nArray = IntStream.range(0, n).toArray();
    List<String> list = new ArrayList<String>();
    combinationStrings(k, nArray, "", list);
    return fromArrayListToArray(list, k);
}

private static void combinationStrings(int k, int[] nArray, String currentString, List<String> list) {
    if (currentString.length() == k) {
        list.add(currentString);
    } else {
        for (int i = 0; i < nArray.length; i++) {
            String oldCurrent = currentString;
            currentString += nArray[i];
            combinationStrings(k, nArray, currentString, list);
            currentString = oldCurrent;
        }
    }
}

private static int[][] fromArrayListToArray(List<String> list, int k) {
    int[][] result = new int[list.size()][k];
    for (int i=0;i<list.size();i++) {
        String[] split = list.get(i).split("\\B");
        for (int j = 0; j < split.length; j++) {
            result[i][j] = Integer.parseInt(split[j]);
        }
    }
    return result;
}

有序取样与替换

1 个答案: