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 | computers | return |
---|---|---|
3 | [[1, 1, 0], [1, 1, 0], [0, 0, 1]] | 2 |
3 | [[1, 1, 0], [1, 1, 1], [0, 1, 1]] | 1 |
문제 풀이
function solution(n, computers) {
var answer = 0;
let network = new Array(n).fill(false);
let queue = [];
const eq = (num) => queue.push(num),
dq = () => queue.shift();
for (let i = 0; i < n; i++) {
if (network[i] === false) {
eq(computers[i]);
while (queue.length > 0) {
let arr = dq();
arr.forEach((el, j) => {
if (el === 1 && network[j] === false && i !== j) {
eq(computers[j]);
network[j] = true;
}
});
}
answer++;
}
}
return answer;
}
dfs/bfs 문제이고 내가 푼 방식은 너비우선 방식으로 문제를 풀었다.
먼저 큐를 만들어주고 반복문으로 n만큼 돌리면서 사용한 네트워크인지 확인해서 사용하지 않았다면 큐에 네트워크 배열을 넣어주고 큐의 길이가 0이 될때까지 while 반복문을 이용하여 돌려줬다. 그러면서 사용하지 않은 네트워크들을 큐에 넣어줬고 while 반복이 끝날 때 answer에 1을 더해줬다. 그 뒤 반복문이 반복되도 이미 사용한 네트워크라면 넘어가게 되고 사용하지 않은 네트워크라면 answer를 늘려가면서 마지막에 answer를 리턴해줬다.