一道简单的并查集的题题目,自己当初没有AC实在是不应该,有些细节处理的不到位,统计根的个数的时候并未走到最后,而是只统计了,他表面所储存的值。
下面给出AC代码:
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 1000 + 10;
int n,m,pather[maxn];
bool used[maxn];
int find(int x)
{
return pather[x]==x? x:pather[x]=find(pather[x]);
}
int main()
{
int T;
scanf("%d",&T);
while(T--)
{
scanf("%d %d",&n,&m);
memset(used,0,sizeof(used));
for(int i=0; i<=n; i++) pather[i]=i;
for(int i=0; i<m; i++)
{
int a,b; scanf("%d %d",&a,&b);
int ax=find(a),bx=find(b);
pather[bx]=ax;
}
int cnt=0;
for(int i=1;i<=n;i++)
{
int a=find(pather[i]);
if(!used[a])
{
cnt++;
used[a]=true;
}
}
printf("%d\n",cnt);
}
return 0;
}
上面是最朴素的做法,下面给出在计数方面稍微优化的点的方法
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 1000 + 10;
int n,m,pather[maxn];
bool used[maxn];
int find(int x)
{
return pather[x]==x? x:pather[x]=find(pather[x]);
}
int main()
{
int T;
scanf("%d",&T);
while(T--)
{
scanf("%d %d",&n,&m);
memset(used,0,sizeof(used));
for(int i=0; i<=n; i++) pather[i]=i;
for(int i=0; i<m; i++)
{
int a,b; scanf("%d %d",&a,&b);
int ax=find(a),bx=find(b);
pather[bx]=ax;
}
int cnt=0;
for(int i=1;i<=n;i++)
{
int a=find(pather[i]);
if(!used[a])
{
cnt++;
used[a]=true;
}
}
printf("%d\n",cnt);
}
return 0;
}
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 1000 + 10;
int n,m,pather[maxn];
bool used[maxn];
int find(int x)
{
return pather[x]==x? x:pather[x]=find(pather[x]);
}
int main()
{
int T;
scanf("%d",&T);
while(T--)
{
scanf("%d %d",&n,&m);
memset(used,0,sizeof(used));
for(int i=0; i<=n; i++) pather[i]=i;
int cnt=n;
for(int i=0; i<m; i++)
{
int a,b; scanf("%d %d",&a,&b);
int ax=find(a),bx=find(b);
if(ax!=bx)
{
pather[bx]=ax;
cnt--;
}
}
printf("%d\n",cnt);
}
return 0;
}