Submission #7133543


Source Code Expand

Copy
#include <bits/stdc++.h>
using namespace std;
using Int = unsigned long long;
typedef pair<int,int> P;
typedef priority_queue<int> Pr;
int const MOD = 1000000007;
template<class T> inline bool chmax(T& a, T b) { if (a < b) { a = b; return 1; } return 0; }
template<class T> inline bool chmin(T& a, T b) { if (a > b) { a = b; return 1; } return 0; }

struct UnionFind {
  vector< int > data;
 
  UnionFind(int sz) {
    data.assign(sz, -1);
  }
 
  bool unite(int x, int y) {
    x = find(x), y = find(y);
    if(x == y) return (false);
    if(data[x] > data[y]) swap(x, y);
    data[x] += data[y];
    data[y] = x;
    return (true);
  }
 
  int find(int k) {
    if(data[k] < 0) return (k);
    return (data[k] = find(data[k]));
  }
 
  int size(int k) {
    return (-data[find(k)]);
  }
};

int main(){
    cin.tie(0); ios::sync_with_stdio(false);

    int n,q; cin>>n>>q;
    UnionFind u(n+1);
    for (int i = 0; i < q; i++){
        int p,a,b; cin>>p>>a>>b;
        if(p)cout<<(u.find(a)==u.find(b)?"Yes":"No")<<'\n';
        else u.unite(a,b);
    }
}

Submission Info

Submission Time
Task B - Union Find
User fellp1
Language C++14 (GCC 5.4.1)
Score 100
Code Size 1102 Byte
Status
Exec Time 53 ms
Memory 1280 KB

Judge Result

Set Name Score / Max Score Test Cases
Sample 0 / 0 00_sample_01.txt
All 100 / 100 00_sample_01.txt, subtask_01_01.txt, subtask_01_02.txt, subtask_01_03.txt, subtask_01_04.txt, subtask_01_05.txt, subtask_01_06.txt, subtask_01_07.txt, subtask_01_08.txt, subtask_01_09.txt, subtask_01_10.txt, subtask_01_11.txt, subtask_01_12.txt, subtask_01_13.txt, subtask_01_14.txt, subtask_01_15.txt, subtask_01_16.txt, subtask_01_17.txt, subtask_01_18.txt
Case Name Status Exec Time Memory
00_sample_01.txt 1 ms 256 KB
subtask_01_01.txt 31 ms 640 KB
subtask_01_02.txt 1 ms 640 KB
subtask_01_03.txt 46 ms 1024 KB
subtask_01_04.txt 53 ms 1280 KB
subtask_01_05.txt 4 ms 256 KB
subtask_01_06.txt 5 ms 640 KB
subtask_01_07.txt 49 ms 896 KB
subtask_01_08.txt 53 ms 1280 KB
subtask_01_09.txt 1 ms 256 KB
subtask_01_10.txt 1 ms 640 KB
subtask_01_11.txt 45 ms 896 KB
subtask_01_12.txt 53 ms 1280 KB
subtask_01_13.txt 40 ms 768 KB
subtask_01_14.txt 1 ms 640 KB
subtask_01_15.txt 46 ms 896 KB
subtask_01_16.txt 53 ms 1280 KB
subtask_01_17.txt 50 ms 1024 KB
subtask_01_18.txt 50 ms 1024 KB