7-20 矩阵运算 - Python

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

输入格式:

输入第一行给出正整数n（1<n≤10）；随后n行，每行给出n个整数，其间以空格分隔。

输出格式:

在一行中给出该矩阵除副对角线、最后一列和最后一行以外的所有元素之和。

输入样例:

4
2 3 4 1
5 6 1 1
7 1 8 1
1 1 1 1

输出样例:

35

代码如下：

n = int(input())
temp = []
sum1 = 0
for i in range(n):
    temp.append(list(map(int, input().split()[0:n])))

for i in range(0,n):
    for j in range(0,n):
        if i!=n-1 and j!=n-1 and i+j!=n-1:
            sum1 +=temp[i][j]
            flag=0
print(sum1)