[Python] 기초 알고리즘 팀 노트 정리

방향 전환

def turn(direction, c):
	# 0 -> 3 
    # 1 -> 0
    # 2 -> 1
    # 3 -> 2
    # 서쪽 방향으로 회전
    if c == "L":
        direction = (direction - 1) % 4
    # 0 -> 1
    # 1 -> 2
    # 2 -> 3
    # 3 -> 0
    # 동쪽 방향으로 회전
    if c == "D":
        direction = (direction + 1) % 4
    return direction
    
 
 turn(0, "L")
 turn(3, "D")

2차원 배열 90도 회전

# a는 2차원 배열이어야 함.
def rotate_a_matrix_by_90_degree(a):
    return list(zip(*a[::-1]))
    
rotate_a_matrix_by_90_degree([[0, 1, 0], [1, 0, 1], [0, 0, 1]])