[c++] 백준 1717 - 집합의 표현

0. [c++] 백준  -

https://www.acmicpc.net/problem/1717

1. 풀이

2. 소스코드

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55

#include<iostream> #include<vector> #include<algorithm> #include<queue> #include<string>   using namespace std;   const int INF = 1000000007; int N, M; int arr[1000001];   int find(int x) {     if (x == arr[x])         return x;     else         return arr[x] = find(arr[x]); }   int main() {     ios_base::sync_with_stdio(0);     cin.tie(0);       cin >> N >> M;       for (int i = 0; i <= N; i++)         arr[i] = i;       int op, a, b;       for (int i = 0; i < M; i++) {         cin >> op >> a >> b;         // 집합 합치기         if (op == 0) {             a = find(a);             b = find(b);             if (a > b)                  arr[a] = b;             else                  arr[b] = a;         }         // 포함관계 출력         else {             a = find(a);             b = find(b);             if (a == b)                 cout << "YES\n";             else                 cout << "NO\n";         }     }         return 0; }

cs

3. 참고

구종만, 「프로그래밍 대회에서 배우는 알고리즘 문제 해결 전략」, 인사이트, 2012, p.216~236.

질문이나 지적 있으시면 댓글로 남겨주세요~

도움 되셨으면 하트 꾹!