编写程序,求某一方阵中对角线元素和。要求:考虑主对角线和副对角线
时间: 2024-05-01 07:23:59 浏览: 79
matlab主对角线元素为0代码-WaveGliderModel:与MSc有关的代码的存储库。论文
以下是一个示例程序,可以计算方阵中主对角线和副对角线上的元素之和:
```python
def diagonal_sum(matrix):
n = len(matrix)
sum1 = 0
sum2 = 0
for i in range(n):
sum1 += matrix[i][i] # 计算主对角线上的元素和
sum2 += matrix[i][n-i-1] # 计算副对角线上的元素和
return sum1 + sum2 # 返回两个对角线上的元素和之和
# 示例用法
matrix = [[1, 2, 3],
[4, 5, 6],
[7, 8, 9]]
print(diagonal_sum(matrix)) # 输出:25
```
该程序首先计算主对角线上的元素和,然后计算副对角线上的元素和,最后将两者之和返回即可。在计算副对角线元素和时,可以使用 `n-i-1` 来计算每个元素在副对角线上的位置。
阅读全文