HDU 1213 并查集

#include<iostream>
using namespace std;

const int MaxN=1001;
int P[MaxN],size[MaxN];

void InitSets(int n)
{
for(int i=0;i<=n;i++)
{
P[i]=i;
size[i]=1;
}
}
int FindSets(int x)
{
int y=x;
while(P[x]!=x)x=P[x];
while(y!=x)
{
int t=P[y];
P[y]=x;
y=t;
}
return x;
}
bool Union(int x,int y)
{
int fx,fy;
fx=FindSets(x);
fy=FindSets(y);
if(fx==fy)return false;
if(size[fx]>=size[fy])
{
P[fy]=fx;
size[fx]+=size[fy];
}
else
{
P[fx]=fy;
size[fy]+=size[fx];
}
return true;
}
int main()
{
int T;
cin>>T;
for(int x=0;x<T;x++)
{
int N,M,i;
cin>>N>>M;
InitSets(N);
for(i=0;i<M;i++)
{
int a,b;
cin>>a>>b;
Union(a,b);
}
int cnt=0;
for(i=1;i<=N;i++)
{
if(P[i]==i)cnt++;
}
cout<<cnt<<endl;
}
return 0;
}

优化——路径压缩

思想：每次查找的时候，如果路径较长，则修改信息，以便下次查找的时候速度更快

步骤：第一步，找到根节点；第二步，修改查找路径上的所有节点，讲他们都指向根节点