[알고리즘 문제풀이] 프로그래머스 네트워크

어제 풀었는데 포스팅 귀찮다고 미뤄버렸음니도..ㅎㅎ.. 코테 뿌셔 🔥 (제발ㅠㅠ)

이번 문제는 프로그래머스 고득점 kit DFS/BFS 분류에 있는 level3 문제다 !
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입니다.

입출력 예

ncomputersreturn
3[[1, 1, 0], [1, 1, 0], [0, 0, 1]]2
3[[1, 1, 0], [1, 1, 1], [0, 1, 1]]1

풀이 방법

문제를 먼저 해석해보면 결국 n개의 컴퓨터라고 칭하는 노드가 있는 그래프이며, 연결된 한 그룹을 네트워크라고 할 때, 총 몇 개의 네트워크가 있는가, 즉 몇 개의 그룹으로 이루어진 그래프인가를 묻는 문제이다.

그렇기때문에 너무나도 당연하게 DFS를 사용해서 문제를 풀면된다. 노드는 0부터 시작하기 때문에 0부터 탐색을 시작해서 DFS로 연결된 끝까지 탐색이 완료되면 네트워크의 개수를 하나 증가시켜준다. 다음 노드도 차근차근 탐색을 시작하되 이미 카운트된 그룹에 포함된 노드는 이미 방문했기에 visited를 이용해서 건너뛰어주면 된다 !

너무 간단한 문제라 따로 설명할 게 없다 ! 코드도 보면 아주 간단하다 !

코드

class Solution {
    private static Boolean[] visited;

    public static void dfs(int start, int[][] computers){
        visited[start] = true;
        for(int i=0; i<computers[start].length; i++){
            if(!visited[i] && computers[start][i] == 1){
                dfs(i, computers);
            }
        }
    }

    public static int solution(int n, int[][] computers) {
        int answer = 0;
        visited = new Boolean[n];
        for(int i=0; i<n; i++){ visited[i] = false; }
        for(int i=0; i<computers.length; i++){
            if(!visited[i]){
                dfs(i, computers);
                answer++;
            }
        }
        return answer;
    }
}

좋은 웹페이지 즐겨찾기