畅通工程(最小生成树)

首先要判断能否构成生成树,刚开始的思路是用hash遍历城市,看是否所有的城市都记录进取,被测试数据误导,如果给这组数据(1,2)(3,4)这么没办法构成单树。

#include<stdio.h>

#include<string.h>



const int INF=0x7fffffff;

const int MAXN=110;



int hash[MAXN];

int pre[MAXN];

int map[MAXN][MAXN];

int dist[MAXN];

int n,m,ans,flag;



void Prim()

{

    int i,j,k;

    int min;

    bool p[MAXN];

    for(i=2; i<=n; i++)

    {

        p[i]=false;

        dist[i]=map[1][i];

        pre[i]=1;

    }

    dist[1]=0;

    p[1]=true;

    for(i=1; i<=n-1; i++)

    {

        min=INF;

        k=0;

        for(j=1; j<=n; j++)

        {

            if(!p[j] && dist[j]<min)

            {

                min=dist[j];

                k=j;

            }

        }

        if(k==0)//判断是否有最小生成树

        {

            flag=1; return ;

        }

        p[k]=true;

        ans+=dist[k];

        for(j=1; j<=n; j++)

        {

            if(!p[j] && map[k][j]!=INF && dist[j]>map[k][j])

            {

                dist[j]=map[k][j];

                pre[j]=k;

            }

        }

    }

}



int main()

{

    int i,a,b,c,j;

    while(scanf("%d%d",&m,&n) && m)

    {

        //  memset(hash,0,sizeof(hash));

        for(i=1; i<=n; i++)

            for(j=1; j<=n; j++)

            {

                if(i==j) map[i][j]=0;

                else map[i][j]=INF;

            }

        for(i=1; i<=m; i++)

        {

            scanf("%d%d%d",&a,&b,&c);

            if(map[a][b]>c) map[a][b]=map[b][a]=c;

            //   hash[a]=hash[b]=1;

        }

        // for(i=1; i<=n; i++)

        //   if(hash[i]!=1) break;

        ans=flag=0;

        Prim();

        if(flag) printf("?\n");

        else  printf("%d\n",ans);

    }

    return 0;

}

你可能感兴趣的:(最小生成树)