求一个(n*n)矩阵对角元素之和java
时间: 2024-10-06 18:02:44 浏览: 1
在Java中,计算一个(n * n)矩阵的对角元素之和可以分为两步:首先创建一个二维数组表示矩阵,然后遍历这个数组,只累加对角线上的元素。
以下是一个简单的示例代码:
```java
public class DiagonalSum {
public static int sumDiagonal(int[][] matrix) {
if (matrix == null || matrix.length == 0) {
return 0;
}
int sum = 0;
for (int i = 0; i < matrix.length; i++) {
// 对左上到右下的对角线求和
sum += matrix[i][i];
// 如果矩阵是奇数行偶数列,则还需要考虑从右下到左上的对角线
if (i + 1 < matrix.length) {
sum += matrix[matrix.length - 1 - i][i];
}
}
return sum;
}
public static void main(String[] args) {
int[][] matrixExample = {{1, 2, 3}, {4, 5, 6}, {7, 8, 9}};
System.out.println("矩阵对角线元素之和: " + sumDiagonal(matrixExample));
}
}
```
在这个例子中,`sumDiagonal`方法接受一个二维数组作为输入,遍历每个元素,如果当前索引(i, i)位于对角线上就将其加入总和。对于奇数行偶数列的矩阵,还需要加上另一个对角线的元素(即倒数第一个元素与当前位置相对应的那个)。