藍橋杯歷屆試題-合根植物(並查集)

藍橋杯歷屆試題-合根植物是一道簡單的並查集問題,並查集算法的核心是把兩個點連通起來的函數和找到某個點的根節點的函數,本文把這兩個函數定義成join()find();
先寫出這兩個函數的原型:

int find(int x)
{
     return a[x]==x?x:a[x]=find(a[x]);//返回節點x的根節點,並且壓縮路徑
}

//如果看不懂壓縮路徑的寫法可以看下面這個
int find(int x)
{
     while(a[x]!=x)
     {
         x=a[x];
     }
     return x;//返回x節點的根節點
}

void join(int x,int y)
{
    int fx=find(x);
    int fy=find(y);
    if(fx!=fy)
     	a[fx]=fy;//讓fx的根節點變爲fy
}

最後寫出這道題的解法:

#include<bits/stdc++.h>
using namespace std;
int a[1000005];
int count=0;
int find(int x)
{
     return a[x]==x?x:a[x]=find(a[x]);
}
void join(int x,int y)
{
    int fx=find(x);
    int fy=find(y);
    if(fx!=fy)
    {
           a[fx]=fy;
           count--;
    }
}
int main()
{
    int i;
    int m,n,c;
    scanf("%d%d",&m,&n);
    c=m*n;
    count=c;
    for(i=1;i<=c;i++)
     {
         a[i]=i;
     }
    int b;
    scanf("%d",&b);
    while(b--)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        join(x,y);
    }
    printf("%d\n",count);
    return 0;
}

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章