[프로그래머스] 실습용 로봇 (JAVA 풀이)

 

문제 접근

 

https://school.programmers.co.kr/learn/courses/15009/lessons/121687

프로그래머스

 

난이도가 그렇게 어렵지 않은 구현 문제이다.

 

방향벡터를 구현하고 로봇의 회전에 따라 어떻게 방향을 지정해줄지 고민하고, 좌표계가 일반적인 배열에서의 구현 문제 처럼 맨 왼쪽 위 위치(인덱스)가 0, 0으로 정해져있는 것이 아니라  인덱스로 일반적인 x, y 좌표계에서 중앙인 0,0 에서 움직이는 것임을 염두해야한다.

 

=> 즉, 보통 배열에서 위쪽으로의 이동은 int[y][x]로 생각하고 y를 -1 해주는데 이 문제는 x, y 좌표계 이므로 위쪽으로 이동한다고하면 정말 (x,  y) 좌표에서 y값을 +1 해줘서 (0, 1)로 생각해야 하는 것임

 

문제 풀이 

 

우선 방향벡터를 어떤순으로 적어놓을지 고민해야함

 

기본적으로 위쪽을 바라보고 있는 상태에서 시작해서

 

오른쪽 회전이라면 위쪽 -> 오른쪽 -> 아래쪽 -> 왼쪽 -> 위쪽 ... 순이 되고

왼쪽 회전이라면 위쪽 -> 왼쪽 -> 아래쪽 -> 오른쪽 -> 위쪽 .. 순이 된다.

 

그래서 위쪽, 오른쪽, 아래쪽, 왼쪽의 방향을 가리키는 방향벡터를 만들어주고

오른쪽 회전이라면 그대로 방향 벡터를 +1씩 증가시키면 오른쪽 회전에 따른 방향을 정할 수 있고

왼쪽 회전이라면 방향 벡터를 -1씩 감소시키면 왼쪽 회전에 따른 방향을 정할 수 있게된다.

=> 이 때, 앞서 말한대로 x, y좌표계의 (0, 0) 에서 시작하는 것이므로 위쪽 방향은 정말 +1을 아래쪽 방향은 -1을 해줘야한다.

 

그럼 시작 좌표 0, 0 을 start 배열에 저장해두고 현재 방향을 나타내는 direction을 위쪽으로 세팅해주고 코드 작성을 시작하면된다.

명령어를 for문으로 돌면서 만약 R이라면 그대로 방향 벡터의 +1 증가방향을 더해주면서 방향을 세팅해주고 L이라면 방향 벡터의 -1 감소 방향으로 빼주면서 방향을 세팅해준다.

 

이 때 만약 방향 벡터의 3번 인덱스에서 1을 증가시키면 indexOutOfBounds 에러가 발생한다. 마찬가지로 0번 인덱스에서 1을 감소해도 똑같다.

 

=> 이를 방지하기 위해 방향을 증가시키는 오른쪽 회전일 경우 4로 모듈러 연산을 해주었고 방향을 감소시키는 왼쪽 회전일 경우 -1을 해준 상태에서 4를 더하고 4로 모듈러 연산을 해서 0번인덱스에서 -1을 감소하면 3이 되게끔 구현하였다.

 

이제 남은 경우인 G, B의 경우는 G는 현재 로봇의 방향이 저장된 direction 방향 그대로 이동하면 되므로 방향 벡터의 direction 인덱스에 있는 값을 더해주고 B는 그 반대 방향으로 후진을 하므로 방향 벡터의 direction 인덱스에 있는 값을 빼준다.

 

전체 코드

 

class Solution {
    
    // 위쪽, 오른쪽, 아래쪽, 왼쪽 방향벡터
    // R이면 현재 방향(인덱스)를 + 1
    // L이면 현재 방향(인덱스)를 - 1
    
    int[] dx = {1, 0, -1, 0};
    int[] dy = {0, 1, 0, -1};
    
    public int[] solution(String command) {
        
        int[] start = {0, 0};
        int direction = 0; // 위쪽 바라보고 시작
        
        for(char c : command.toCharArray()) {
            
            if(c == 'R') {
                direction = (direction + 1) % 4;    
            } else if(c == 'L') {
                direction = ((direction - 1) + 4) % 4;
            } else if(c == 'G') {
                start[0] += dx[direction];
                start[1] += dy[direction];
            } else { // 후진의 경우 현재 방향의 반대로 움직임
                start[0] -= dx[direction];
                start[1] -= dy[direction];
            }
        }
            
        return start;
    }
}