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

문제링크
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입니다.

입출력 예

문제풀이

1. 연결되어 있는 그래프 배열을 생성한다.
2. 방문을 표시할 배열을 생성한다.
3. 인자값으로 넘겨받은 인덱스의 방문배열을 true로 바꾸고 해당 그래프에 연결된 모든 값들을 true로 변경한다. 변경이 완료되면 true를 반환한다.
4. n까지 돌면서 방문되지 않은 인덱스를 dfs함수를 호출하여 answer값을 1씩 증가한다.

코드

function solution(n, computers) {
    var answer = 0;
    let graph = Array.from(Array(n), () => Array());
    for(let i=0; i<n; i++){
        for(let j=0; j<n; j++)
            if(computers[i][j] == 1 && i != j) graph[i].push(j);
    }
    let visited = new Array(n).fill(false);
    const dfs = (i) => {
        visited[i] = true;
        graph[i].map(v => {
            if(!visited[v]) dfs(v);
        })
        return true;
    }
    for(let i=0; i<n; i++){
        if(!visited[i])
            if(dfs(i)) answer++;
    }
    return answer;
}