[프로그래머스 Lv.3 / C++] Day79. 네트워크

2024. 08. 21 - 코딩테스트 스터디 Day79

 
 

01. 문제 설명

네트워크란 컴퓨터 상호 간에 정보를 교환할 수 있도록 연결된 형태를 의미합니다. 예를 들어, 컴퓨터 A와 컴퓨터 B가 직접적으로 연결되어 있고, 컴퓨터 B와 컴퓨터 C가 직접적으로 연결되어 있을 때 컴퓨터 A와 컴퓨터 C도 간접적으로 연결되어 정보를 교환할 수 있습니다. 따라서 컴퓨터 A, B, C는 모두 같은 네트워크 상에 있다고 할 수 있습니다.
컴퓨터의 개수 n, 연결에 대한 정보가 담긴 2차원 배열 computers가 매개변수로 주어질 때, 네트워크의 개수를 return 하도록 solution 함수를 작성하시오.

 
 

02. 제한사항

• 컴퓨터의 개수 n은 1 이상 200 이하인 자연수입니다.
• 각 컴퓨터는 0부터 n-1인 정수로 표현합니다.
• i번 컴퓨터와 j번 컴퓨터가 연결되어 있으면 computers[i][j]를 1로 표현합니다.
• computer[i][i]는 항상 1입니다.

 
 

03. 입출력 예

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개의 네트워크가 있습니다.

 
 

4. 풀이 및 답

#include <string>
#include <vector>

using namespace std;

int check[200] = { 0, };

void dfs(int i, int n, vector<vector<int>> computers) {
    check[i] = 1;
    
    for (int j = 0; j < n; j++) {
        if (check[j] == 0 && computers[i][j] == 1) {
            dfs(j, n, computers);
        }
    }
}

int solution(int n, vector<vector<int>> computers) {
    int answer = 0;
    
    for (int i = 0; i < n; i++) {
        if (check[i] == 0) {
            dfs(i, n, computers);
            answer++;
        }
    }
    
    return answer;
}

컴퓨터의 탐색 여부를 체크하는 check 배열을 선언하고 0으로 초기화했다.
 
0부터 n까지 반복문을 돌면서 아직 탐색하지 않은 컴퓨터를 dfs 함수를 통해 탐색한다.
 
dfs 함수는 한 컴퓨터와 연결된 컴퓨터를 발견하면 dfs 함수를 재귀호출하여 탐색해 나간다.
예를 들어, 1번 컴퓨터를 탐색하던 도중 2번 컴퓨터와 연결되어 있다는 것을 확인하게 되면 2번 컴퓨터를 탐색하기 시작한다. 이때, 2번 컴퓨터와 연결된 컴퓨터 중 1번 컴퓨터는 이미 탐색했기 때문에 건너뛴다.
 
 

정답!

---

끝~
 
참고한 게시글

[프로그래머스] 네트워크(BFS,DFS) / C++