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

 

문제

문제 링크 : 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

 

풀이

DFS를 활용해서 풀었다. 전체 컴퓨터의 수를 배열로 저장한 다음, DFS로 컴퓨터를 마킹했다.

어차피 같은 네트워크 안에 있으면 DFS로 모두 마킹됨으로 한번 DFS 끝날때마다 네트워크가 하나 늘었다고 판단했다.

마킹되지 않은 나머지 컴퓨터들에 대해서도 DFS를 진행했다.

시간복잡도는 N개의배열을 N번 호출하므로 $O(N^2)$인 것 같다.

 

코드

class Solution {
	public int solution(int n, int[][] computers) {
    	int[] computerCheck = new int[n];
        int answer = CheckNetwork(computerCheck, computers);
        return answer;
    }
    
    public int CheckNetwork(int[] computerCheck, int[][] computers) {
    	int answer = 0;
    	for(int i=0; i<computerCheck.length; i++) {
    		if(computerCheck[i] ==0) {
    			DFS(i,computerCheck,computers);
    			answer++;
    		}
    	}
    	return answer;
    }
    
    public void DFS(int node,int[] computerCheck, int[][] computers) {
    	for(int i=0; i<computerCheck.length; i++) {
    		if(computerCheck[i] != 1 && computers[node][i] == 1) {
    			computerCheck[i] = 1;
    			DFS(i, computerCheck, computers);
    		}
    	}
    	return;
    }
}

 

결과