백준 13901번 로봇 C++

https://www.acmicpc.net/problem/13901

13901번: 로봇

 

실버3 문제치고는 어려운 것 같은 구현 문제

 

#include <iostream>
#include <string>
#include <vector>
#include <math.h>
#include <algorithm>
#include <utility>
#include <stack>
#include <queue>
#include <math.h>
#include <set>
#include <map>
#include <list>
#include <unordered_map>
#include <unordered_set>
#include <iomanip>
#include <limits.h>

using namespace std;
using int64 = long long; 

int dRow[4] = { -1,1,0,0 };
int dCol[4] = { 0,0,-1,1 };

int r, c, k;
int dirIdx = 0;
vector<int> dirs(4);
vector<vector<int>> board;

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);

    cin >> r >> c >> k;
    board = vector<vector<int>>(r, vector<int>(c, 0));
    for (int i = 0; i < k; i++) {
        int br, bc;
        cin >> br >> bc;
        board[br][bc] = -1;
    }
    int sr, sc;
    cin >> sr >> sc;
    board[sr][sc] = 1;
    for (int i = 0; i < 4; i++) {
        int dir;
        cin >> dir;
        dirs[i] = dir - 1;
    }
    while (true) {
        int currDir = dirs[dirIdx];
        bool canGo = false;
        for (int i = 0; i < 4; i++) {
            int nextIdx = (dirIdx + i) % 4;
            int nextDir = dirs[nextIdx];
            int nextRow = sr + dRow[nextDir];
            int nextCol = sc + dCol[nextDir];
            if (nextRow < 0 || nextRow >= r) continue;
            if (nextCol < 0 || nextCol >= c) continue;
            if (board[nextRow][nextCol] != 0) continue;
            board[nextRow][nextCol] = 1;
            dirIdx = nextIdx;
            sr = nextRow;
            sc = nextCol;
            canGo = true;
            break;
        }
        if (canGo == false) break;
    }
    cout << sr << " " << sc;
}