hdu1102 最小生成树

来源:互联网 发布:sql语言select语句 编辑:程序博客网 时间:2024/05/16 15:35
#include<stdio.h>
#include<string.h>
int cost[105][105];
int vis[105];
int lowcost[105];
int n;
const int min=20000;
int prim()
{
    memset(vis,0,sizeof(vis));
    int min1=min;
    for(int i=0;i<n;i++)
    lowcost[i]=cost[0][i];
    int k;
    int sum=0;
    vis[0]=1;
    for(int i=1;i<n;i++)
    { min1=min;
        for(int j=0;j<n;j++)
        {
            if(vis[j]==0&&min1>lowcost[j])
            {
                min1=lowcost[j];
                k=j;
            }
        }
        sum+=min1;
        vis[k]=1;
        for(int j=0;j<n;j++)
        {
            if(vis[j]==0&&lowcost[j]>cost[k][j])
            lowcost[j]=cost[k][j];
        }
    }
    return sum;
}
int main()
{
    int q,a,b;
    while(scanf("%d",&n)==1)
    {
        for(int i=0;i<n;i++)
    for(int j=0;j<n;j++)
    {
        scanf("%d",&cost[i][j]);
        if(i==j)
        cost[i][j]=min;
    }
        scanf("%d",&q);
    for(int i=0;i<q;i++)
    {
        scanf("%d%d",&a,&b);
        cost[a-1][b-1]=cost[b-1][a-1]=0;
    }
    int min2=prim();
    printf("%d\n",min2);
    }
    return 0;
}
0 0