[프로그래머스 : 레벨 3] 네트워크 : DFS(Java)

이번 문제는 생각외로 쉬웠다.

어떤 알고리즘으로 문제를 풀어야 할지를 먼저 생각하니 금방 풀렸던것 같다.

 

문제부터 살펴보자.

 

문제 설명

 

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

 

입출력 예 설명

 

예제 #1
아래와 같이 2개의 네트워크가 있습니다.

예제 #2
아래와 같이 1개의 네트워크가 있습니다.

 

 

나는 이번 문제는 DFS 알고리즘을 통해서 풀었다.

이유는 더이상 자식이 없는 지점까지 가서 없다면 return1을 해서 이어지는 구도는 끝까지 이어지게 하고

끊긴다면 1을 return 해야했기 때문이다.

 

(DFS 알고리즘에 관한건 아래 링크를 참조하시면 된다.)

https://wpioneer.tistory.com/130 

[자료구조] DFS(Depth First Search) Java 구현

 

 

그럼 소스를 살펴보자.

 

    public int solution(int n, int[][] computers) {
        int answer = 0;
        
        //이번 문제는 DFS 알로리즘으로 풀어야 한다.
        //왜냐하면 연결이 없을때 바로 벗어나야하기때문이다.
        //그리고 이번 문제는 인접행렬을 통해서 DFS를 구현해야한다.
        
        int[] visit = new int[n]; //방문체크를 위해서
        int count = 0;
        
        for(int i = 0; i<n;i++) {
        	if(visit[i] != 1) {
        		count += dfs(i, visit,computers,computers[i]);
        	}
        }
        
        return answer = count;
    }
	private int dfs(int index, int[] visit, int[][] computers, int[] computer) {
		// TODO Auto-generated method stub
		int count = 0;
		
		visit[index] = 1;
		boolean flag = true; //자식이 없는 지점을 체크하기위해서 
		for(int i = 0; i<computers.length;i++) {
			if(computer[i] == 1 && index != i && visit[i] != 1) { 
				//연결되어 있고 [2][2]인 부분이 아니고 방문하지 않았다면 진입
				flag = false;
				count = dfs(i,visit,computers,computers[i]);
				
			}
		}
		
		if(flag)
			count = 1;
		
		return count;
	}

 

 

자세한 내용은 내가 주석으로 설명했으니 확인해보면 될것 같다.

 

느낀점

1. 무슨 알고리즘을 써야할지 먼저 생각하자.