[프로그래머스] 공원 산책 c++

문제 설명

문제 바로가기

지나다니는 길을 'O', 장애물을 'X'로 나타낸 직사각형 격자 모양의 공원에서 로봇 강아지가 산책을 하려합니다. 산책은 로봇 강아지에 미리 입력된 명령에 따라 진행하며, 명령은 다음과 같은 형식으로 주어집니다.

• ["방향 거리", "방향 거리" … ]

예를 들어 "E 5"는 로봇 강아지가 현재 위치에서 동쪽으로 5칸 이동했다는 의미입니다. 로봇 강아지는 명령을 수행하기 전에 다음 두 가지를 먼저 확인합니다.

• 주어진 방향으로 이동할 때 공원을 벗어나는지 확인합니다.
• 주어진 방향으로 이동 중 장애물을 만나는지 확인합니다.

위 두 가지중 어느 하나라도 해당된다면, 로봇 강아지는 해당 명령을 무시하고 다음 명령을 수행합니다.
공원의 가로 길이가 W, 세로 길이가 H라고 할 때, 공원의 좌측 상단의 좌표는 (0, 0), 우측 하단의 좌표는 (H - 1, W - 1) 입니다.

제한 사항

• 3 ≤ park의 길이 ≤ 50
  • 3 ≤ park[i]의 길이 ≤ 50
    • park[i]는 다음 문자들로 이루어져 있으며 시작지점은 하나만 주어집니다.
      • S : 시작 지점
      • O : 이동 가능한 통로
      • X : 장애물
  • park는 직사각형 모양입니다.
• 1 ≤ routes의 길이 ≤ 50
  • routes의 각 원소는 로봇 강아지가 수행할 명령어를 나타냅니다.
  • 로봇 강아지는 routes의 첫 번째 원소부터 순서대로 명령을 수행합니다.
  • routes의 원소는 "op n"과 같은 구조로 이루어져 있으며, op는 이동할 방향, n은 이동할 칸의 수를 의미합니다.
    • op는 다음 네 가지중 하나로 이루어져 있습니다.
      • N : 북쪽으로 주어진 칸만큼 이동합니다.
      • S : 남쪽으로 주어진 칸만큼 이동합니다.
      • W : 서쪽으로 주어진 칸만큼 이동합니다.
      • E : 동쪽으로 주어진 칸만큼 이동합니다.
    • 1 ≤ n ≤ 9

입출력 예

park	routes	result
["SOO","OOO","OOO"]	["E 2","S 2","W 1"]	[2,1]
["SOO","OXX","OOO"]	["E 2","S 2","W 1"]	[0,1]
["OSO","OOO","OXO","OOO"]	["E 2","S 3","W 1"]	[0,0]

풀이

이동한 좌표의 값이 'X" 인지만 확인하는 문제인 줄 알았으나 이동하는 좌표의 경로에 'X'도 검사해야하는 문제였다.

생각보다 간단하지 않아서 어떻게 풀어나갈지 고민하게 되었다.

참고로, 코딩할때는 (0.0)부터 시작하는 좌표들이고, x y가 수학적 좌표랑 반대로 표현되어 다르게 생각해야 한다.

 

이동 경로에 'X'가 있다면 이동하지 않고 다음 경로를 적용해야하기 때문에 경로 값을 검사하는 함수를 추가하였다. 

경로가 항상 값이 증가되는 형태로 이동되지 않고 W, N 방향은 값이 감소되어하기 때문에 while문을 통해 방향에 따른 값을 변경시키는 함수를 만들었습니다.

 

#include <string>
#include <vector>
#include <map>

using namespace std;

int dist[6] = {1, 0, -1, 0, 1};
map<char, int> m = { {'S',0 }, {'W', 1},{'N',2},{'E', 3} };

bool checkMap(int startx, int endx, int starty, int endy, int index,const vector<string>& park) {
    
	while( (startx != endx) || (starty != endy) ){
        startx += dist[index];
        starty += dist[index+1];
        if(park[startx][starty] == 'X'){
            return false;
        }
    }
	return true;
}

vector<int> solution(vector<string> park, vector<string> routes) {
    
    int x = -1, y = -1;
    for(int i = 0; i < park.size(); i++){
        if(x != -1){
            break;
        }
        for(int j = 0; j < park[0].size(); j++){
            if(y != -1){
                break;
            }
            else if(park[i][j] == 'S'){
                x = i;
                y = j;
                break;
            }
        }
    }    
    
    for(const auto& it : routes){
        int row = dist[m[it[0]]] * (it[2] - '0');
        int col = dist[m[it[0]]+1] * (it[2] - '0');
        
        if( x + row >= park.size() || y + col >= park[0].size() || x + row < 0 || y + col < 0){
            continue;
        }
        if(park[x+row][y+col] == 'X'){
            continue;
        }
        if (!checkMap(x, x + row, y, y + col, m[it[0]], park)){
		    continue;
	   }    
        x += row;
        y += col;
    }
    
    return { x, y };
}