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

(6/24 - O)

 

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

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

이런 테스트케이스에서 막히고, 또 저런 테스트케이스에서 막혀서 푸는 데 너무 오래 걸린 문제.. dfs/bfs의 다양한 문제들을 풀어 보지 않아서 그런가 보다. 응용을 빠르게 하지 못 하는 것 같다.

 

구현

방문하지 않은 모든 노드를 방문한다. 방문 처리를 하고, 해당 노드와 이어진 노드를 방문한다.

 

코드

// 프로그래머스 고득점 kit - 네트워크
// 6/24 - O

#include <string>
#include <vector>

using namespace std;
bool visited[201];
int answer = 0;
void dfs(vector<vector<int>>& v, int size, int start, int& cnt){
    if (start >= size) return;
    if (visited[start] == true) return;
    visited[start] = true;
    for (int i = 0; i < size; i++) {
        if (v[start][i] == 1) {
            dfs(v, size, i, cnt);
        }
    }
    cnt++;
}

int solution(int n, vector<vector<int>> computers) {
    int answer = 0;

    for (int i = 0; i < n; i++) {
        int cnt = 0;
        dfs(computers, n, i, cnt);
        if (cnt > 0) {
            answer++;
        }
    }

    return answer;
}

int main() {
    vector<vector<int>> v;

    //v.push_back({ 1,0, 0, 1 });
    //v.push_back({ 0, 1, 1, 0 });
    //v.push_back({ 0, 1, 1, 0 });
    //v.push_back({ 1,1, 0, 1 });

    //v.push_back({ 1,0, 0 });
    //v.push_back({ 0, 1, 0 });
    //v.push_back({ 0, 0, 1 });

    v.push_back({ 1, 1, 0 });
    v.push_back({ 1, 1, 0 });
    v.push_back({ 0, 0, 1 });


    //v.push_back({ 1, 1, 0 });
    //v.push_back({ 1, 1, 1 });
    //v.push_back({ 0, 1, 1 });

    //v.push_back({ 1, 0, 1, 1, 0, 0 });
    //v.push_back({ 0, 1, 0, 0, 1, 1 });
    //v.push_back({ 1, 0, 1, 1, 1, 1 });
    //v.push_back({ 1, 0, 1, 1, 1, 1 });
    //v.push_back({ 0, 1, 1, 1, 1, 1 });
    //v.push_back({ 0, 1, 1, 1, 1, 1 });

    solution(3, v);
}