POJ 1258 Agri-Net

標籤：style   blog   io   2014   for   re   

題意：就是要你求最小產生樹所需的最小距離

思路：Kruskal演算法

AC代碼：

#include<stdio.h>#include<string.h>#include<algorithm>using namespace std;#define N 125000int u[N],v[N],w[N],r[N];int f[520],str[520][520];int find(int x){    if(x!=f[x])        f[x]=find(f[x]);    return f[x];}int cmp(const int x,const int y)    //間接排序距離{    return w[x]<w[y];}int main(){    int x,y,sum,e,i,j,n;    while(scanf("%d",&n)!=EOF)    {        int cnt=0;        memset(str,0,sizeof(str));        for(i=1;i<=n;i++)            for(j=1;j<=n;j++)            {                int val;                scanf("%d",&val);                if(val==0)                    continue;                if(!(str[i][j]&&str[j][i]))                {                    u[cnt]=i;                    v[cnt]=j;                    w[cnt++]=val;                    str[i][j]=str[j][i]=1;                }            }        for(i=0;i<=n;i++)f[i]=i;        for(i=0;i<=cnt;i++)r[i]=i;        sort(r,r+cnt,cmp);        sum=0;        for(i=0;i<=cnt;i++)        {            e=r[i];            x=find(u[e]);            y=find(v[e]);            if(x!=y){                sum+=w[e];                f[x]=y;            }        }        printf("%d\n",sum);    }}