畅通工程(最小生成树)

首先要判断能否构成生成树,刚开始的思路是用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;
}
posted @ 2012-08-07 20:21  calmound  阅读(287)  评论(1编辑  收藏  举报