hackerrank - Diagonal Difference - Python3

문제: https://www.hackerrank.com/challenges/diagonal-difference/problem

정방 행렬(square matrix)에서 대각선 합의 절대 차이를 구하시오.

예를 들어 다음과 같은 정방 행렬이 있습니다.

1 2 3
4 5 6
9 8 9

1 + 5 + 9 = 15, 3 + 5 + 9 = 17. 차이의 절댓값은 |15 - 17| = 2입니다.

Input Format

첫 줄은 n. 행과 열의 개수입니다. nxn 행렬이라는 의미입니다.

다음으로 n줄 만큼의 arr[i]가 나옵니다. 각 줄은 space로 구분된 n개의 정수가 나옵니다. 각 항목은 arr[i][j]입니다.

 

Constraints

-100 <= arr[i][j] <= 100

 

Sample Input

3
11 2 4
4 5 6
10 8 -12

Sample Output

15

11 + 5 -12 = 4

4 + 5 + 10 = 19

|4-19| = 15

 

아래는 코드 템플릿입니다.

#!/bin/python3

import math
import os
import random
import re
import sys

#
# Complete the 'diagonalDifference' function below.
#
# The function is expected to return an INTEGER.
# The function accepts 2D_INTEGER_ARRAY arr as parameter.
#

def diagonalDifference(arr):
    # Write your code here

if __name__ == '__main__':
    fptr = open(os.environ['OUTPUT_PATH'], 'w')

    n = int(input().strip())

    arr = []

    for _ in range(n):
        arr.append(list(map(int, input().rstrip().split())))

    result = diagonalDifference(arr)

    fptr.write(str(result) + '\n')

    fptr.close()

diagonalDifference 함수를 완성합니다.

def diagonalDifference(arr):
    a = 0
    b = 0
    len_arr = len(arr)
    for i in range(len_arr):
        a += arr[i][i]
        b += arr[i][len_arr-i-1]
    return abs(a-b)

3x3 매트릭스인 경우 arr[0][0], arr[1][1], arr[2][2] 의 합과 arr[0][2], arr[1][1], arr[2][0]합의 차이를 구하면 됩니다.

a = arr[0][0] + arr[1][1] + arr[2][2]

b = arr[0][2] + arr[1][1] + arr[2][0]

로 각각의 합을 구하고 a-b로 차이를 구합니다. abs함수로 절댓값을 구했습니다.