< 알고리즘 문제 풀이 > 리코쳇 로봇

문제 링크 > 코딩테스트 연습 - 리코쳇 로봇 | 프로그래머스 스쿨 (programmers.co.kr)

프로그래머스

 

솔루션 >

def solution(board):
    answer = 0
    
    visited = []
    row, col = len(board), len(board[0])
    for i in range(row):
        visited.append([0 for _ in range(col)])
        for j in range(col):            
            if board[i][j]=='R':
                rx, ry = i, j
            elif board[i][j]=='G':
                gx, gy = i, j

    # U D L R
    dx = [-1, 1, 0, 0]
    dy = [0, 0, -1, 1]
    
    while True:
        mv = []
        mv.append((rx, ry, 0))
        
        visited[rx][ry] = 1
        
        while len(mv) != 0:
            px, py, count = mv.pop(0)
            
            if (px, py) == (gx, gy):
                visited[rx][ry] = 1
                return count
               
            for i in range(4):
                # 각 방향에 따라 움직인 후 위치 변수 생성
                nx, ny = px, py
                                         
                # 방향의 끝까지 이동
                while True:                    
                    if i == 0: # U
                        if nx+dx[i] >= 0 and board[nx+dx[i]][ny] != 'D':
                            nx, ny = nx+dx[i], ny+dy[i]
                        else:
                            break
                    elif i == 1: # D
                        if nx+dx[i] < row and board[nx+dx[i]][ny] != 'D':
                            nx, ny = nx+dx[i], ny+dy[i]
                        else:
                            break
                    elif i == 2: # L
                        if ny+dy[i] >= 0 and board[nx][ny+dy[i]] != 'D':
                            nx, ny = nx+dx[i], ny+dy[i]
                        else:
                            break
                    elif i == 3: # R
                        if ny+dy[i] < col and board[nx][ny+dy[i]] != 'D':
                            nx, ny = nx+dx[i], ny+dy[i]
                        else:
                            break
                    
                if visited[nx][ny] != 1:
                    visited[nx][ny] = 1
                    mv.append((nx, ny, count + 1))

        return -1