矩阵元素之和 Java

Question

我正在编写一个带有 2 个参数、一个 int 矩阵和 1 个 int 数组的方法。该方法必须返回数组 b[] 中具有倍数的 a[][] 元素的总和。

这是方法代码

public static int sum(int[][] a, int[] b) {
    int matrixSum = 0;

    if (a != null && b != null) {
        for (int i = 0; i < a.length; i++) {
            for (int j = 0; j < a.length; j++) {
                boolean multiple = false;
                boolean nullRow = false;
                if (a[i] == null || a[i].length < 1) {
                    nullRow= true;
                } else {
                    for (int k = 0; k < b.length && !multiple; k++) {
                        if (b[k] % a[i][j] == 0) {
                            matrixSum += a[i][j];
                            multiple = true;
                        }
                    }
                }
            }
        }
    }
    return matrixSum;
}

这是测试类

public static void main(String[] args) {
    int[][] a = {{2, 3, 1}, {1, 1, 1}, {2, 2, 2}};
    int[][] a1 = {{2, 3}, {1, 1, 1}, {2, 2, 2}};
    int[][] a2 = {{2, 3, 4}, {1, 1}, {2, 2, 2}};
    int[][] a3 = {{2, 3, 4}, {1, 2, 3}, {1, 1}};

    int[] b = {7, 8, 7, 20};


    System.out.println(sum(a, b)); // 15
    System.out.println(sum(a1, b)); // 11
    System.out.println(sum(a2, b)); // 14
    System.out.println(sum(a3, b)); // 11

}

在某些情况下它可以在其他情况下不工作，有些结果不正确，而其他时候我收到 OutBound 错误

Answer 1

试试这个：

public static int sum(int[][] a, int[] b) {

    if (a == null || b== null)  return 0;
    int matrixSum = 0;

    for (int i = 0; i < a.length; i++) {
        if(a[i] == null || a[i].length <1 )     continue;
        for (int j = 0;  j < a[i].length; j++) {
            for (int k = 0; k < b.length ; k++) {
                if (b[k] % a[i][j] == 0) {
                    matrixSum += a[i][j];
                    break;
                }
            }
        }
    }

    return matrixSum;
}

使用Stream可以更简单更简洁：

public static int sum(int[][] a, int[] b) {

    return Stream.of(a).flatMapToInt( row -> IntStream.of(row)).
                 filter(i->isMultipleOf(i,b)).sum();
}

private static boolean isMultipleOf(int i, int[] b) {
    for(int j : b){
        if(j%i == 0) return true;
    }
    return false;
}

Answer 2

我很确定 IndexOutOfBounds 的问题在于：

for (int j = 0; j < a.length; j++)

需要询问第二维的长度，如：

for (int j = 0; j

Answer 3

此代码有效，谢谢大家

public static int sum(int[][] a, int[] b) {
    int matrixSum = 0;

    if (a != null && b != null) {
        for (int i = 0; i < a.length; i++) {
            if (a[i] == null) {
                boolean nullRow = true;
            } else {
                for (int j = 0; j < a[i].length; j++) {
                    boolean multiple = false;
                    for (int k = 0; k < b.length && !multiple; k++) {
                        if (b[k] % a[i][j] == 0) {
                            matrixSum += a[i][j];
                            multiple = true;
                        }
                    }
                }
            }
        }
    }

    return matrixSum;
}