藍橋杯歷屆試題-合根植物是一道簡單的並查集問題,並查集算法的核心是把兩個點連通起來的函數和找到某個點的根節點的函數,本文把這兩個函數定義成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;
}