【PTA-python】第4章-19 矩阵运算 (20 分)--减法实现

分析

这里学习如何存储矩阵
for 循环 list.extend()去连接另一个列表。
特别处理：第一行要减去一个元素，最后一行要减去n个元素。
f其他行都是减去两个元素。
另外注意*作为乘法，不能直接n(n-1)，会报错TypeError: 'int' object is not callable

题目

第4章-19 矩阵运算 (20 分)

给定一个n×n的方阵，本题要求计算该矩阵除副对角线、最后一列和最后一行以外的所有元素之和。副对角线为从矩阵的右上角至左下角的连线。

解法

n=int(input())
lst=[]
s=0
for i in range(n):
    lst.extend(list(map(int,input().split())))
for i in range(n*n):
    s+=lst[i]
for i in range(1,n-1):
    s=s-lst[(i+1)*n-1]-lst[(i+1)*n-1-i]
for i in range(n):
    s=s-lst[n*(n-1)+i]
print(s-lst[n-1])