[백준] 2606번: 바이러스(BFS) - C++

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

2606번: 바이러스

---

처음 코드(틀림)

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

int N,T;
int map[101][101];
int visited[101];
int result = 0;
queue<int> q;

void reset()
{
    for(int i = 1; i<=N;i++)
    {
        visited[i]=0;
        for(int j = 1; j<=N;j++)
            map[i][j]=0;
    }
}

void BFS(int a)
{
    q.push(a);
    
    while(!q.empty())
    {
        int b = q.front();
        q.pop();
        
        for(int i = b+1;i<=N;i++)
        {
            if(map[b][i] == 1 && !visited[i])
            {
                visited[i] = true;
                q.push(i);
                result++;
            }
        }
    }
}

int main()
{
    cin>>N;
    cin>>T;
    reset();
    
    for(int i = 0;i<T;i++)
    {
        int n1,n2;
        cin>>n1>>n2;
        
        map[n1][n2] = 1;
    }
    BFS(1);
    cout<<result;
}

처음에는 다음과 같이 오른쪽 삼각형에만 저장을하고 탐색하는 방법을 선택하였다. 하지만, 이 방법은 단방향으로 연결시킨 것이기 때문에 예를 들어, 1은 5와 연결되어있지만 5는 1과 연결되지 않은 것으로 간주한다. 따라서 양방향으로 연결시켜 문제를 해결해야한다.

 

맞은 코드

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

int N,T;
int map[101][101];
int visited[101];
int result = 0;
queue<int> q;

void reset()
{
    for(int i = 1; i<=N;i++)
    {
        visited[i]=0;
        for(int j = 1; j<=N;j++)
            map[i][j]=0;
    }
}

void BFS(int a)
{
    q.push(a);
    visited[a] = true;
    
    while(!q.empty())
    {
        int b = q.front();
        q.pop();
        
        for(int i = 1;i<=N;i++)
        {
            if(map[b][i] == 1 && !visited[i])
            {
                visited[i] = true;
                q.push(i);
                result++;
            }
        }
    }
}

int main()
{
    cin>>N;
    cin>>T;
    reset();
    
    for(int i = 0;i<T;i++)
    {
        int n1,n2;
        cin>>n1>>n2;
        
        map[n1][n2] = 1;
        map[n2][n1] = 1;
    }
    BFS(1);
    cout<<result;
}

양방향으로 연결시켜 문제를 해결할 수 있었다!