DSU.cpp

#include<bits/stdc++.h>

#define  pf                  printf
#define  sc                  scanf
#define  PI                  2*acos(0.0)
#define  fast                ios_base::sync_with_stdio(false);cin.tie(0); cout.tie(0)
#define  mem(a,b)            memset(a, b, sizeof(a) )
#define  gcd(a,b)            __gcd(a,b)
#define  lcm(a,b)            (a*(b/gcd(a,b)))
#define  MAX                 1234567899
#define  mod                 1000000007
#define  all(v)              (v).begin(),(v).end()
#define  vSort(v)            sort(all(v))
#define  maxSort(v)          sort(all(v),greater<int>())
#define  Unique(v)           v.erase(unique(all(v)),v.end())
#define  sqr(x)              ((x)*(x))
#define  qube(x)             ((x)*(x)*(x))
#define  deci(n)             cout<<fixed<<setprecision(n)
#define  sci(n)              sc("%d",&(n))
#define  scii(x,y)           sc("%d %d",&(x),&(y))
#define  scl(x)              sc("%lld",&(x))
#define  scll(x,y)           sc("%lld %lld",&(x),&(y))
#define  vi                  vector<int>
#define  hi                  pf(" HI \n")
#define  pp                  pair<int,int>
#define  pb(x)               push_back(x)
#define  mp(x,y)             make_pair(x,y)
#define  F                   first
#define  S                   second
#define  check(n,pos)        (n & (1<<(pos)))
#define  biton(n,pos)        (n  (1<<(pos)))
#define  bitoff(n,pos)       (n & ~(1<<(pos)))
typedef  long long int       lli;
typedef  unsigned long long int ulli;
int dx4[] = { 0, 0, -1, 1 } ;
int dy4[] = { -1, 1, 0, 0 } ;
bool valid( int r, int c, int x, int y )
{
    if( x >= 1 && x <= r && y >= 1 && y <= c ) return 1 ;
    return 0 ;
}
using namespace std;
int a[2000000];
int Size[200000];
void initialize(int n )
{
    for(int i=1; i<=n; i++)
    {
        a[i]=i;
        Size[i]=1;
    }
}
int Root(int i)
{
    while(a[i]!=i)
    {
        a[i]=a[a[i]];
        i=a[i];
    }
    return i;
}
void Union(int x,int y)
{
    int roota = Root(x);
    int rootb = Root(y);
    if(roota!=rootb)
    {
        if(Size[roota]<Size[rootb])
        {
            a[roota] = a[rootb];
            Size[rootb]+=Size[roota];
        }
        else
        {
            a[rootb]=a[roota];
            Size[roota]+=Size[rootb];
        }
    }
}
bool Find(int x,int y)
{
    if(Root(x)==Root(y))
        return true;
    else
        false;
}
int main ()
{
    initialize(100000);
    int m;
    int x,y;
    cin >> m;
    for(int i=1; i<=m; i++)
    {
        cin >> x >> y;
        Union(x,y);
    }
    for(int i=1; i<=m; i++)
    {
        cin >> x>> y;
        bool flag= Find(x,y);
        if(flag)
            pf("YES\n");
        else
            pf("NO\n");
    }
    return 0;
}