int n, e;
cin >> n >> e;
dsu_initialize(n);
bool cycle = false;
প্রথমে number of nodes এবং number of edges ইনপুট নিয়া হয়েছে, এরপর dsu_initialize function কে কল করে parent array এবং group_size array কে যথাক্রমে -1 এবং 1 দিয়ে initialize করা হয়েছে।
while (e--)
{
int a, b;
cin >> a >> b;
int leaderA = dsu_find(a);
int leaderB = dsu_find(b);
if (leaderA == leaderB)
{
cycle = true;
}
else
{
dsu_union_by_size(a, b);
}
}
এরপর number of edges বার লুপ চালিয়ে edge গুলোকে ইনপুট নিয়া হয়েছে এবং সেইখানে নোডগুলোর leader কে বের করা হয়েছে dsu_find function কে কল করে , যদি নোড ২টির leader মিলে যায় , তাহলে cycle detect করা হবে, অন্যথায় dsu_union_by_size funciton কে কল করে নোড ২টি কে union করা হবে।
dsu_find এবং dsu_union_by_size function কিভাবে কাজ করে সেইটি আমরা আগের মডিউলে দেখেছি।
সম্পূর্ণ কোড
#include <bits/stdc++.h>
using namespace std;
const int N = 1e5 + 5;
int parent[N];
int group_size[N];
void dsu_initialize(int n)
{
for (int i = 0; i < n; i++)
{
parent[i] = -1;
group_size[i] = 1;
}
}
int dsu_find(int node)
{
if (parent[node] == -1)
return node;
int leader = dsu_find(parent[node]);
parent[node] = leader;
return leader;
}
void dsu_union_by_size(int node1, int node2)
{
int leaderA = dsu_find(node1);
int leaderB = dsu_find(node2);
if (group_size[leaderA] > group_size[leaderB])
{
parent[leaderB] = leaderA;
group_size[leaderA] += group_size[leaderB];
}
else
{
parent[leaderA] = leaderB;
group_size[leaderB] += group_size[leaderA];
}
}
int main()
{
int n, e;
cin >> n >> e;
dsu_initialize(n);
bool cycle = false;
while (e--)
{
int a, b;
cin >> a >> b;
int leaderA = dsu_find(a);
int leaderB = dsu_find(b);
if (leaderA == leaderB)
{
cycle = true;
}
else
{
dsu_union_by_size(a, b);
}
}
if (cycle)
{
cout << "Cycle found" << endl;
}
else
{
cout << "Cycle not found" << endl;
}
return 0;
}