[c++] 알고리즘 문제 해결 전략 1권 :: 외발 뛰기 (문제 ID: JUMPGAME, 난이도 : 하)

https://www.algospot.com/judge/problem/read/JUMPGAME

문제

 땅따먹기를 하다 질린 재하와 영훈이는 땅따먹기의 변종인 새로운 게임을 하기로 했습니다. 이 게임은 그림과 같이 n*n 크기의 격자에 각 1부터 9 사이의 정수를 쓴 상태로 시작합니다. 각 차례인 사람은 맨 왼쪽 윗 칸에서 시작해 외발로 뛰어서 오른쪽 아래 칸으로 내려가야 합니다. 이 때 각 칸에 적혀 있는 숫자만큼 오른쪽이나 아래 칸으로 움직일 수 있으며, 중간에 게임판 밖으로 벗어나면 안 됩니다.

 균형을 잃어서 다른 발로 서거나 넘어져도 게임에서 집니다만, 재하와 영훈이는 젊고 활기차기 때문에 외발로 뛰어다니는 것은 아무것도 아닙니다. 다만 걱정되는 것은 주어진 게임판에 시작점에서 끝점으로 가는 방법이 존재하지 않을 수도 있다는 것입니다. 예를 들어 그림 (a)의 게임판에서는 사각형으로 표시된 칸들을 통해 끝에 도달할 수 있지만, 숫자가 하나 바뀐 그림 (b)에서는 그럴 수가 없습니다.

 게임판이 주어질 때 왼쪽 위의 시작점에서 오른쪽 아래의 시작점에 도달할 수 있는 방법이 있는지 확인하는 프로그램을 작성하세요.

입력

 입력의 첫 줄에는 테스트 케이스의 수 C(C <= 50)가 주어집니다. 각 테스트 케이스의 첫 줄에는 격자의 크기 n(2 <= n <= 100)이 주어지고, 그 후 n줄에 각 n개의 숫자로 왼쪽 위부터 각 칸에 쓰인 숫자들이 주어집니다. 오른쪽 아래 있는 끝 점 위치에는 0이 주어집니다.

출력

 각 테스트 케이스마다 한 줄로, 시작점에서 끝 점으로 도달할 수 있을 경우 "YES"를, 아닐 경우 "NO"를 출력합니다. (따옴표 제외)

나의 답

#include <iostream>
#include <algorithm>
#include <queue>
#include <vector>


using namespace std;

int main() {
    int test_case;
    cin >> test_case;
    while (test_case--) {
        int n;
        cin >> n;
        int** arr = (int**)malloc(n * sizeof(int*));
        for (int i = 0; i < n; i++) {
            arr[i] = (int*)malloc(n * sizeof(int));
        }

        for (int i = 0; i < n; i++) {
            for (int j = 0; j < n; j++) {
                cin >> arr[i][j];
            }
        }

        queue<pair<int, int>> num_q;
        num_q.push(make_pair(0, 0));

        vector<vector<int>> check(n, vector<int>(n));

        while (num_q.size()) {
            int y = num_q.front().first; int x = num_q.front().second;
            num_q.pop();
            check[y][x] = 1;
            if (y == (n - 1) && x == (n - 1))
                break;

            int jump = arr[y][x];

            if (y + jump < n && !check[y + jump][x]) // 아래쪽
                num_q.push(make_pair(y + jump, x));
            if (x + jump < n && !check[y][x + jump]) // 오른쪽
                num_q.push(make_pair(y, x + jump));
        }
        if (check[n - 1][n - 1])
            cout << "YES";
        else
            cout << "NO";
    }
}

 일단 손으로 알고리즘을 짤 때, Queue를 이용한 bfs를 활용해야겠다는 생각을 했고 queue에 들어갈 수 있는 위치의 조건은 맵 안이고, 아직 들르지 않은 곳이었다. 또한 기저사례는 queue가 비거나 끝에 다다른 경우이다.

 

 위에서 구현 패턴을 생각해봤듯이 기저사례인 queue가 비거나 끝에 다다른 경우는 앞부분에 배치하여 오류가 없도록 하였고, memset은 사용할 생각을 못하였는데 들른 곳을 체크하는 check의 자료구조를 2차원 vector로 정의하여 자동 초기화하기 쉬워서 괜찮았다.

재귀 호출에서 시작하기

 책에서는 재귀 호출을 구현한 후, 메모이제이션을 통해 중복되는 경우를 제거하였다. 따라서 bfs 방식을 쓴 본인과는 조금의 차이가 있었다.