最小生成树，并查集的思想 nyoj1239

来源：互联网发布：便利店文化知乎编辑：程序博客网时间：2024/04/30 09:41

#include<stdio.h>#include<string.h>#include<algorithm>using namespace std;int n,pre[10010],cnt;struct node{    int x,y;    int val;} p[10010];void init(){    for(int i=0; i<10010; i++)        pre[i]=i;}int cmp(node s1,node s2){    return s1.val<s2.val;}int find(int x){    return x==pre[x]?x:find(pre[x]);}int join(int x,int y){    int fx=find(x);    int fy=find(y);    if(fx!=fy)    {        pre[fx]=fy;        return 1;    }    return 0;}int main(){    int t;    scanf("%d",&t);    while(t--)    {        init();        cnt=0;        scanf("%d",&n);        for(int i=0; i<=n; i++)        {            for(int j=1; j<=n; j++)            {                scanf("%d",&p[cnt].val);                p[cnt].x=i,p[cnt++].y=j;            }        }        sort(p,p+cnt,cmp);        int sum=0;        for(int i=0; i<cnt; i++)        {            if(join(p[i].x,p[i].y))                sum+=p[i].val;        }        printf("%d\n",sum);    }    return 0;}

0 0