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

네트워크

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

입출력 예ncomputersreturn

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개의 네트워크가 있습니다.

 

 

풀이방법

-union-find와 DFS를 이용해서 풀이가 가능하다

 

 

첫 번째 풀이

function getParent(parent,a){
    if(parent[a]==a) return a;
    return parent[a]=getParent(parent,parent[a]);
}
function unionParent(parent,a,b){
    a=getParent(parent,a);
    b=getParent(parent,b);
    if(a<b) parent[b]=a;
    else parent[a]=b;
}
function findParent(parent,a,b,){
    a=getParent(parent,a);
    b=getParent(parent,b);
    if(a==b)  return 1;
    else return 0;
}
function solution(n, computers) {
    var answer = 0;
    let arr=Array.from({length:n},(_,i)=>i);
    for(var i=0;i<computers.length;i++){
        for(var j=0;j<computers[i].length;j++){   
            if(computers[i][j]==1) unionParent(arr,i,j);
        }
    }
    for(var i=0;i<n;i++){
        getParent(arr,i)
    }
    
    return [...new Set(arr)].length;
}

-union-find를 이용했다. 

위 공식은 외워두는 것이 좋다.

마지막에 한번 getParent를 통해서 parent테이블인 arr을 최신화 시켜줘야 답이 나온다.

 

 

 

두 번째 풀이(DFS를 이용한 풀이)

function solution(n, computers) {
    let visited = [false];
    let answer = 0;

    function dfs(i) {
        visited[i] = true;
        for(let j=0; j<computers[i].length; j++) {
            if(computers[i][j]===1 && !visited[j]){
                dfs(j);
            }
        }
    }
    
    for (let i=0; i < computers.length; i++) {
        if (!visited[i]) {
            dfs(i)
            answer++;
        }
    }
    return answer;
}

-DFS를 이용해서 방문하지 않은 노드를 확인하면서 뻗어간다.

-DFS를 방문하면 그 노드는 방문했다고 표시를 해줘야 더 깊이 들어가지 않는다.