네트워크 C++(dfs/bfs)[프로그래머스]

※ 저의 풀이가 무조건적인 정답은 아닙니다.

다른 코드가 좀더 효율적이고 좋을 수 있습니다.

다른사람들의 풀이는 언제나 참고만 하시기 바랍니다.

문제 주소입니다.

https://programmers.co.kr/learn/courses/30/lessons/43162

코딩테스트 연습 - 네트워크 | 프로그래머스

---

1. 문제 설명

2. 문제 해석

3. 소스 코드

3.1 주석 없는 코드

3.2 주석 있는 코드

3.3 테스트 코드

4. 결과

---

1. 문제 설명

네트워크란 컴퓨터 상호 간에 정보를 교환할 수 있도록 연결된 형태를 의미합니다. 예를 들어, 컴퓨터 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을 보시면 {1,1,0}, {1,1,0}, {0,0,1} 이렇게 되어있습니다. 1번과 2번 컴퓨터는 네트워크 연결되어있고 3번 컴퓨터는 네트워크가 분리되어 있습니다. 따라서 네트워크가 총 2개입니다.   예제 2를 보시면 {1,1,0}, {1,1,1}, {0,1,1} 이렇게 되어 있습니다. 1번은 2번 컴퓨터와 네트워크가 연결되어있고 2번은 1,3번 컴퓨터와 네트워크가 연결되어있고 3번은 2번 컴퓨터와 네트워크가 연결되어 있습니다. 따라서 네트워크는 총 1개입니다.   그렇다면 어떻게 풀면 될까요? 네트워크가 분리되어 있는지 확인을 하면됩니다. 확인을 하기위해서는 네트워크 하나에 연결된 모든 컴퓨터를 없애버리면 됩니다.   처음에 n개의 컴퓨터 대수가 입력이 됩니다. 그렇다면 n번만큼 도는 반복문을 작성을 해주시고 제한사항을 보시면 computer[i][i] 항상 1이기때문에 이것을 이용해서 연결된 컴퓨터인지 아닌지 확인을 하시면 됩니다. computer[i][i]가 1이라면 dfs를 이용해서 탐색을 시작합니다. dfs함수에 들어가면 computer[i][i]를 0으로 바꿔줍니다. 그래야 연결된 컴퓨터를 삭제한것으로 처리되기 때문입니다. 이제 노드만큼 순회(재귀)하며 연결된 노드들을 전부 0으로 바꿔주고 dfs함수를 탈출하면 네트워크 개수를 추가합니다. 연결이 되지않은 컴퓨터는 재귀함수로 갈수 없기때문에 나중에 따로 체크가 가능하게 됩니다.   이해가 잘되지 않으시다면 코드를 봐주세요.   예제 1의경우 반복문을 사용하면 i=0 일때 1,2 가 0으로 바뀌게 되고 answer가 1이됩니다. i=1 일때 2번이 이미 삭제된 컴퓨터이므로 넘어갑니다. i=2 일때 3번 컴퓨터는 0으로 바뀌게 되고 answer가 2가 됩니다.   예제 2의경우 반복문을 사용하면 i=0 일때 1,2,3 컴퓨터는 0으로 바뀌게 되고 answer가 1이됩니다. i=1 일때 2번 컴퓨터는 이미 삭제된 컴퓨터이므로 넘어갑니다. i=2 일때 3번 컴퓨터는 이미 삭제된 컴퓨터이므로 넘어갑니다.

---

3. 소스코드

3.1 주석없는 코드

#include <vector>
using namespace std;
 
bool dfs(vector<vector<int>>& computers, int n) {
    if (!computers[n][n])    return false;
    computers[n][n] = 0;
    for (int i = 0; i < computers.size(); i++) {
        if (computers[n][i])    dfs(computers, i);
    }
    return true;
}
 
int solution(int n, vector<vector<int>> computers) {
    int answer = 0;
    for (int i = 0; i < n; i++) {
        if (computers[i][i] && dfs(computers, i))        answer++;
    }
    return answer;
}

3.2 주석있는 코드

#include <vector>
using namespace std;
 
bool dfs(vector<vector<int>>& computers, int n) {
    //이미 순회한 노드라면 리턴(재귀함수 탈출조건)
    if (!computers[n][n])    return false;
    //순회했다고 변경하기
    computers[n][n] = 0;
    //노드수만큼 반복
    for (int i = 0; i < computers.size(); i++) {
        //연결된 노드가 있다면 재귀
        if (computers[n][i])    dfs(computers, i);
    }
    return true;
}
 
int solution(int n, vector<vector<int>> computers) {
    int answer = 0;
    for (int i = 0; i < n; i++) {
        //순회하지 않은 노드라면 dfs탐색후 answer증가
        if (computers[i][i] && dfs(computers, i))
            answer++;
    }
    return answer;
}

3.3 테스트 코드 추가

#include <iostream>
#include <vector>
#include <string>
using namespace std;
 
bool dfs(vector<vector<int>>& computers, int n) {
    //이미 순회한 노드라면 리턴(재귀함수 탈출조건)
    if (!computers[n][n])    return false;
    //순회했다고 변경하기
    computers[n][n] = 0;
    //노드수만큼 반복
    for (int i = 0; i < computers.size(); i++) {
        //연결된 노드가 있다면 재귀
        if (computers[n][i])    dfs(computers, i);
    }
    return true;
}
 
int solution(int n, vector<vector<int>> computers) {
    int answer = 0;
    for (int i = 0; i < n; i++) {
        //순회하지 않은 노드라면 dfs탐색후 answer증가
        if (computers[i][i] && dfs(computers, i))
            answer++;
    }
    return answer;
}
 
void print(int n, vector<vector<int>> computers, int answer){
    int t = solution(n, computers);
    if (t == answer)    cout << "정답" << endl;
    else    cout << "틀림" << endl;
}
 
int main() {
    print(3, { {1,1,0}, {1,1,0}, {0,0,1} }, 2);
    print(3, { {1,1,0}, {1,1,1}, {0,1,1} }, 1);
    return 0;
}

---

4. 결과