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

https://school.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입니다.

 

🖊️ 문제 풀이

나는 이 문제를 인접리스트와 DFS를 이용해 해결했다.

먼저 computers 데이터들을 인접리스트 형태로 어떤 컴퓨터가 서로 연결되어있는지를 기록해주었다.

인접리스트를 아직 방문하지 않은 경우에  DFS를 돌리고, 네트워크 개수를 +1 해주었다.

 

 

재귀 형태의 DFS 함수에서는 먼저 DFS 함수의 매개변수에 해당하는 노드를 방문 처리 해준다.

해당 노드와 연결된 노드로 이동하면서 아직 방문하지 않았을 경우에는 연결된 노드 DFS 함수를 적용시켜준다.

 

이렇게 인접리스트랑 DFS를 이용해서 연결된 컴퓨터를 확인하고 네트워크 갯수를 계산할 수 있다.

 

 

🖊️ 코드

package 프로그래머스Lv3;

import java.util.ArrayList;

public class 네트워크 {
  public int solution(int n, int[][] computers) {
    int answer = 0;
    boolean[] visit = new boolean[n];
    ArrayList<Integer>[] list = new ArrayList[n];

    for (int i = 0; i < n; i++) {
      list[i] = new ArrayList<>();
    }

    for (int i = 0; i < n; i++) {
      for (int j = 0; j < n; j++) {
        if(i==j){
          continue;
        }
        if(computers[i][j] == 1){
          list[i].add(j);
          list[j].add(i);
        }
      }
    }

    for (int i = 0; i < n; i++) {
      if(!visit[i]){
        dfs(i, list, visit);
        answer++;
      }
    }
    return answer;
  }

  private static void dfs(int i, ArrayList<Integer>[] list, boolean[] visit) {
    if(visit[i]){
      return;
    }

    visit[i] = true;
    for(int item : list[i]){
      if(!visit[item]){
        dfs(item, list, visit);
      }
    }
  }

}