[python] 프로그래머스 - 빛의 경로 사이클(월간 코드 챌린지 시즌3)

 

🤔문제 해결

💨 문제에서 시키는대로 그대로 풀었다. (구현)

 

💫 visited 라는 3차원 배열을 만든다. 제일 안쪽의 원소에는 'U','D','R','L' 이 들어갈 수 있는데 해당 위치(i, j) 에서 이동한 방향을 담고 있다. 싸이클이 다르면 절대로 이동경로가 하나라도 똑같은게 나올 수 없기 때문에. 이동할 때 UDRL 이 있는지 확인을 하고 이동한다.

 

💻소스 코드

def solution(grid):
    answer = []
    N, M = len(grid), len(grid[0])
    visited = [[set() for _ in range(M)] for _ in range(N)]

    # U, D, L, R
    for i in range(N):
        for j in range(M):
            for k in 'UDLR':
                cnt = 0
                while True:
                    if k in visited[i][j]:  # 이미 이동했던 적이 있는지 확인
                        break
                    visited[i][j].add(k)  # 없다면 싸이클 시작
                    cnt += 1
                    
                    # 이동
                    if k == 'U':
                        i -= 1
                        if i < 0:
                            i = N - 1
                    elif k == 'D':
                        i += 1
                        if i == N:
                            i = 0
                    elif k == 'L':
                        j -= 1
                        if j < 0:
                            j = M - 1
                    elif k == 'R':
                        j += 1
                        if j == M:
                            j = 0
                    
                    # 다음 방향 체크
                    if grid[i][j] == 'L':
                        if k == 'U':
                            k = 'L'
                        elif k == 'D':
                            k = 'R'
                        elif k == 'L':
                            k = 'D'
                        elif k == 'R':
                            k = 'U'
                    elif grid[i][j] == 'R':
                        if k == 'U':
                            k = 'R'
                        elif k == 'D':
                            k = 'L'
                        elif k == 'L':
                            k = 'U'
                        elif k == 'R':
                            k = 'D'
                    
                if cnt > 0:
                    answer.append(cnt)

    return sorted(answer)

 

📕문제 확인

출처: 프로그래머스

링크: 빛의 경로 싸이클