본문 바로가기

Language/C++

[c++][백준 2146][BFS] 다리 만들기 - 컴도리돌이

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