/******************************************************************************
Welcome to GDB Online.
GDB online is an online compiler and debugger tool for C, C++, Python, Java, PHP, Ruby, Perl,
C#, OCaml, VB, Swift, Pascal, Fortran, Haskell, Objective-C, Assembly, HTML, CSS, JS, SQLite, Prolog.
Code, Compile, Run and Debug online from anywhere in world.
*******************************************************************************/
#include <bits/stdc++.h>
using namespace std;
class DisjointSet
{
public:
vector<int>parent;
vector<int>size;
DisjointSet(int N)
{
parent.resize(N+1);
size.resize(N+1,1);
for(int i=0;i<=N;i++)
parent[i]=i;
}
int fup(int x)
{
if(x==parent[x])return x;
return parent[x]=fup(parent[x]);
}
void ubs(int u,int v)
{
int upu=fup(u);
int upv=fup(v);
if(upu==upv)return ;
if(size[upu]<size[upv])
{
parent[upu]=upv;
size[upv]=size[upv]+size[upu];
}
else
{
parent[upv]=upu;
size[upu]=size[upu]+size[upv];
}
}
};
int main()
{
int n=5;
int edges=4;
vector<int>sol;
vector<int>cof{2,2,1,1};
vector<int>cot{1,3,3,4};
vector<int>q{4,2,5};
DisjointSet ds(n);
for(int i=0;i<edges;i++)
{
ds.ubs(cof[i],cot[i]);
}
for(int i=0;i<q.size();i++)
{
int par=ds.fup(q[i]);
sol.push_back(ds.size[par]);
}
for(auto&x:sol)cout<<x<<" ";
return 0;
}