문제 설명
플레이어 A와 플레이어 B가 서로 게임을 합니다. 당신은 이 게임이 끝날 때까지 양 플레이어가 캐릭터를 몇 번 움직이게 될지 예측하려고 합니다.
각 플레이어는 자신의 캐릭터 하나를 보드 위에 올려놓고 게임을 시작합니다. 게임 보드는 1x1 크기 정사각 격자로 이루어져 있으며, 보드 안에는 발판이 있는 부분과 없는 부분이 있습니다. 발판이 있는 곳에만 캐릭터가 서있을 수 있으며, 처음 캐릭터를 올려놓는 곳은 항상 발판이 있는 곳입니다. 캐릭터는 발판이 있는 곳으로만 이동할 수 있으며, 보드 밖으로 이동할 수 없습니다. 밟고 있던 발판은 그 위에 있던 캐릭터가 다른 곳으로 이동하여 다른 발판을 밞음과 동시에 사라집니다. 양 플레이어는 번갈아가며 자기 차례에 자신의 캐릭터를 상하좌우로 인접한 4개의 칸 중에서 발판이 있는 칸으로 옮겨야 합니다.
다음과 같은 2가지 상황에서 패자와 승자가 정해지며, 게임이 종료됩니다.
- 움직일 차례인데 캐릭터의 상하좌우 주변 4칸이 모두 발판이 없거나 보드 밖이라서 이동할 수 없는 경우, 해당 차례 플레이어는 패배합니다.
- 두 캐릭터가 같은 발판 위에 있을 때, 상대 플레이어의 캐릭터가 다른 발판으로 이동하여 자신의 캐릭터가 서있던 발판이 사라지게 되면 패배합니다.
게임은 항상 플레이어 A가 먼저 시작합니다. 양 플레이어는 최적의 플레이를 합니다. 즉, 이길 수 있는 플레이어는 최대한 빨리 승리하도록 플레이하고, 질 수밖에 없는 플레이어는 최대한 오래 버티도록 플레이합니다. '이길 수 있는 플레이어'는 실수만 하지 않는다면 항상 이기는 플레이어를 의미하며, '질 수밖에 없는 플레이어'는 최선을 다해도 상대가 실수하지 않으면 항상 질 수밖에 없는 플레이어를 의미합니다. 최대한 오래 버틴다는 것은 양 플레이어가 캐릭터를 움직이는 횟수를 최대화한다는 것을 의미합니다.
아래 그림은 초기 보드의 상태와 각 플레이어의 위치를 나타내는 예시입니다.
위와 같은 경우, 플레이어 A는 실수만 하지 않는다면 항상 이길 수 있습니다. 따라서 플레이어 A는 이길 수 있는 플레이어이며, B는 질 수밖에 없는 플레이어입니다. 다음은 A와 B가 최적의 플레이를 하는 과정을 나타냅니다.
- 플레이어 A가 초기 위치 (1, 0)에서 (1, 1)로 이동합니다. 플레이어 A가 (0, 0)이나 (2, 0)으로 이동할 경우 승리를 보장할 수 없습니다. 따라서 무조건 이길 방법이 있는 (1, 1)로 이동합니다.
- 플레이어 B는 (1, 1)로 이동할 경우, 바로 다음 차례에 A가 위 또는 아래 방향으로 이동하면 발판이 없어져 패배하게 됩니다. 질 수밖에 없는 플레이어는 최대한 오래 버티도록 플레이하기 때문에 (1, 1)로 이동하지 않습니다. (1, 2)에서 위쪽 칸인 (0, 2)로 이동합니다.
- A가 (1, 1)에서 (0, 1)로 이동합니다.
- B에게는 남은 선택지가 (0, 1)밖에 없습니다. 따라서 (0, 2)에서 (0, 1)로 이동합니다.
- A가 (0, 1)에서 (0, 0)으로 이동합니다. 이동을 완료함과 동시에 B가 서있던 (0, 1)의 발판이 사라집니다. B가 패배합니다.
- 만약 과정 2에서 B가 아래쪽 칸인 (2, 2)로 이동하더라도 A는 (2, 1)로 이동하면 됩니다. 이후 B가 (2, 1)로 이동, 다음 차례에 A가 (2, 0)으로 이동하면 B가 패배합니다.
위 예시에서 양 플레이어가 최적의 플레이를 했을 경우, 캐릭터의 이동 횟수 합은 5입니다. 최적의 플레이를 하는 방법은 여러 가지일 수 있으나, 이동한 횟수는 모두 5로 같습니다.
게임 보드의 초기 상태를 나타내는 2차원 정수 배열 board와 플레이어 A의 캐릭터 초기 위치를 나타내는 정수 배열 aloc, 플레이어 B의 캐릭터 초기 위치를 나타내는 정수 배열 bloc이 매개변수로 주어집니다. 양 플레이어가 최적의 플레이를 했을 때, 두 캐릭터가 움직인 횟수의 합을 return 하도록 solution 함수를 완성해주세요.
https://school.programmers.co.kr/learn/courses/30/lessons/92345
제한 사항
- 1 ≤ board의 세로 길이 ≤ 5
- 1 ≤ board의 가로 길이 ≤ 5
- board의 원소는 0 또는 1입니다.
- 0은 발판이 없음을, 1은 발판이 있음을 나타냅니다.
- 게임 보드의 좌측 상단 좌표는 (0, 0), 우측 하단 좌표는 (board의 세로 길이 - 1, board의 가로 길이 - 1)입니다.
- aloc과 bloc은 각각 플레이어 A의 캐릭터와 플레이어 B의 캐릭터 초기 위치를 나타내는 좌표값이며 [r, c] 형태입니다.
- r은 몇 번째 행인지를 나타냅니다.
- 0 ≤ r < board의 세로 길이
- c는 몇 번째 열인지를 나타냅니다.
- 0 ≤ c < board의 가로 길이
- 초기 보드의 aloc과 bloc 위치는 항상 발판이 있는 곳입니다.
- aloc과 bloc이 같을 수 있습니다.
- 상대 플레이어의 캐릭터가 있는 칸으로 이동할 수 있습니다.
풀이
생각했던 것보다 어려웠던 문제이다.
기본적으로는 DFS를 이용하여 A, B가 번갈아가며 이동한다고 생각하면 된다.
DFS가 종료되는 조건은 다음과 같다.
- 움직일 수 있는 발판이 없다 → 패배
- 움직인다면 같이 있던 다른 플레이어가 떨어진다 → 승리
위의 상황이 아니라면 갈 수 있는 4방향을 탐색하여 DFS를 진행하면 된다.
DFS의 함수의 반환값은 [승리가능 여부, 이동 횟수]이다.
반환값의 값들은 승리가능 여부에 따라 달라진다.
우선, 이동할 수 있는 방향 중 하나라도 승리가 가능한 루트가 있다면 승리가능 여부는 true이다.
그리고 승리가능 여부가 true라면 최소 이동 거리를 반환하여 최대한 빨리 승리해야 한다.
이 상황은 문제의 조건인 "최적의 플레이를 합니다" 조건 때문이다.
반대로 단 하나라도 이길 수 있는 루트가 없다면 최대한 오래 끌어 이동 횟수를 늘려야 한다.
int cnt = bWin ? minCnt : maxCnt;
이러한 개념을 바탕으로 A와 B가 서로 번갈아가며 플레이하면 된다.
따라서, 2개의 함수가 필요하다.
하지만 코드는 같기 때문에 매개변수로 누구의 차례인지 나타내는 방법도 가능하다.
하지만 코드가 복잡해질 수 있기 때문에 분리하여 구현하였다.
코드를 작성하다 보면 두 개의 함수를 코딩해야 하기 때문에 승리 여부가 헷갈릴 수 있다
한 가지 팁은 반환값을 받는 변수를 갖는 플레이어가 승리하는지 생각하면 된다.
전체 코드
#include <string>
#include <vector>
#include <iostream>
using namespace std;
vector<int> dy = {1, 0, -1, 0};
vector<int> dx = {0, -1, 0, 1};
pair<bool, int> moveA(vector<vector<int>> board, vector<int>& aloc, vector<int>& bloc);
pair<bool, int> moveB(vector<vector<int>> board, vector<int>& aloc, vector<int>& bloc);
pair<bool, int> moveA(vector<vector<int>> board, vector<int>& aloc, vector<int>& bloc)
{
//승리가능 여부
bool bWin = false;
//이동 여부
bool bMove = false;
int maxCnt = 0;
int minCnt = 999;
for(int i = 0; i < 4; i++)
{
int aY = aloc[0] + dy[i];
int aX = aloc[1] + dx[i];
if(aY < 0 || aY >= board.size() || aX < 0 || aX >= board[0].size()) continue;
if(board[aY][aX] == 0) continue;
if(aloc[0] == bloc[0] && aloc[1] == bloc[1])
{
//A가 움직여서 이김
return {false, 1};
}
bMove = true;
board[aloc[0]][aloc[1]] = 0;
vector<int> tempA = {aY, aX};
auto [winTemp, cnt] = moveB(board, tempA, bloc);
if(winTemp)
{
//A Win
bWin = true;
minCnt = min(cnt+1, minCnt);
}
else
{
maxCnt = max(cnt+1, maxCnt);
}
}
//움직이지 못함 -> 패배
if(!bMove)
{
return {true, 0};
}
//승리가능
if(bWin)
{
return {false, minCnt};
}
//패배
return {true, maxCnt};
}
pair<bool, int> moveB(vector<vector<int>> board, vector<int>& aloc, vector<int>& bloc)
{
//승리가능 여부
bool bWin = false;
//이동 여부
bool bMove = false;
int maxCnt = 0;
int minCnt = 999;
for(int i = 0; i < 4; i++)
{
int bY = bloc[0] + dy[i];
int bX = bloc[1] + dx[i];
if(bY < 0 || bY >= board.size() || bX < 0 || bX >= board[0].size()) continue;
if(board[bY][bX] == 0) continue;
if(aloc[0] == bloc[0] && aloc[1] == bloc[1])
{
//B가 움직여서 이김
return {false, 1};
}
bMove = true;
board[bloc[0]][bloc[1]] = 0;
vector<int> tempB = {bY, bX};
auto [winTemp, cnt] = moveA(board, aloc, tempB);
if(winTemp)
{
//B Win
bWin = true;
minCnt = min(cnt+1, minCnt);
}
else
{
maxCnt = max(cnt+1, maxCnt);
}
}
//움직이지 못함 -> 패배
if(!bMove)
{
return {true, 0};
}
//승리가능
if(bWin)
{
return {false, minCnt};
}
//패배
return {true, maxCnt};
}
int solution(vector<vector<int>> board, vector<int> aloc, vector<int> bloc) {
auto [win, cnt] = moveA(board, aloc, bloc);
return cnt;
}