[프로그래머스] 네트워크 c++

출처: 프로그래머스

https://programmers.co.kr/learn/courses/30/lessons/43162?language=cpp 

코딩테스트 연습 - 네트워크

#include <string>
#include <vector>
using namespace std;
bool visited[201]; //방문검사

void dfs(vector<vector<int>> &arr, int idx){
    visited[idx] = true;
    for(int i=0; i<arr[idx].size(); i++){
        if(arr[idx][i] == 1 && visited[i] == false){ //1이고 아직 방문하지 않은 컴
            // arr[idx][i] = 0;
            // visited[i] = true;
            dfs(arr, i); //dfs 진행
        }
    }
}

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

dfs로 풀이했다.

 

연결되어 있는 것이 1로 표시되어 있다.

dfs로 computers 벡터를 돌면서 인접한 1을 찾고 dfs진행이 끝나면 인접한 1 찾는 과정이 끝난 것이다.

끝날 때마다 answer++해준다.