차세대 영농인 한나는 강원도 고랭지에서 유기농 배추를 재배하기로 하였다. 농약을 쓰지 않고 배추를 재배하려면 배추를 해충으로부터 보호하는 것이 중요하기 때문에, 한나는 해충 방지에 효과적인 배추흰지렁이를 구입하기로 결심한다. 이 지렁이는 배추근처에 서식하며 해충을 잡아 먹음으로써 배추를 보호한다. 특히, 어떤 배추에 배추흰지렁이가 한 마리라도 살고 있으면 이 지렁이는 인접한 다른 배추로 이동할 수 있어, 그 배추들 역시 해충으로부터 보호받을 수 있다. 한 배추의 상하좌우 네 방향에 다른 배추가 위치한 경우에 서로 인접해있는 것이다.
한나가 배추를 재배하는 땅은 고르지 못해서 배추를 군데군데 심어 놓았다. 배추들이 모여있는 곳에는 배추흰지렁이가 한 마리만 있으면 되므로 서로 인접해있는 배추들이 몇 군데에 퍼져있는지 조사하면 총 몇 마리의 지렁이가 필요한지 알 수 있다. 예를 들어 배추밭이 아래와 같이 구성되어 있으면 최소 5마리의 배추흰지렁이가 필요하다. 0은 배추가 심어져 있지 않은 땅이고, 1은 배추가 심어져 있는 땅을 나타낸다.
입력
입력의 첫 줄에는 테스트 케이스의 개수 T가 주어진다. 그 다음 줄부터 각각의 테스트 케이스에 대해 첫째 줄에는 배추를 심은 배추밭의 가로길이 M(1 ≤ M ≤ 50)과 세로길이 N(1 ≤ N ≤ 50), 그리고 배추가 심어져 있는 위치의 개수 K(1 ≤ K ≤ 2500)이 주어진다. 그 다음 K줄에는 배추의 위치 X(0 ≤ X ≤ M-1), Y(0 ≤ Y ≤ N-1)가 주어진다. 두 배추의 위치가 같은 경우는 없다.
출력
각 테스트 케이스에 대해 필요한 최소의 배추흰지렁이 마리 수를 출력한다.
간단한 bfs 풀이로 접근하였는데, 메모리초과 오류가 테스트케이스에서 지속적으로 발생하여 해당 원인을 조사하였다.
문제는 bfs로 탐색한 상하좌우의 grid 원소를 pop
할 시에 visited=1
처리 한 것.
위 그림과 같이pop
할시에 방문 처리를 할 경우 중복해서 queue 에 push
되는 원소가 존재하게 된다. (2,1)의 경우처럼 queue에 삽입 되었지만 탐색 순서의 후순위로 밀려 중복되는 경우가 발생하기 때문.
따라서 queue 에 push
할 경우 visited
처리를 해 주는 것으로 해결하였다.
#include <bits/stdc++.h>
using namespace std;
int N, M, T, K; //M 가로 N 세로, K 배추개수, T 테케 개수
int dx[4] = { 1,0,-1,0 };
int dy[4] = { 0,1,0,-1 };
bool inRange(int _x, int _y) {
return _x > -1 && _x<M&& _y > -1 && _y < N;
}
bool CanGo(int _x, int _y, vector<vector<int>>&_grid, vector<vector<int>>& _visited) {
return inRange(_x, _y) && _grid[_x][_y] && !_visited[_x][_y];
}
int solution(vector<vector<int>>&_v) {
int answer = 0;
vector<vector<int>> visited(M,vector<int>(N));
queue<pair<int, int>> q;
for (int i = 0; i < M; i++) {
for (int j = 0; j < N; j++) {
if (_v[i][j] == 1 && !visited[i][j]) {
q.push({ i,j });
visited[i][j] = 1;
while (!q.empty()) {
int n_y = q.front().second;
int n_x = q.front().first;
q.pop();
for (int k = 0; k < 4; k++) {
int next_x = n_x + dx[k];
int next_y = n_y + dy[k];
if (CanGo(next_x, next_y, _v, visited)) {
q.push({ next_x, next_y }); visited[next_x][next_y] = 1;
}
}
}
answer++;
}
}
}
return answer;
}
int main() {
cin >> T;
while (T--) {
//입력
cin >> M >> N >> K;
vector<vector<int>> v(M,vector<int>(N));
for (int i = 0; i < K; i++) {
int x, y;
cin >> x >> y;
v[x][y] = 1;
}
cout << solution(v) << "\n";
}
return 0;
}