poj1251

来源:互联网 发布:笨办法学python英文版 编辑:程序博客网 时间:2024/06/01 12:49

给出一个无向图,求最小生成树

#include<stdio.h>
#include<iostream>
#include<String.h>
using namespace std;
int map[28][28],n,a[28],vis[28];
void prim()
{
    int i,j,min,min_f;
    int sum=0;
    for(i=1;i<=n;i++)
    {
        a[i]=map[1][i];
        vis[i]=0;
    }
    vis[1]=1;
    for(i=1; i<n; i++)
    {
        min=100000000;
        min_f=2;
        for(j=1; j<=n; j++)
        {
            if(!vis[j]&&a[j]<min)
            {
                min_f=j;
                min=a[j];
            }
        }
        sum+=a[min_f];
        vis[min_f]=1;
        for(j=2; j<=n; j++)
        {
            if(!vis[j]&&a[j]>map[min_f][j])
                a[j]=map[min_f][j];
        }
    }
    cout<<sum<<endl;
}
int main()
{
    int i,j,m,aa,s;
    char d;
    while(scanf("%d",&n)&&n!=0)
    {
        getchar();
        for(i=1; i<=n; i++)
        {
            for(j=1; j<=n; j++)
            {
                map[i][j]=100000000;
            }
        }
        for(i=1; i<n; i++)
        {
            cin>>d>>m;
            for(j=0; j<m; j++)
            {
                cin>>d>>aa;
                s=d-'A'+1;
                map[i][s]=map[s][i]=aa;
            }
        }
        prim();
    }
    return 0;