본문 바로가기
Coding Test/Programmers

[Programmers] (Summer/Winter Coding(~2018)) Lv 2. 방문 길이

by song.ift 2023. 7. 10.

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

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

 

#include <string>
#include <unordered_map>
#include <set>
using namespace std;

#define POS 5

bool move(int& x, int& y, int moveX, int moveY)
{
    if (x + moveX < -POS || x + moveX > POS 
        || y + moveY < -POS || y + moveY > POS)
        return false;

    x += moveX;
    y += moveY;
    return true;
}

int solution(string dirs) {
    int answer = 0, x = 0, y = 0;
    enum { U, D, R, L };
    int dirX[] = { 0, 0, 1, -1 },
        dirY[] = { 1, -1, 0, 0 };
    unordered_map<int, set<int>> m;

    for (size_t i = 0; i < dirs.length(); ++i)
    {
        int dirIdx = -1, prevIdx = -1;

        if (dirs[i] == 'U') dirIdx = U, prevIdx = D;
        else if (dirs[i] == 'D') dirIdx = D, prevIdx = U;
        else if (dirs[i] == 'R') dirIdx = R, prevIdx = L;
        else if (dirs[i] == 'L') dirIdx = L, prevIdx = R;

        const int prevID = x * 100 + y; // 자신만의 이전 인덱스 고유 ID
        if (move(x, y, dirX[dirIdx], dirY[dirIdx]))
        {
            const int newID = x * 100 + y; // 자신만의 새로운 인덱스 고유 ID
            if (m.find(prevID) != m.end()
                && m[prevID].find(dirIdx) != m[prevID].end())
                continue;

            m[prevID].insert(dirIdx); // prev -> new 방향
            m[newID].insert(prevIdx); // new -> prev 방향
            ++answer;
        }
    }
    return answer;
}

GitHub : https://github.com/developeSHG/Algorithm-Baekjoon_Programmers/blob/deccad56a27b8dfe2223f8bf88c1d6a77bd93c48/%ED%94%84%EB%A1%9C%EA%B7%B8%EB%9E%98%EB%A8%B8%EC%8A%A4/lv2/49994.%E2%80%85%EB%B0%A9%EB%AC%B8%E2%80%85%EA%B8%B8%EC%9D%B4/%EB%B0%A9%EB%AC%B8%E2%80%85%EA%B8%B8%EC%9D%B4.cpp

 

댓글