[Level.0] 특별한 이차원 배열(2)

문제 설명: n × n 크기의 이차원 배열 arr이 매개변수로 주어질 때, arr이 다음을 만족하면 1을 아니라면 0을 return 하는 solution 함수를 작성해 주세요.

• 0 ≤ i, j < n인 정수 i, j에 대하여 arr[i][j] = arr[j][i]

제한사항

• 1 ≤ arr의 길이 = arr의 원소의 길이 ≤ 100
• 1 ≤ arr의 원소의 원소 ≤ 1,000
• 모든 arr의 원소의 길이는 같습니다.

문제 해결

def solution(arr):
    for i in range(len(arr)):
        for j in range(i):
            if arr[i][j] != arr[j][i]:
                return 0
    return 1

 

주석: j를 i까지만 돌리면 불필요한 중복 검사(i=1, j=2와 i=2, j=1)를 줄일 수 있습니다.