HDU 1102 Constructing Roads（Prim求最小生成树）

题目地址：点击打开链接

思路：只要把已经建立的路之间的距离设为0，就变成最小生成树的模板题

AC代码：

#include<stdio.h>
#include<string.h>
int main()
{
	int n,q,i,j,a[110][110],visit[110],lowcost[110],min,k,sum;
	int x,y;
	while(scanf("%d",&n)!=EOF)
	{
		sum=0;
		memset(a,0,sizeof(a));
		memset(visit,0,sizeof(visit));
		for(i=1;i<=n;i++)
		{
			for(j=1;j<=n;j++)
			{
				scanf("%d",&a[i][j]);
			}
		}
		scanf("%d",&q);
		for(i=1;i<=q;i++)
		{
			scanf("%d%d",&x,&y);
			a[x][y] = 0;
			a[y][x] = 0;
		}
		for(i=1;i<=n;i++)
		{
		    lowcost[i] = a[1][i];
		}
        visit[1] = 1;
        for(i=1; i<n; i++)
        {
            min = 10000000;
            for(j=1; j<=n; j++)
            {
                if(visit[j] == 0 && lowcost[j] < min)
                {
                    min = lowcost[j];
                    k = j;
                }
            }
            visit[k] = 1;
            sum += min;
            for(j=1; j<=n; j++)
            {
                if(visit[j] == 0 && a[k][j] < lowcost[j])
                    lowcost[j] = a[k][j];
            }
        }
        printf("%d\n",sum);
	}
	return 0;
}