মডিউল ১১_২ঃ DSU দিয়ে Cycle Detection Undirected গ্রাফে ইমপ্লিমেন্টেশন

DSU দিয়ে Cycle Detection Undirected গ্রাফে ইমপ্লিমেন্টেশন

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;
}