PS/깊이 우선 탐색, 너비 우선 탐색[DFS, BFS]

백준 1012 유기농 배추(c++)

SeungbeomKim 2022. 8. 5. 02:34

https://www.acmicpc.net/problem/1012

 

1012번: 유기농 배추

차세대 영농인 한나는 강원도 고랭지에서 유기농 배추를 재배하기로 하였다. 농약을 쓰지 않고 배추를 재배하려면 배추를 해충으로부터 보호하는 것이 중요하기 때문에, 한나는 해충 방지에 

www.acmicpc.net

<코드>

#include <iostream>
#include <cstring>
#include <vector>
#define max 51
using namespace std;
int mov[4][2] = {{-1,0},{1,0},{0,-1},{0,1}};
int board[max][max];
bool visited[max][max] = {0};
int n,m,x,y;
int cnt=0;
void dfs(int a,int b)
{
	visited[a][b] = true;
	for(int i=0;i<4;i++){
		int sr = a + mov[i][0];
		int sc = b + mov[i][1];
		if(sr<0 || sc<0 || sr>m-1 || sc>n-1) continue;
		if(visited[sr][sc]) continue;
		if(!visited[sr][sc] && board[sr][sc]){
			visited[sr][sc] = true;
			dfs(sr,sc);
		} 
	}
}
int main()
{
	int t;
	cin>>t;
	while(t--){
		cnt = 0;
		int k;
		cin>>m>>n>>k;
		for(int i=0;i<k;i++)
		{
			cin>>x>>y;
			board[x][y]=1;
		}
		for(int i=0;i<m;i++)
		{
			for(int j=0;j<n;j++){
				if(!visited[i][j] && board[i][j])
				{
					dfs(i,j);
					cnt++;
				}
			}
		}
		cout<<cnt<<'\n';
		memset(board,0,sizeof(board));
		memset(visited,0,sizeof(visited));
		
	}
}

<풀이과정>

이 문제는 dfs를 이용하여 간편히 풀 수 있다. 배추가 있는 곳(위치)에 1의 값을 주고 상하좌우에 연결된 부분이 있는지 찾아가면 된다. dfs, bfs도 풀다 보니, 유형이 조금씩만 변형되고 알고리즘 자체 과정(탐색)은 비슷한 거 같다.