[백준] 3190번 - 뱀

문제

입력

출력

풀이

코드

#include <vector>
#include <algorithm>
#include <iostream>
#include <queue>
#include <map>
using namespace std;

vector<pair<int, int>> dir = { {0, 1}, {1, 0}, {0, -1}, {-1, 0} };

pair<int, int> Destination(const pair<int, int>& a, const pair<int, int>& b) {
	pair<int, int> tmp;
	tmp.first = a.first + b.first;
	tmp.second = a.second + b.second;
	return tmp;
}

void Rotation(char n, int& idx) {
	idx += (n == 'D') ? 1 : -1;
	if (idx > 3) idx = 0;
	else if (idx < 0) idx = 3;
}

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

	int N, K;
	cin >> N >> K;
	vector<vector<int>> board(N, vector<int>(N)); // 0빈곳 1몸 2사과
	board[0][0] = 1;
	for (int i = 0; i < K; ++i) {
		int row, col;
		cin >> row >> col;
		board[row - 1][col - 1] = 2;
	}

	int L;
	cin >> L;
	map<int, char> commands;
	for (int i = 0; i < L; ++i) {
		int time;
		char rot;
		cin >> time >> rot;
		commands[time] = rot;
	}

	int answer = 0;
	int idx = 0;
	deque<pair<int, int>> snake;
	snake.push_back(make_pair(0, 0));
	while (true) {
		answer += 1;
		pair<int, int> head = snake.front();
		
		pair<int, int> dest = Destination(head, dir[idx]);
		

		if (dest.first < 0 || dest.first >= N || dest.second < 0 || dest.second >= N || board[dest.first][dest.second] == 1) {
			cout << answer;
			break;
		}

		snake.push_front(dest); // 머리 추가
		if (board[dest.first][dest.second] == 2) { // 목적지 사과 있으면
			board[dest.first][dest.second] = 1; // 목적지 사과 먹고 몸으로 변경
		}
		else {
			board[dest.first][dest.second] = 1; // 목적지 몸으로 변경
			pair<int, int> tail = snake.back();
			board[tail.first][tail.second] = 0; // 꼬리 맵에서 지우기
			snake.pop_back(); // 꼬리 제거
		}

		if(commands.count(answer)) 
			Rotation(commands[answer], idx); // 카운트 넘어가면 회전
	}
	
	return 0;
}

후기