코딩테스트/💯프로그래머스 코딩테스트 연습
깊이/너비 우선 탐색(DFS/BFS)_네트워크
1son
2023. 2. 17. 11:40
문제 설명💛
네트워크란 컴퓨터 상호 간에 정보를 교환할 수 있도록 연결된 형태를 의미합니다.
예를 들어, 컴퓨터 A와 컴퓨터 B가 직접적으로 연결되어있고, 컴퓨터 B와 컴퓨터 C가 직접적으로 연결되어 있을 때
컴퓨터 A와 컴퓨터 C도 간접적으로 연결되어 정보를 교환할 수 있습니다.
따라서 컴퓨터 A, B, C는 모두 같은 네트워크 상에 있다고 할 수 있습니다.
문제 핵심✔️
컴퓨터의 개수 n, 연결에 대한 정보가 담긴 2차원 배열 computers가 매개변수로 주어질 때,
네트워크의 개수를 return 하도록 solution 함수를 작성하시오.
[프로그래머스] 네트워크 문제풀이 (Java)
[프로그래머스] 네트워크 문제풀이 (Java)
velog.io
참고해서 풀어보자.
이 문제는 DFS문제이다.
재귀호출을 이용해서 해결한다.
풀이 🔥
class Solution {
public int solution(int n, int[][] computers) {
int answer = 0; //answer 0으로 초기화
boolean[] check = new boolean[n]; //n개수만큼 boolean배열 만들고
for(int i=0;i<n;i++){
if(!check[i]){ //check[i]가 false인게 있다면, dfs호출하고 answer++해준다.
dfs(computers, i, check);
answer++;
}
}
return answer;
}
boolean[] dfs(int[][] computers, int i, boolean[] check){
check[i]=true; //true로 변경하고
for(int j=0;j<computers.length;j++){
if(i!=j && computers[i][j]==1 && check[j]==false){ //자기 자신이 아니며, false이며, 값이 1인것
check=dfs(computers, j, check); //재귀호출
}
}
return check;
}
}
@Test
public void 정답() {
Assert.assertEquals(2, solution(3, new int[][]{{1, 1, 0}, {1, 1, 0}, {0, 0, 1}}));
Assert.assertEquals(1, solution(3, new int[][]{{1, 1, 0}, {1, 1, 1}, {0, 1, 1}}));
}
@Test 코드 작성하는 법도 외워두자.