Answer to Question #188630 in Python for phani

Question #188630
Matrix Diagonal Sum

Given a square matrix, print the sum all diagonal elements of the matrix.

Input

The first line of input will be an integer N, denoting the no. of rows in the matrix. The following N lines will contain space-separated integers denoting the elements of each row of the matrix.

Output

The output should be a single integer denoting the sum of all the diagonal elements.

Explanation

For example, if the given N is 3, and the given matrix is
1 2 3
4 5 6
7 8 9
The diagonal and anti-diagonal elements of the above matrix are 1, 3, 5, 7, 9. So the output should be 25

Sample Input 
3
1 2 3
4 5 6
7 8 9
Sample Output 
25


1
Expert's answer
2021-05-03T10:26:56-0400
n = int(input())
a = []
for i in range(n):
    a.append(list(map(int, input().split())))
sum_first_diagonal = sum(a[i][i] for i in range(n))
sum_second_diagonal = 0
for i in range(n):
    for j in range(n):
        if i + j == n - 1:
            sum_second_diagonal += a[i][j]
if n % 2 == 0:
    sum = sum_first_diagonal + sum_second_diagonal
else:
    sum = sum_first_diagonal + sum_second_diagonal - a[int((n - 1) / 2)][int((n - 1) / 2)]
print(sum)

Need a fast expert's response?

Submit order

and get a quick answer at the best price

for any assignment or question with DETAILED EXPLANATIONS!

Comments

No comments. Be the first!

Leave a comment

LATEST TUTORIALS
New on Blog
APPROVED BY CLIENTS