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

백준 14502 연구소(c++)

SeungbeomKim 2022. 8. 9. 20:57

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

 

14502번: 연구소

인체에 치명적인 바이러스를 연구하던 연구소에서 바이러스가 유출되었다. 다행히 바이러스는 아직 퍼지지 않았고, 바이러스의 확산을 막기 위해서 연구소에 벽을 세우려고 한다. 연구소는 크

www.acmicpc.net

<코드>

#include <iostream>
#include <queue>
#include <algorithm>
#include <vector>
using namespace std;

int n,m;
int map[8][8];
int mov[4][2] = {{-1,0},{1,0},{0,1},{0,-1}};

int ret = 0;

void bfs(){
	bool visited[8][8] = {false};
	queue<pair<int,int>> q;
	int backup[8][8];
	for(int i=0;i<n;i++){
		for(int j=0;j<m;j++){
			backup[i][j] = map[i][j];
			if(backup[i][j]==2){
				q.push({i,j});
				visited[i][j] = true;
			}
		}
	}
	while(!q.empty()){
		pair<int,int> cur = q.front();
		q.pop();
		for(int i=0;i<4;i++)
		{
			int sr = cur.first + mov[i][0];
			int sc = cur.second + mov[i][1];
			
			if(sr<0 || sc<0 || sr>n-1 || sc >m-1) continue;
			if(!visited[sr][sc] && !backup[sr][sc])
			{
				visited[sr][sc] = true;
				backup[sr][sc] = 2;
				q.push({sr,sc});
			}
		}
	}
	int candi = 0;

	for(int i=0;i<n;i++)
	{
		for(int j=0;j<m;j++){
			if(!backup[i][j]){
				++candi;
			}
		}
	}
	if(ret < candi){
		ret = candi;
	}
}
void dfs(int count,int x,int y)
{
	if(count == 3) {
		bfs();
		return;	
	}
	for(int i=x;i<n;i++){
		for(int j=y;j<m;j++){
			if(!map[i][j]){
				map[i][j] = 1;
				dfs(count+1,i,j);
				map[i][j] = 0;
			}
		}
		y=0;
	}
}
int main()
{
	cin>>n>>m;
	for(int i=0;i<n;i++)
	{
		for(int j=0;j<m;j++)
		{
			cin>>map[i][j];
		}
	}
	dfs(0,0,0);
	
	cout<<ret<<'\n';
}

<풀이과정>

이 문제는 dfs와 bfs가 혼합된 문제이다. 백트래킹(backtracking) 방식으로 벽을 설치한 후 바이러스에 감염된 방을 하나하나 queue에 push해주고 안전 영역을 하나하나씩 세주면 된다.

bfs, dfs가 혼합된 백트래킹 방식의 문제는 처음 풀어보는데 접근 자체가 어려운거 같다.

이제 백트래킹 코드 구조와 사용 목적에 대해서도 차차 공부해야겠다..

<참고 자료>

https://www.youtube.com/watch?v=FAGeo3IH2UE&t=523s