python-n矩阵主对角线元素之和

404 阅读1分钟

题目

评测机会通过执行命令 python main.py 来执行你的代码。

你的代码需要从标准输入流(控制台)中读入数据 n 和 nums(共n + 1行,矩阵按行顺序输入,每行 n 个整数,以空格分隔),计算出结果后并打印到标准输出流(控制台)中。

解法一 先存矩阵,再将对角线上元素一个一个相加 < n = int (input())

nums = [] for _ in range(n): nums.append(input().split())

answer = 0

for i in range(n): answer += int(nums[i][i])

print(answer) />

解法二 在输入的时候就把元素相加 输入流是按\n划分行,空格划分行内数字

所以可以直接加

<

n = int( input() )

answer = 0 for i in range(n): now_line = input().split() answer += int( now_line[i] )

print(answer)

>