Language/C++
[c++][백준 2146][BFS] 다리 만들기 - 컴도리돌이
컴도리돌이
2021. 11. 21. 09:00
728x90
728x90
2146번: 다리 만들기
여러 섬으로 이루어진 나라가 있다. 이 나라의 대통령은 섬을 잇는 다리를 만들겠다는 공약으로 인기몰이를 해 당선될 수 있었다. 하지만 막상 대통령에 취임하자, 다리를 놓는다는 것이 아깝다
www.acmicpc.net
* 풀이 과정
1. 육지와 육지 사이의 구분이 필요하기 때문에 육지마다 번호를 표시하였다.(처음 육지를 1로 입력받는 것을 -1로 입력받았다.)
2. 육지와 육지 사이의 다리를 놓는다.
1) 번호에 따른 육지의 x, y 값들을 queue에 넣고 방문 표시를 한다.
2) 다른 육지 번호가 나오면 다리의 값을 return 한다.
3) 바다가 나오면 queue에 넣고 다리가 설치했다고 가정을 하고 bridge 값도 +1을 시켜서 queue에 넣는다. 해당 좌표는 방문 표시를 한다.
* 함수 설명
land_number : 육지의 번호를 매기는 함수.
short_bridge : 육지와 육지 사이의 최소 다리 건설 값을 반환해주는 함수.
#include <iostream>
#include <queue>
#include <vector>
using namespace std;
int n;
int dx[] = {-1,1,0,0};
int dy[] = {0,0,-1,1};
vector<vector<int>> matrix;
vector<vector<bool>> visited;
struct info
{
int x;
int y;
int bridge;
};
void input()
{
cin >> n;
matrix.resize(n,vector<int>(n,0));
visited.resize(n,vector<bool>(n,false));
for(int i=0; i<n ; i++)
for(int j=0; j<n ; j++)
{
cin >> matrix[i][j];
if(matrix[i][j]) matrix[i][j] = -1;
}
}
void land_number(int i,int j,int num)
{
queue<pair<int,int>> q;
q.push({i,j});
visited[i][j] = true;
matrix[i][j] = num;
while(!q.empty())
{
int x = q.front().first;
int y = q.front().second;
q.pop();
for(int i=0; i<4; i++)
{
int nx = x + dx[i];
int ny = y + dy[i];
if(nx >=0 and ny >= 0 and nx < n and ny < n and !visited[nx][ny] and matrix[nx][ny] == -1)
{
visited[nx][ny] = true;
q.push({nx,ny});
matrix[nx][ny] = num;
}
}
}
}
int short_bridge(int num)
{
queue<info> q;
for(int i=0; i<n; i++)
{
for(int j=0; j<n; j++)
{
if(matrix[i][j] == num)
{
visited[i][j] = true;
q.push({i,j,0});
}
}
}
while(!q.empty())
{
int x = q.front().x;
int y = q.front().y;
int b = q.front().bridge;
q.pop();
for(int i=0; i<4; i++)
{
int nx = x + dx[i];
int ny = y + dy[i];
if(nx >= 0 and ny >= 0 and nx < n and ny < n)
{
if(matrix[nx][ny] and matrix[nx][ny] != num) return b;
if(!matrix[nx][ny] and !visited[nx][ny])
{
visited[nx][ny] = true;
q.push({nx,ny,b+1});
}
}
}
}
}
void solve()
{
int num= 1,answer = 987654321;
for(int i=0; i<n; i++)
for(int j=0; j<n; j++)
if(matrix[i][j] and !visited[i][j])
{
land_number(i,j,num);
num++;
}
for(int i=1; i< num; i++)
{
answer = min(answer,short_bridge(i));
visited.clear();
visited.resize(n,vector<bool>(n,false));
}
cout << answer << '\n';
}
int main()
{
input();
solve();
return 0;
}
728x90
728x90