2차원 리스트를 돌리기(rotate_a_matrix_by_90_degree())

나동빈님의 '이것이 코딩테스트다'를 풀다가 2차원 리스트를 시계방향으로 90도 돌리는 함수를 만들어야하는 상황에 처했다.(두둥)

 

책에는 간단하게 식만 나와있었는데 외울 수는 있었지만 앞으로 계속 쓰려면 머리속으로 단단히 이해해두는게 좋을 거라고 생각해 이렇게 정리하는 글을 쓴다. 혹시 나와 같은 상황을 겪는 읽은이가 있다면 도움이 되기를..!

 

결국 코드는 아래와 같다!

#2차원 리스트를 시계방향으로 90도 회전시키는 함수
def rotate_a_matrix_by_90_degree(a):
    n = len(a)#기존 2차원 리스트 행 길이 구하기
    m = len(a[0])#기존 2차원 리스트 열 길이 구하기
    new_list = [[0]*n for _ in range(m)]
    for i in range(n):
        for j in range(m):
            new_list[j][n-1-i] = a[i][j]
    return new_list