[Lucky Algorithm] Diagonal Difference (3/77)

Diagonal Difference (Hacker Rank)

Given a square matrix of size N X N, calculate the absolute difference between the sums of its diagonals.

#!/bin/python3

import sys

def diagonal(arr):
    pri_dia, sec_dia = 0, 0
    for i,a in enumerate(arr):
        pri_dia += int(a[i])
        sec_dia += int(a[len(arr)-i-1])
    result = abs(pri_dia - sec_dia)

    return result

n = int(input().strip())
a = []
for a_i in range(n):
    a_t = [int(a_temp) for a_temp in input().strip().split(' ')]
    a.append(a_t)

print(diagonal(a))

행운의 77문제 프로젝트

  행운의 77문제 프로젝트는 한 달동안 알고리즘 문제 77개를 푸는 프로젝트입니다.