본문 바로가기

Algorithm_BOJ(백준)/유니온파인드(UnionFind)

[백준 7511 c++ O] 소셜 네트워킹 어플리케이션

728x90
반응형

- 풀이 링크:

https://github.com/xhaktmchl/Algorithm_study/blob/main/BOJ/%EC%9C%A0%EB%8B%88%EC%98%A8%ED%8C%8C%EC%9D%B8%EB%93%9C(UnionFind)/%5B%EB%B0%B1%EC%A4%80%207511%20c%2B%2B%20O%5D%20%EC%86%8C%EC%85%9C%20%EB%84%A4%ED%8A%B8%EC%9B%8C%ED%82%B9%20%EC%96%B4%ED%94%8C%EB%A6%AC%EC%BC%80%EC%9D%B4%EC%85%98.cpp 

 

GitHub - xhaktmchl/Algorithm_study: 알고리즘 이론 및 문제풀이

알고리즘 이론 및 문제풀이. Contribute to xhaktmchl/Algorithm_study development by creating an account on GitHub.

github.com

#include <iostream>
#include <algorithm> // fill_n, min,max, swap
//#include <map> // 중복 x 
//#include <string> // getline
//#include <cstring> // memset, strok, strstr
//#include <vector>
//#include <queue>
//#include <set> // 트리, 중복 x
//#include <unordered_set>
//#include <cmath>
using namespace std;
/*
[백준 7511 c++ O] 소셜 네트워킹 어플리케이션
접근: bfs: 간선을 추가하고 bfs로 친구인지 아닌지 확인 -> 시간초과 예상
접근2: 유니온파인드: 간선추가하면서 유니온 -> 파인드로 같은 집합인지 확인
시간복잡도:
풀이1:
    //1.테케, 입력
    //2.유니온:친구 간선 추가
    //부모님 배열 초기화
    //같은 집합 아니면 유니온
    //3.파인드: 같은 집합이면 1, 아니면 0
*/
int T,n, uCnt,fCnt,n1,n2;
int par[1000000 + 5];

void init() {
    for (int i = 0; i < n; i++) par[i] = i;
}

int fin(int node) {
    if (par[node] == node) return node;
    else return par[node] = fin(par[node]); // 경로 압축: 시간단축
}

void uni(int a, int b) {
    int rootA = fin(a);
    int rootB = fin(b);
    par[rootA] = rootB;
}

int main()
{
    ios::sync_with_stdio(false); cin.tie(nullptr); cout.tie(0);
    //1.테케, 입력
    cin >> T;
    for (int t = 1; t <= T; t++) {
        //2.유니온:친구 간선 추가
        //부모님 배열 초기화
        cin >> n;
        init();
        
        //같은 집합 아니면 유니온
        cin >> uCnt;
        for (int i = 0; i < uCnt; i++) {
            cin >> n1 >> n2;
            if (fin(n1) != fin(n2)) uni(n1, n2);
        }
        //3.파인드: 같은 집합이면 1, 아니면 0
        cin >> fCnt;
        cout << "Scenario "<<t<<":" << '\n';
        for (int i = 0; i < fCnt; i++) {
            cin >> n1 >> n2;
            if (fin(n1) == fin(n2)) cout << 1 << '\n';
            else cout << 0 << '\n';
        }cout << '\n';
    }
    return 0;
}
반응형