[프로그래머스] - 네트워크(C/C++)

프로그래머스

[문제 설명]

네트워크란 컴퓨터 상호 간에 정보를 교환할 수 있도록 연결된 형태를 의미합니다. 예를 들어, 컴퓨터 A와 컴퓨터 B가 직접적으로 연결되어있고, 컴퓨터 B와 컴퓨터 C가 직접적으로 연결되어 있을 때 컴퓨터 A와 컴퓨터 C도 간접적으로 연결되어 정보를 교환할 수 있습니다. 따라서 컴퓨터 A, B, C는 모두 같은 네트워크 상에 있다고 할 수 있습니다.

컴퓨터의 개수 n, 연결에 대한 정보가 담긴 2차원 배열 computers가 매개변수로 주어질 때, 네트워크의 개수를 return 하도록 solution 함수를 작성하시오.

제한사항

• 컴퓨터의 개수 n은 1 이상 200 이하인 자연수입니다.
• 각 컴퓨터는 0부터 n-1인 정수로 표현합니다.
• i번 컴퓨터와 j번 컴퓨터가 연결되어 있으면 computers[i][j]를 1로 표현합니다.
• computer[i][i]는 항상 1입니다.

입출력 예

 

n	computers	return
3	[[1, 1, 0], [1, 1, 0], [0, 0, 1]]	2
3	[[1, 1, 0], [1, 1, 1], [0, 1, 1]]	1

입출력 예 설명

예제 #1
아래와 같이 2개의 네트워크가 있습니다.

예제 #2
아래와 같이 1개의 네트워크가 있습니다.

---

[문제 풀이]

네트워크의 개수를 출력하는 문제인데, 간단하게 컴퓨터들이 몇개의 그룹으로 묶어져있는지 확인하면 된다.

BFS를 이용하여 컴퓨터 한대당 연결되어 있는 모든 컴퓨터를 확인하고 방문 처리를 해준다.

더이상 연결되어 있는 컴퓨터가 없다면, 컴퓨터의 개수만큼의 반복문을 통해 방문하지 않은 컴퓨터가 있는지 확인한다.

각각의 컴퓨터가 방문했는지 안했는지 체크하고, 방문하지 않는 컴퓨터를 만났을때 answer의 값을 1씩 증가시킨다.

전체 컴퓨터 개수만큼 반복문을 완료했을때, answer에 들어있는 값은 네트워크의 개수이다.

 

 

파이팅 !

코드가 깨끗하고 명확하게 보일진 모르겠지만, 누군가에게 도움이 될 수도.. 있기에..

더보기

#include <string>
#include <vector>
#include <queue>
using namespace std;
vector<int> isvisited;

int solution(int n, vector<vector<int>> computers) {
    int answer = 0;
    isvisited = vector<int>(n,0);
    int count = 0;
    for(int i =0; i < n; i++)
    {
        if(isvisited[i]) continue;
        count++;
        queue<int> q;
        q.push(i);
        while(!q.empty())
        {
            int index = q.front();
            q.pop();
            if(isvisited[index]) continue;
            isvisited[index] = count;
            for(int j =0; j < n; j++)
            {
                if(i == j || isvisited[j]) continue;
                if(computers[index][j] == 1) q.push(j);
            }
        }
    }

    return answer = count;
}