1012번 : 유기농 배추
차세대 영농인 한나는 강원도 고랭지에서 유기농 배추를 재배하기로 하였다. 농약을 쓰지 않고 배추를 재배하려면 배추를 해충으로부터 보호하는 것이 중요하기 때문에, 한나는 해충 방지에 효과적인 배추흰지렁이를 구입하기로 결심한다. 이 지렁이는 배추근처에 서식하며 해충을 잡아 먹음으로써 배추를 보호한다. 특히, 어떤 배추에 배추흰지렁이가 한 마리라도 살고 있으면 이 지렁이는 인접한 다른 배추로 이동할 수 있어, 그 배추들 역시 해충으로부터 보호받을 수 있다. 한 배추의 상하좌우 네 방향에 다른 배추가 위치한 경우에 서로 인접해있는 것이다.
한나가 배추를 재배하는 땅은 고르지 못해서 배추를 군데군데 심어 놓았다. 배추들이 모여있는 곳에는 배추흰지렁이가 한 마리만 있으면 되므로 서로 인접해있는 배추들이 몇 군데에 퍼져있는지 조사하면 총 몇 마리의 지렁이가 필요한지 알 수 있다. 예를 들어 배추밭이 아래와 같이 구성되어 있으면 최소 5마리의 배추흰지렁이가 필요하다. 0은 배추가 심어져 있지 않은 땅이고, 1은 배추가 심어져 있는 땅을 나타낸다.
1 | 1 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 |
0 | 1 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 |
0 | 0 | 0 | 0 | 1 | 0 | 0 | 0 | 0 | 0 |
0 | 0 | 0 | 0 | 1 | 0 | 0 | 0 | 0 | 0 |
0 | 0 | 1 | 1 | 0 | 0 | 0 | 1 | 1 | 1 |
0 | 0 | 0 | 0 | 1 | 0 | 0 | 1 | 1 | 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)가 주어진다. 두 배추의 위치가 같은 경우는 없다.
출력
각 테스트 케이스에 대해 필요한 최소의 배추흰지렁이 마리 수를 출력한다.
생각해 볼 점
2667번 문제와 동일하게 풀었습니다.
BFS와 큐를 이용하여 풀었습니다.
visited 배열 대신, matrix를 방문했으면 1 -> 2로 표기하여 구분했습니다.
코드
#include <iostream>
#include <queue>
using namespace std;
class graph
{
private:
int M, N; //가로 , 세로
int **map;
int result;
int move_x[4] = {0, -1, 0, 1}; //상하좌우 움직임
int move_y[4] = {-1, 0, 1, 0};
queue<pair<int, int>> q;
public:
graph(int N, int M, int K);
~graph();
int get_result();
void bfs();
};
//생성자
graph::graph(int M, int N, int K)
{
//이차원 배열 초기화
this->M = M;
this->N = N;
map = new int*[N + 2]; //상 하 좌 우 이동 시 인덱스를 넘어가지 않도록 + 2
result = 0;
for(int i = 0; i < N + 2; i++)
{
map[i] = new int[M + 2];
fill_n(map[i], M + 2, 0);
}
//그래프 입력
for(int i = 0; i < K; i++)
{
int x, y;
scanf("%d %d", &x, &y);
map[y + 1][x + 1] = 1;
}
}
//소멸자
graph::~graph()
{
for(int i = 0; i < N + 2; i++) delete[] map[i];
delete[] map;
}
//결과 값을 반환하는 함수, BFS 함수를 실행함
int graph::get_result()
{
for(int i = 1; i < N + 1; i++)
{
for(int j = 1; j < M + 1; j++)
{
if(map[i][j] == 1)
{
q.push(make_pair(j, i));
map[i][j] = 2; //2는 방문했음을 뜻함
result++;
bfs();
}
}
}
return result;
}
//BFS 알고리즘
void graph::bfs()
{
if(q.empty()) return;
pair<int, int> current = q.front();
q.pop();
for(int i = 0; i < 4; i++)
{
int new_x = current.first + move_x[i];
int new_y = current.second + move_y[i];
if(map[new_y][new_x] == 1)
{
q.push(make_pair(new_x, new_y));
map[new_y][new_x] = 2; //방문했다고 표기
}
}
bfs();
}
int main()
{
int T;
scanf("%d", &T);
for(int i = 0; i < T; i++)
{
int M, N, K;
scanf("%d %d %d", &M, &N, &K);
graph g(M, N, K);
printf("%d\n", g.get_result());
}
return 0;
}
그 외
(X, Y)와 이차원 배열의 matrix[y][x]를 늘 주의해야합니다.
한번 헷갈려서 코드를 잘못 작성하면 저처럼 낭패를 보기 쉽습니다.
'공부 및 정리 > 백준 코드' 카테고리의 다른 글
[C++]백준 - 11066번 문제 (0) | 2021.07.28 |
---|---|
[C++]백준 - 2178번 문제 (0) | 2021.07.28 |
[C++]백준 - 2667번 문제 (0) | 2021.07.25 |
[C++]백준 - 12015번 문제 (0) | 2021.07.25 |
[C++]백준 - 2606번 문제 (0) | 2021.07.23 |