今天做了一道并查集的最小生成树的问题,刚开始wa,后来和学长谈了以下,才发现自己学的那么差,路径压缩没有,sort函数中的cmp函数还没有完全掌握,qsort函数也没有完全掌握,并的操作不规范,虽然也可以并成一类,但头结点不一定是最小的,而且set[x]与x的大小关系也不能确定,原来这么差劲呀,看来还是学业未精呀......
下面是正确代码,实现了路径压缩和大并小的操作:
#include <stdio.h>
#include <stdlib.h>
#include <algorithm>
using namespace std;
#define Max 510
struct Por
{
int from;
int to;
int value;
}node[Max*Max];
int set[Max];
int find(int x);
bool Merge(int a,int b);
int t,n;
int ans;
int cmp(const void *a,const void *b){
return ((Por*)a)->value - ((Por*)b)->value;
}
int main()
{
scanf("%d",&t);
while(t--){
scanf("%d",&n);
int i,j,k=0;
for(i=0;i<=n;i++)
set[i]=i;
for(i=1;i<=n;i++){
for(j=1;j<=n;j++){
scanf("%d",&node[k].value);
node[k].from=i;
node[k++].to=j;
}
}
qsort(node,k,sizeof(Por),cmp);
for(i=0;i<k;i++){
if(Merge(node[i].from,node[i].to)){
ans=node[i].value;
}
}
printf("%d\n",ans);
}
return 0;
}
int find(int x){
if(x==set[x])
return x=set[x];
set[x]=find(set[x]);
return set[x];
}
bool Merge(int a,int b){
int x=find(a);
int y=find(b);
if(x!=y){
if(x < y)
set[y]=x;
else
set[x]=y;
return true;
}
return false;
}