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

DFS/BFS를 활용하는 Lv3문제입니다.

---

문제 설명

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

---

< 내 풀이 >

위의 입력은 인접행렬임 노드랑 연결된 edge를 확인할 수 있음

dfs가 더 익숙해서 dfs를 활용해 풀기로함

 

노드를 순서대로 방문하고 방문 노드의 연결을 확인함

 

기본 구상은 위와 같고 코드를 보는게 더 이해가 빨리될 듯함

<코드>

class Solution {
    public void dfs(int[][] computers, boolean[][] link, int idx, int n){
        for(int i=0; i<n; i++){
            if(computers[idx][i]==1 && !link[idx][i]){
                link[idx][i] = link[i][idx] = true;
                dfs(computers, link, i, n);
            }
        }
    }
    public int solution(int n, int[][] computers) {
        int answer = 0;
        boolean[][] link = new boolean[n][n];
        for(int i=0; i<n; i++){
            if(!link[i][i]){
                answer++;
                dfs(computers, link, i, n);
            }
        }
        return answer;
    }
}

link라는 연결 유무를 확인하는 boolean 2차원 배열을 생성하고 활용한다.

 

전제 조건에 computers[i][i]는 항상 1이라는 내용이 있다. 노드의 self loop가 존재한다는 것이다.

따라서 computers[i][i]는 항상 true지만 link[i][i]는 초기상태에 false 값을 가져 방문 여부 확인에 사용되기 적절하다.

 

노드를 순서대로 방문할 때 link[i][i]의 값이 false라면 해당 노드에 대한 연결(네트워크)상태를 체크하지 않은 것이다.

 

answer(네트워크 개수)를 증가 시키고 이 노드 i로부터 시작되는 연결(네트워크)를 확인하기 위해 dfs를 수행한다.

 

dfs는 해당 노드에서부터의 연결을 살피게 되는데

만일 연결이 존재하고 최초의 연결 확인일 경우(link값이 false)

해당 구간의 link를 true로 바꿔주고 i노드와 연결된 새로운 노드의 연결을 살피는 dfs를 수행하게 한다.

재귀적으로 수행되며 네트워크 체크 과정을 갖게 될것이다.

 

문제출처: https://programmers.co.kr/learn/courses/30/lessons/43162?language=java

코딩테스트 연습 - 네트워크 | 프로그래머스