输入一个二维矩阵(4*4),通过函数求对角线元素之和
时间: 2023-10-07 13:12:23 浏览: 82
C语言程序设计-分别求一个3×3的整型矩阵两对角线元素之和(或之积)。
5星 · 资源好评率100%
好的,可以通过如下的 Python 代码实现:
```python
def diagonal_sum(matrix):
n = len(matrix)
total_sum = 0
for i in range(n):
total_sum += matrix[i][i] + matrix[i][n-i-1]
return total_sum - matrix[n//2][n//2] if n % 2 == 1 else total_sum
# 测试
matrix = [[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12], [13, 14, 15, 16]]
print(diagonal_sum(matrix)) # 输出 68
```
这里的 `diagonal_sum` 函数接受一个二维矩阵作为输入,返回对角线元素之和。具体实现方式是,先遍历第一个对角线(从左上角到右下角),再遍历第二个对角线(从右上角到左下角),将对角线元素累加起来。注意,如果矩阵的维度是奇数,则中心位置的元素会被重复计算,需要减去一次。
阅读全文