hdu 1879 继续畅通工程

题目：http://acm.hdu.edu.cn/showproblem.php?pid=1879

还是最小生成树。。。一开始没想到。。瞎改模版。。

然后细想才发现。。

如果已经修的路就把这两点的权值改成0就 OK了。。

下面是AC代码：

#include<iostream>
using namespace std;
#define N 9999999
int map[101][101];
int mark[101];
bool visited[101];
int n;
int prim()
{
	int i,j,k,sum=0,min;

	memset(mark,0,sizeof(mark));
	memset(visited,0,sizeof(visited));


	k=1;
	visited[k]=1;

	for(i=2;i<=n;i++)
	{
		mark[i]=map[k][i];               //计算k到各顶点的距离
	}

	min=N;

	for(i=1;i<=n-1;i++)                  
	{
		for(j=1;j<=n;j++)
			if(mark[j]<min&&!visited[j])
			{
				min=mark[j];
				k=j;
			}

		visited[k]=1;

		for(j=1;j<=n;j++)
		{
			if(mark[j]>map[k][j]&&!visited[j])
				mark[j]=map[k][j];
		}
	
		sum+=min;
		min=N;
	}

	return sum;

}
int main()
{
	int i,j;
	int a,b,s,x;
	while(cin>>n&&n)
	{
		for(i=1;i<=n;i++)
			for(j=1;j<=n;j++)
			{
				if(i==j)
				{
					map[i][j]=0;	
				}
				else
				{
					map[i][j]=N;
				}
			}

		for(i=1;i<=n*(n-1)/2;i++)
		{
			scanf("%d%d%d%d",&a,&b,&s,&x);
			if(x==0)
			{
				map[a][b]=s;
			    map[b][a]=s;
			}
			else
			{
				map[a][b]=0;
				map[b][a]=0;
			}
		}

		printf("%d\n",prim());
	}
	return 0;
}