[프로그래머스] 네트워크

https://programmers.co.kr/learn/courses/30/lessons/43162

코딩테스트 연습 - 네트워크

문제 설명

네트워크란 컴퓨터 상호 간에 정보를 교환할 수 있도록 연결된 형태를 의미합니다. 예를 들어, 컴퓨터 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	computer	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개의 네트워크가 있습니다.

 

 

풀이

오히려 그림이 많이 나올 수록 문제 풀이는 간단해진다! 

이 문제는 그래프 문제와도 유사하다. 간선, 접점으로 이루어져 있으니까.

나는 이 문제에서 1번으로 시작해서 끝까지 네트워크가 퍼져 나가야 한다고 생각했으므로 DFS를 활용하였다. 하지만 BFS를 활용해서도 모든 탐색이 가능하다.

 

단 문제에서 1번만 시작하는 것이 아니라 연결하는 접점별로 분류 해야 하기 때문에 다른 DFS 문제와 같이 처음 시작점을이 아닌 전체의 점에서 시작 해야한다.

또 연결된 점이라는 것을 visit 배열을 통해 파악하고 전체 점에서 시작하되 방문하지 않았던 점만 DFS 알고리즘을 이용하여 네트워크를 연결해 준다.

 

#include <string>
#include <vector>
#include <iostream>
#include <algorithm>

using namespace std;
vector<int> network[201];
bool visit[200] ={false,};

void dfs(int sp) { //DFS 알고리즘
    
    visit[sp] = true;
 
    for(int i=0; i<network[sp].size();i++) {
        
        int npoint = network[sp][i];
       
        if(visit[npoint] == false) {
            visit[npoint] = true;
            dfs(npoint);   
        }   
    }
    

}

int solution(int n, vector<vector<int>> computers) {
   
      for(int i = 0 ; i < n; i++) {
        vector<int> vc;
        for(int j = 0 ; j < n; j++) {     
            if(i == j) continue; // 시작점과 끝점이 동일한 것은 제외
            if(computers[i][j] == 1) { // 연결 되었을 때
                vc.push_back(j);
            }
        }
       
        sort(vc.begin(),vc.end()); // 그래프 문제를 풀기 위해서는 정렬이 반드시 되어있어야 한다. 물론 해당 문제는 그래프문제가 아니기 때문에 정렬을 굳이 해줄 필요는 없다.
        network[i] = vc;
    }
    
    int answer=0;
    
    for(int i = 0;i < n;i++) {
        if(visit[i] == false) {
            dfs(i); //방문하지 않은 점만 DFS 알고리즘 대입
            answer++;
        }
    }
 
    
    return answer;
}

끝