SDUTACM 图结构练习——最小生成树

た入场券 2022-09-25 07:22 141阅读 0赞

## 题目描述 ##

有n个城市，其中有些城市之间可以修建公路，修建不同的公路费用是不同的。现在我们想知道，最少花多少钱修公路可以将所有的城市连在一起，使在任意一城市出发，可以到达其他任意的城市。

## 输入 ##

输入包含多组数据，格式如下。

第一行包括两个整数n m，代表城市个数和可以修建的公路个数。(n <= 100， m <=10000)

剩下m行每行3个正整数a b c，代表城市a 和城市b之间可以修建一条公路，代价为c。

## 输出 ##

每组输出占一行，仅输出最小花费。

## 示例输入 ##

3 2
    1 2 1
    1 3 1
    1 0

## 示例输出 ##

2
    0

## 提示 ##

#include<stdio.h>
    #include<stdlib.h>
    #include<string.h>
    #define INF 100000;
    int k[101][101],lowc[10001],vis[10001];
    int prim(int n)
    {
            int i,j,p;
    	int min,sum=0;
    	memset(vis,0,sizeof(vis));
    	vis[0]=1;
    	for(i=1;i<n;i++)
    		lowc[i]=k[0][i];
    	for(i=1;i<n;i++)
    	{
                    min=INF;
    		p=-1;
    		for(j=0;j<n;j++)
    		{
    			if(vis[j]==0&&min>lowc[j])
    			{
    				min=lowc[j];
                                    p=j;
    			}
    		}
    		sum=sum+min;
    	        vis[p]=1;
    	        for(j=0;j<n;j++)
    		{
    		    if(vis[j]==0&&lowc[j]>k[p][j])
    			    lowc[j]=k[p][j];
    		}
    	}
    	return sum;
    }
    int main()
    {
    	int n,m,i,j,a,b,w;
    	while(scanf("%d%d",&n,&m)!=EOF)
    	{

for(i=0;i<n;i++)
    		{
    			for(j=0;j<n;j++)
    			{
    				if(i==j)
                                            k[i][j]=0;
    				else
    					k[i][j]=INF;
    			}
    		}
    		for(i=0;i<m;i++)
    		{
    			scanf("%d%d",&a,&b);
    			scanf("%d",&w);
    			if(k[a-1][b-1]>w)
    			{
    				k[a-1][b-1]=w;
    				k[b-1][a-1]=w;
    			}
    		}
    		printf("%d\n",prim(n));
    	}
    }