在一个N行N列的方阵(或称N阶方阵)中,从左上角到右下角这一斜线上有N个数据元素,这个斜线称为方阵的主对角线。给你一个方阵,请求方阵主对角线上数据的和。 输入 第一行是N(N<100),表示下边是一个N阶方阵。接下来N行N列用空格间隔放置正整数(int型)。 输出 N阶方阵主对角线上数据的和。
时间: 2024-03-09 10:49:26 浏览: 40
C语言程序设计-用函数求一个N阶方阵右下三角元素的和(包括副对角线上的元素);
4星 · 用户满意度95%
好的,这是一道简单的算法题,我可以帮助你解决。以下是我给出的 Python 代码实现:
```python
n = int(input())
matrix = []
for i in range(n):
row = list(map(int, input().split()))
matrix.append(row)
sum = 0
for i in range(n):
sum += matrix[i][i]
print(sum)
```
首先,我们读入一个整数 `n`,表示方阵的阶数。接着,我们读入 `n` 行 `n` 列的矩阵,用一个二维列表 `matrix` 存储它们。然后,我们遍历主对角线上的元素,即第 `i` 行第 `i` 列的元素,将它们相加,最后输出结果。
希望这个代码能够帮助你解决问题。如果你还有其他问题,请继续提问。
阅读全文