글
[삼성 SW 역량 테스트] 구슬탈출2
삼성 SW 역량 테스트 기출 풀이
2019. 1. 10. 21:48
#include <stdio.h> #include <queue> #include <algorithm> using namespace std; struct INFO { int ry, rx, by, bx, count; }; INFO start; char map[10][11]; int bfs() { const int dy[] = { -1, 1, 0, 0 }; const int dx[] = { 0, 0, -1, 1 }; int visited[10][10][10][10] = { 0, }; queue<INFO> q; q.push(start); visited[start.ry][start.rx][start.by][start.bx] = 1; int ret = -1; while (!q.empty()) { INFO cur = q.front(); q.pop(); if (cur.count > 10) break; if (map[cur.ry][cur.rx] == 'O' && map[cur.by][cur.bx] != 'O') { ret = cur.count; break; } for (int dir = 0; dir < 4; ++dir) { int next_ry = cur.ry; int next_rx = cur.rx; int next_by = cur.by; int next_bx = cur.bx; while (1) { if (map[next_ry][next_rx] != '#' && map[next_ry][next_rx] != 'O') { next_ry += dy[dir], next_rx += dx[dir]; } else { if (map[next_ry][next_rx] == '#') { next_ry -= dy[dir], next_rx -= dx[dir]; } break; } } while (1) { if (map[next_by][next_bx] != '#' && map[next_by][next_bx] != 'O') { next_by += dy[dir], next_bx += dx[dir]; } else { if (map[next_by][next_bx] == '#') { next_by -= dy[dir], next_bx -= dx[dir]; } break; } } if (next_ry == next_by && next_rx == next_bx) { if (map[next_ry][next_rx] != 'O') { int red_dist = abs(next_ry - cur.ry) + abs(next_rx - cur.rx); int blue_dist = abs(next_by - cur.by) + abs(next_bx - cur.bx); if (red_dist > blue_dist) { next_ry -= dy[dir], next_rx -= dx[dir]; } else { next_by -= dy[dir], next_bx -= dx[dir]; } } } if (visited[next_ry][next_rx][next_by][next_bx] == 0) { visited[next_ry][next_rx][next_by][next_bx] = 1; INFO next; next.ry = next_ry; next.rx = next_rx; next.by = next_by; next.bx = next_bx; next.count = cur.count + 1; q.push(next); } } } return ret; } int main() { int n, m; scanf("%d %d", &n, &m); for (int i = 0; i < n; ++i) { scanf("%s", map[i]); } for (int y = 0; y < n; ++y) { for (int x = 0; x < m; ++x) { if (map[y][x] == 'R') { start.ry = y, start.rx = x; } if (map[y][x] == 'B') { start.by = y, start.bx = x; } } } start.count = 0; int ret = bfs(); printf("%d\n", ret); return 0; }