프로그래머스 - 방문 길이

문제

https://school.programmers.co.kr/learn/courses/30/lessons/49994

프로그래머스

이동 경로 중 처음 방문한 간선의 개수를 구하는 문제였다.

 

문제 풀이

처음에는 구조체를 만들어서 구조체 [11][11] map을 만들고, 각 LRUD마다 반대 방향의 RLDU 체킹을 일일이 넣어줬는데 너무 복잡하고 어느 부분이 틀린지도 잘 모르겠어서 다시 풀었다. 다시 풀면서 3차원 배열로 변경하고 nx, ny와 x,y의 LR / RL / UD / DU 계산을 (di+2)%4로 해서 간단하게 풀 수 있었다.

#include <string>
using namespace std;

int visit[11][11][4]; //0123 왼위오아
int x,y;

int dx[4] = {-1,0,1,0};
int dy[4] = {0,1,0,-1};

int solution(string dirs) {
    int answer = 0;
    int di;

    x=y=5;
    
    for(int i=0;i<dirs.size();i++){
        if(dirs[i] == 'L'){
            di = 0;
        }
        else if(dirs[i] == 'U'){
            di = 1;
        }
        else if(dirs[i] == 'R'){
            di = 2;
        }
        else if(dirs[i] == 'D'){
            di = 3;
        }
        
        int nx = x+dx[di];
        int ny = y+dy[di];
        
        if(nx >= 0 && nx < 11 && ny >=0 && ny <11){
            if(visit[nx][ny][((di+2)%4)] == 0 && visit[x][y][di] == 0){
                visit[nx][ny][((di+2)%4)] = 1;
                visit[x][y][di] = 1;
                answer++;
            }
            x = nx;
            y = ny;
        }
    }
    return answer;
}