java - 如何循环数组的“n”维？

问题描述

我正在尝试创建一个方法来总结潜在项目并从数组中返回该总和。以下是一些预期的示例输入：

arraySum(new int[10]); // 10
arraySum(new int[2][5]); // 10
arraySum(new int[5][5][5]); // 125

问题是我永远无法真正知道我正在处理多少个维度。到目前为止，我发现调用String.valueOf(array)数组会返回一个字符串，该字符串包含[数组中存在的许多维度的字符：

String.valueOf(new int[10]); // [I@hash_code
String.valueOf(new int[5][2]); // [[I@hash_code
String.valueOf(new int[5][5][5]); // [[[I@hash_code

我可以使用它来找出该数组存在的确切维度数量String.valueOf(array).split("\\[").length - 1。但是，我不确定从这一点上我能做什么。

public int arraySum(final Object array) {
    checkArgument(array.getClass().isArray()); // from com.google.common.base.Preconditions
    final int dimensions = String.valueOf(array).split("\\[").length - 1;
    int sum = 0;

    // somehow loop n-times over here to count up

    return sum;
}

通常在对多维数组进行迭代时，您会在彼此内部有多个循环，但在我的情况下，我需要 n 循环，我显然无法对其进行硬编码。我能做些什么？

标签： javaarraysfor-loop

这样做：

public static int arraySum(final Object[] array) {
    if(array.length == 0) return 0;
    return array.length * ((array[0] instanceof Object[]) ? arraySum((Object[]) array[0]) : 1);
}

您还可以为基元数组重载此方法。以下是我将如何以最少的代码重复来做到这一点：

private static int doArraySum(Object array) {
    if(array == null || !array.getClass().isArray()) return 1;
    int length = Array.getLength(array);
    if(length == 0) return 0;
    return length * doArraySum(Array.get(array, 0));
}

public static int arraySum(Object[] array) {
    return doArraySum(array);
}

public static int arraySum(int[] array) {
    return doArraySum(array);
}

// other primitives

如果子数组的长度可能不相等，则应使用加法而不是乘法：

private static int doArraySum(Object array) {
    if (array == null || !array.getClass().isArray()) return 1;
    return IntStream
            .range(0, Array.getLength(array))
            .map(i -> doArraySum(Array.get(array, i)))
            .sum();
}

例子：

int[][] array = new int[2][];
array[0] = new int[2];
array[1] = new int[3];
arraySum(array); // 5

java - 如何循环数组的“n”维？

问题描述

解决方案

推荐阅读