BOJ

[백준] 3190번 뱀 - C++

Queue, 구현

Posted by dongjune on November 6, 2020

문제

3190번 뱀

풀이

뱀 게임을 구현하는 문제이다.

  • 무한루프로 t를 1씩 증가시키며 벽에 부딪힐 경우 break 해줬다.
  • 꼬리와 머리의 위치와 방향을 따로 저장했다.
1
2
3
4
5
6
7
struct Pos // 위치 정보 자료형
{
    int y, x;
};

int head_dir = 2, tail_dir = 2;   // 초기 방향 오른쪽
Pos head = {1, 1}, tail = {1, 1}; // 머리와 꼬리의 위치 정보  
  • direction 이라는 큐에 방향 전환 정보를 저장하여 큐의 top 과 t의 시간이 같은 경우 방향을 전환 했다.
1
2
3
4
5
6
7
8
9
10
11
// 머리 방향 전환
if (!direction.empty() && t == direction.front().t)
{
    if (direction.front().d == 'L') // 왼쪽
        head_dir = turnLeft(head_dir);
    else // 오른쪽
        head_dir = turnRight(head_dir);
    joint.push({head.y, head.x, head_dir});

    direction.pop();
}
  • 뱀이 꺾이는 관절 부분의 정보 역시 큐로 저장하였다. 관절 부분의 위치와 그 위치에서 머리의 방향 정보를 저장했다. 꼬리의 위치가 큐의 top에 도달하면 꼬리의 방향 전환을 해주었다.
1
2
3
4
5
6
7
8
9
10
11
struct Joint // 뱀의 꺾인 부분 자료형
{
    int y, x, d;
};

// 꼬리가 뱀의 꺾인 부분에 도달하면 -> 꼬리 방향 전환
if (!joint.empty() && tail.y == joint.front().y && tail.x == joint.front().x)
{
    tail_dir = joint.front().d;
    joint.pop();
}

소스 코드의 주석을 통해 자세히 살펴 보자.

소스 코드

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
#include <iostream>
#include <vector>
#include <queue>

using namespace std;

/* 변수 선언 */
struct Pos // 위치 정보 자료형
{
    int y, x;
};
struct Dir // 방향 전환 정보 자료형
{
    int t;
    char d;
};
struct Joint // 뱀의 꺾인 부분 자료형
{
    int y, x, d;
};
int board[101][101]; // 2차원 보드 배열
int n, k, l;
// 방향 : 0 왼쪽, 1 위 ,2 오른쪽, 3 아래
int dy[4] = {0, -1, 0, 1};
int dx[4] = {-1, 0, 1, 0};
queue<Dir> direction;

// 입력
void input()
{
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);

    cin >> n >> k;
    int y, x;
    while (k--)
    {
        cin >> y >> x;
        board[y][x] = 2;
    }
    cin >> l;
    int t;
    char d;
    while (l--)
    {
        cin >> t >> d;
        direction.push({t, d});
    }
}

// 오른쪽 회전
int turnRight(int d)
{
    d += 1;
    d = d >= 4 ? d - 4 : d;
    return d;
}

// 왼쪽 회전
int turnLeft(int d)
{
    d -= 1;
    d = d < 0 ? d + 4 : d;
    return d;
}

// 벽이나 자신에게 부딪힌 경우 true 반환
bool isEnd(int y, int x)
{
    // 벽에 부딪힌 경우
    if (y > n || y <= 0 || x > n || x <= 0)
    {
        return true;
    }
    // 자신과 부딪힌 경우
    if (board[y][x] == 1)
    {
        return true;
    }
    return false;
}

// 결과 값 출력
void solve()
{
    int t = 0;                        // 결과 값
    int head_dir = 2, tail_dir = 2;   // 초기 방향 오른쪽
    Pos head = {1, 1}, tail = {1, 1}; // 머리와 꼬리의 위치 정보
    queue<Joint> joint;               // 꺾인 부분 저장하는 큐
    board[1][1] = 1;                  // 왼쪽 위 시작
    while (true)
    {
        t++;
        // 머리의 이동
        head.y += dy[head_dir];
        head.x += dx[head_dir];

        // 부딪힌 경우 break;
        if (isEnd(head.y, head.x))
            break;

        // 머리가 간 곳에 사과가 없으면 꼬리 한 칸 땡기기
        if (board[head.y][head.x] != 2)
        {
            board[tail.y][tail.x] = 0;
            tail.y += dy[tail_dir];
            tail.x += dx[tail_dir];
        }
        board[head.y][head.x] = 1; // 머리가 이동 한 곳 1로 치환

        // 머리 방향 전환
        if (!direction.empty() && t == direction.front().t)
        {
            if (direction.front().d == 'L') // 왼쪽
                head_dir = turnLeft(head_dir);
            else // 오른쪽
                head_dir = turnRight(head_dir);
            joint.push({head.y, head.x, head_dir});

            direction.pop();
        }
        // 꼬리가 뱀의 꺾인 부분에 도달하면 -> 꼬리 방향 전환
        if (!joint.empty() && tail.y == joint.front().y && tail.x == joint.front().x)
        {
            tail_dir = joint.front().d;
            joint.pop();
        }
    }
    cout << t << '\n';
}

int main()
{
    input();
    solve();

    return 0;
}