프로그래머스 : 네트워크(Java)

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개의 네트워크가 존재합니다.

 

2. 아래의 경우 1개의 네트워크가 존재합니다.

 

---

[풀이]

 

 visited라는 방문에 대한 boolean배열을 만들어서 해당 컴퓨터를 방문했는지 안했는지를 알 수 있게 하였다. DFS를 구현할 때는 스택을 이용해서 구현하였다. solution 메소드에서 반복문을 돌면서 해당 컴퓨터를 방문안했다면 DFS의 매개변수로 해당 컴퓨터 번호를 넣어서 DFS 메소드가 실행되도록 하였다.

 

 문제를 풀며 주의할 점이라면, 아무래도 DFS를 구현하는 부분이라고 생각한다. 그외에는 for문을 돌면서 방문하지 않은 컴퓨터 번호를 DFS에 넣는 부분을 주의하면 될 것 같다.

 

---

[코드]

 

import java.util.Stack;

class Solution {
    static boolean[] visited;
    static int[][] arr;
    static int N;
    
    public int solution(int n, int[][] computers) {
        int answer = 0;
        N = n;
        arr = computers;
        visited = new boolean[N];

        for(int i = 0; i< n; i++){
            if(visited[i] == false){
                dfs(i);
                answer++;
            }
        }

        return answer;
    }

    public static void dfs(int start){
        Stack<Integer> stack = new Stack<>();
        stack.push(start);
        visited[start] = true;

        while(!stack.isEmpty()){
            for(int i = 0; i< N; i++){
                if(arr[start][i] == 1 && visited[i] == false){
                    visited[i] = true;
                    stack.push(i);
                    dfs(i);
                }
            }
            stack.pop();
        }
    }
}