java - 将数组扩展为表示索引的次数与原始值的大小相同

Question

示例:expand(new int[]{3, 2, 5}) -> {0, 0, 0, 1, 1, 2, 2, 2, 2, 2}
我试图让它创建一个新数组来打印索引 3、3 次。所以 3 就是 0,0,0。

public static int[] expand(int[] input) {
    int c = 0;
    int[] myArray = new int[sum(input)];
    if(input.length == 0){
        return new int[0];
    }
    for(int i = 0; i < input.length; i++) {
        int a = input[i];
        for(int j = c; j < a; j++) {
            c += j;
            myArray[j] = i;

        }
    }

    return myArray;
}

目前这仅部分有效，我似乎无法弄清楚如何正确地遍历整个数组。此外，索引零似乎被跳过。

Answer 1

你很接近!看来您只需要稍微修改一下嵌套的 for 循环即可:

for (int j = 0; j < a; j++) {
    myArray[c++] = i;
}

由于您使用 c 来跟踪当前索引，这只是将索引 c 处的元素设置为 i 并递增c。您还可以删除 a 并使用 input[i] 代替它。

注意:从 0 启动 j 比从 c 启动更容易。