문제 링크 https://www.acmicpc.net/problem/13565
문제
인제대학교 생화학연구실에 재직중인 석교수는 전류가 침투(percolate) 할 수 있는 섬유 물질을 개발하고 있다. 이 섬유 물질은 2차원 M × N 격자로 표현될 수 있다. 편의상 2차원 격자의 위쪽을 바깥쪽(outer side), 아래쪽을 안쪽(inner side)라고 생각하기로 한다. 또한 각 격자는 검은색 아니면 흰색인데, 검은색은 전류를 차단하는 물질임을 뜻하고 흰색은 전류가 통할 수 있는 물질임을 뜻한다. 전류는 섬유 물질의 가장 바깥쪽 흰색 격자들에 공급되고, 이후에는 상하좌우로 인접한 흰색 격자들로 전달될 수 있다.
김 교수가 개발한 섬유 물질을 나타내는 정보가 2차원 격자 형태로 주어질 때, 바깥쪽에서 흘려 준 전류가 안쪽까지 침투될 수 있는지 아닌지를 판단하는 프로그램을 작성하시오.
입력
첫째 줄에는 격자의 크기를 나타내는 M (2 ≤ M ≤ 1,000) 과 N (2 ≤ N ≤ 1,000) 이 주어진다. M줄에 걸쳐서, N개의 0 또는 1 이 공백 없이 주어진다. 0은 전류가 잘 통하는 흰색, 1은 전류가 통하지 않는 검은색 격자임을 뜻한다.
출력
바깥에서 흘려준 전류가 안쪽까지 잘 전달되면 YES를 출력한다.
그렇지 않으면 NO를 출력한다.
풀이 과정
문제를 잘 읽어야한다. 위쪽을 바깥쪽, 아래쪽을 안쪽이라고 생각한다.
0이면 전류가 잘 통하는 흰색이고, 1이면 전류가 통하지 않는 검은색 격자이다.
01이 붙어서 입력되므로 scanf(“%1d”)를 사용해서 입력 받아준다.
첫번째 줄이고, 전류가 통하는 0일 때 queue에 push를 해주며 방문 표시를 한다.
bfs 과정에서는 전류가 통하지 않는 1일 때와 방문 했을 때를 제외하고는 상하좌우로 방문을 해주며 전류가 통할 때의 자리를 2로 바꿔준다.
bfs가 끝나고 맨 마지막 줄에 전류가 통했다는 뜻인 2가 있다면 YES, 2가 없다면 바깥쪽에서 흘려준 전류가 안쪽까지 전달되지 않았다는 뜻이므로 NO를 출력해준다.
#include <vector>
#include <algorithm>
#include <iostream>
#include <queue>
using namespace std;
int M, N;
int num[1001][1001];
bool c[1001][1001];
int dx[4] = { 1, 0, -1, 0 };
int dy[4] = { 0, -1, 0, 1 };
queue <pair <int, int>> q;
void bfs() {
while (!q.empty()) {
int x = q.front().first;
int y = q.front().second;
q.pop();
for (int i = 0; i < 4; i++) {
int nx = x + dx[i];
int ny = y + dy[i];
if (nx < 1 || nx > M || ny < 1 || ny > N)
continue;
if (num[nx][ny] != 1 && !c[nx][ny]) {
c[nx][ny] = 1;
num[nx][ny] = 2;
q.push({ nx, ny });
}
}
}
}
int main() {
cin >> M >> N;
for (int i = 1; i <= M; i++) {
for (int j = 1; j <= N; j++) {
scanf("%1d", &num[i][j]);
if (i == 1 && !num[i][j]) {
q.push({ i, j });
c[i][j] = 1;
}
}
}
bfs();
bool ans = false;
for (int i = 1; i <= N; i++) {
if (num[M][i] == 2) ans = true;
}
if (ans) cout << "YES";
else cout << "NO";
}
댓글남기기