Cod sursa(job #3267089)

Utilizator BogdanBurescuBogdan Burescu BogdanBurescu Data 11 ianuarie 2025 09:30:07
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.71 kb
#include <fstream>
#include <queue>
#include <string>
#include <vector>
#include <map>
#include <set>
#include <algorithm>
#include <cmath>

using namespace std;

ifstream cin ("disjoint.in");
ofstream cout ("disjoint.out");

long long int t[200005],m,c,u,v,curr,vec,x,y,ok,mask,cost,maxi,b,poz,pozc,n,k,i,j,sp[200005],a[200005],l[200005],r[200005],ll,rr,mij,mini,p,ans,nr,sum,mod=1e9+7;
string s;
//vector<int>g[200005];
queue<int>q;


struct graf
{
    int x,y;
};

vector<int> parent,rnk;

void union_find(int n)
{
    parent.resize(n);
    rnk.resize(n,0);
    for(int i=0; i<n; i++)
        parent[i]=i;
}

int findd(int u)
{
    if(parent[u]!=u)
        parent[u]=findd(parent[u]);
    return parent[u];
}

void unite(int u, int v)
{
    int rootu=findd(u);
    int rootv=findd(v);
    if(rootu!=rootv)
    {
        if(rnk[rootu]>rnk[rootv])
            parent[rootv]=rootu;
        else if(rnk[rootu]<rnk[rootv])
            parent[rootu]=rootv;
        else
        {
            parent[rootv]=rootu;
            rnk[rootu]++;
        }
    }
}

int main()
{
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
    cin>>n>>m;
    vector<graf>g(m);
    for(i=0; i<m; i++)
    {
        cin>>t[i];
        cin>>g[i].x>>g[i].y;
        g[i].x--;
        g[i].y--;
    }

    union_find(n);

    for(i=0; i<m; i++)
    {
        u=g[i].x;
        v=g[i].y;
        if(t[i]==1)
        {
            if(findd(u)!=findd(v))
                unite(u, v);
        }
        else
        {
            if(findd(u)==findd(v))
                cout<<"DA"<<'\n';
            else
                cout<<"NU"<<'\n';
        }
    }
    return 0;
}