Diagonal Difference HackerRank Solution ( Python )

DiagonalDifference.py

n = int(input())
lst = []
for i in range(n):
    lst.append([int(x) for x in input().split()])
a, b = 0, 0
for i in range(n):
    a += lst[i][i]
    b += lst[i][n-i-1]
print(abs(a-b))

"""
Alternate solution: 

n = int(input())
lst = []
for i in range(n):
    lst.append([int(x) for x in input().split()])
a, b = 0, 0
for i in range(n):
    a += lst[i][i]
    b += lst[i][-i-1]
print(abs(a-b))

"""