[C++] 백준(BOJ) - 2636 : 치즈
인프런에 있는 큰돌님의 강의 10주완성 C++ 코딩테스트 | 알고리즘 코딩테스트를 듣고 정리한 필기입니다.
문제
아래 <그림 1>과 같이 정사각형 칸들로 이루어진 사각형 모양의 판이 있고, 그 위에 얇은 치즈(회색으로 표시된 부분)가 놓여 있다. 판의 가장자리(<그림 1>에서 네모 칸에 X친 부분)에는 치즈가 놓여 있지 않으며 치즈에는 하나 이상의 구멍이 있을 수 있다.
이 치즈를 공기 중에 놓으면 녹게 되는데 공기와 접촉된 칸은 한 시간이 지나면 녹아 없어진다. 치즈의 구멍 속에는 공기가 없지만 구멍을 둘러싼 치즈가 녹아서 구멍이 열리면 구멍 속으로 공기가 들어가게 된다. <그림 1>의 경우, 치즈의 구멍을 둘러싼 치즈는 녹지 않고 ‘c’로 표시된 부분만 한 시간 후에 녹아 없어져서 <그림 2>와 같이 된다.
<그림 1> 원래 치즈 모양
다시 한 시간 후에는 <그림 2>에서 ‘c’로 표시된 부분이 녹아 없어져서 <그림 3>과 같이 된다.
<그림 2> 한 시간 후의 치즈 모양
<그림 3> 두 시간 후의 치즈 모양
<그림 3>은 원래 치즈의 두 시간 후 모양을 나타내고 있으며, 남은 조각들은 한 시간이 더 지나면 모두 녹아 없어진다. 그러므로 처음 치즈가 모두 녹아 없어지는 데는 세 시간이 걸린다. <그림 3>과 같이 치즈가 녹는 과정에서 여러 조각으로 나누어 질 수도 있다.
입력으로 사각형 모양의 판의 크기와 한 조각의 치즈가 판 위에 주어졌을 때, 공기 중에서 치즈가 모두 녹아 없어지는 데 걸리는 시간과 모두 녹기 한 시간 전에 남아있는 치즈조각이 놓여 있는 칸의 개수를 구하는 프로그램을 작성하시오.
입력
첫째 줄에는 사각형 모양 판의 세로와 가로의 길이가 양의 정수로 주어진다. 세로와 가로의 길이는 최대 100이다. 판의 각 가로줄의 모양이 윗 줄부터 차례로 둘째 줄부터 마지막 줄까지 주어진다. 치즈가 없는 칸은 0, 치즈가 있는 칸은 1로 주어지며 각 숫자 사이에는 빈칸이 하나씩 있다.
출력
첫째 줄에는 치즈가 모두 녹아서 없어지는 데 걸리는 시간을 출력하고, 둘째 줄에는 모두 녹기 한 시간 전에 남아있는 치즈조각이 놓여 있는 칸의 개수를 출력한다.
예제 입력
13 12
0 0 0 0 0 0 0 0 0 0 0 0
0 0 0 0 0 0 0 0 0 0 0 0
0 0 0 0 0 0 0 1 1 0 0 0
0 1 1 1 0 0 0 1 1 0 0 0
0 1 1 1 1 1 1 0 0 0 0 0
0 1 1 1 1 1 0 1 1 0 0 0
0 1 1 1 1 0 0 1 1 0 0 0
0 0 1 1 0 0 0 1 1 0 0 0
0 0 1 1 1 1 1 1 1 0 0 0
0 0 1 1 1 1 1 1 1 0 0 0
0 0 1 1 1 1 1 1 1 0 0 0
0 0 1 1 1 1 1 1 1 0 0 0
0 0 0 0 0 0 0 0 0 0 0 0
예제 출력
3
5
알고리즘
- 구현
- DFS
풀이
- {0,0} 을 기준으로 DFS를 돌린다. 1인 지점을 찾아낸다
- 1인 지점은 pair형 좌표 벡터에 담는다
- 담아놓은 좌표 벡터 기준으로 0으로 초기화 시킨 후
- 치즈가 모두 녹았는지 체크
코드
#include <bits/stdc++.h>
using namespace std;
int a[104][104], visited[104][104];
int n, m, cnt1, cnt2;
const int dy[] = { -1,0,1,0 };
const int dx[] = { 0, 1, 0 ,-1 };
vector<pair<int, int>> v;
void go(int y, int x)
{
visited[y][x] = 1;
if (a[y][x] == 1) // 2. 1인 지점은 pair형 좌표 벡터에 담는다
{
v.push_back({ y,x });
return;
}
for (int i = 0; i < 4; i++)
{
int ny = y + dy[i];
int nx = x + dx[i];
if (ny < 0 || nx < 0 || ny >= n || nx >= m || visited[ny][nx]) continue;
go(ny, nx);
}
return;
}
int main()
{
ios_base::sync_with_stdio(false);
cin.tie(NULL);
cin >> n >> m;
for (int i = 0; i < n; i++)
{
for (int j = 0; j < m; j++)
{
cin >> a[i][j];
}
}
while (true)
{
fill(&visited[0][0], &visited[0][0] + 104 * 104, 0); // visited 배열 초기화
v.clear(); // 값이 1인 지점을 담을 v 배열 초기화
go(0, 0); // 1. {0,0} 을 기준으로 DFS를 돌린다
cnt2 = v.size();
for (pair<int, int> b : v) // 3. 담아놓은 좌표 벡터 기준으로 0으로 초기화 시킨 후
{
a[b.first][b.second] = 0;
}
// 4. 치즈가 모두 녹았는지 체크
bool flag = 0;
for (int i = 0; i < n; i++)
{
for (int j = 0; j < m; j++)
{
if (a[i][j] != 0) flag = 1;
}
}
cnt1++;
if (!flag) break;
}
cout << cnt1 << "\n" << cnt2 << '\n';
return 0;
}
평가
- 탐색은 어떤 지점에서든 (보통은 {0,0} 기준) 시작을 한 후, 여러 번 반복 해야하는 경우 종료 조건을 둔 while 반복문을 이용하여 원하는 정보를 얻어낸다.
- 탐색을 통해 찾아 낸 지점이 이차원 좌표 상일 경우 pair<int, int> 자료형을 이용하여 벡터에 담아 저장해 놓을 수 있다
- 크기가 크지 않은 이차원 벡터의 경우 flag 불린 값을 적절히 활용하여 전체 탐색을 해나 갈 수 있다.
- visited 배열과 좌표 벡터의 경우 반복문을 여러번 사용할 때 항상 초기화 하여야 함을 명심한다. fill 함수 or memset 함수
