Python HackerRank : Diagonal Difference

문제

• N X N matrixt
• 1차 대각행렬의 합 : A
• 2차 대각행렬의 합 : B
• |A-B| 반환   

# version1
def diagonalDifference(arr):
    L = 0
    R = 0
    for i in range(len(arr)):
        L += arr[i][i]
        R += arr[i][len(arr)-i-1]
    return abs(L-R)


# version2
def diagonalDifference(arr):
    #left
    L=[0]*len(arr)
    for i in range(len(arr)):
        L[i] = arr[i][i]
    diag1=sum(L)
    
    #Right
    R=[0]*len(arr)
    for i in range(len(arr)):
        R[i] = arr[i][len(arr)-1-i]
    diag2=sum(R)

    return abs(diag1 - diag2)

 

Tip

# 왼쪽에서 시작하는 대각선 인덱스 확인 
a[0][0], a[1][1],a[2][2]

# 오른쪽에서 시작하는 대각선 인덱스 확인 
a[0][2], a[1][1],a[2][0]

 

참고 : https://www.hackerrank.com/challenges/diagonal-difference/problem