编写程序,求某一方阵中对角线元素之和,要求:考虑主对角线和副对角线
时间: 2024-01-08 09:47:21 浏览: 67
C语言程序设计-分别求一个3×3的整型矩阵两对角线元素之和(或之积)。
5星 · 资源好评率100%
好的,以下是 Python 语言的程序示例:
```python
def diagonal_sum(matrix):
n = len(matrix)
sum_main = 0
sum_secondary = 0
for i in range(n):
sum_main += matrix[i][i]
sum_secondary += matrix[i][n-i-1]
return sum_main + sum_secondary
```
这个函数接受一个方阵(二维数组)作为输入,返回该方阵所有对角线元素之和。其中,`sum_main` 记录主对角线元素之和,`sum_secondary` 记录副对角线元素之和。对于一个 `n` 行 `n` 列的方阵,主对角线元素的下标为 `(i, i)`,副对角线元素的下标为 `(i, n-i-1)`。
你可以调用这个函数,将一个方阵作为参数传入,即可得到对角线元素之和。
阅读全文