HDU1879 继续畅通工程（Prim算法）

Problem Description

省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通（但不一定有直接的公路相连，只要能间接通过公路可达即可）。现得到城镇道路统计表，表中列出了任意两城镇间修建道路的费用，以及该道路是否已经修通的状态。现请你编写程序，计算出全省畅通需要的最低成本。

 

Input

测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( 1< N < 100 )；随后的 N(N-1)/2 行对应村庄间道路的成本及修建状态，每行给4个正整数，分别是两个村庄的编号（从1编号到N），此两村庄间道路的成本，以及修建状态：1表示已建，0表示未建。

当N为0时输入结束。

 

Output

每个测试用例的输出占一行，输出全省畅通需要的最低成本。

 

Sample Input

  
  

   
   3
1 2 1 0
1 3 2 0
2 3 4 0
3
1 2 1 0
1 3 2 0
2 3 4 1
3
1 2 1 0
1 3 2 1
2 3 4 1
0
  
  

 

Sample Output

  
  

   
   3
1
0
  
  

 
在讲这道题和Prim算法前，我们要先搞清楚最小生成树的概念。
要搞清楚最小生成树的概念就先要搞清楚生成树的概念。
生成树：若一个连通图有n个结点，它的生成树是有n-1条边的连通图，即包含其所有顶点，并有保持图连通的最少的边。
图的生成树不唯一。从不同的顶点出发进行遍历，可以得到不同的生成树。
最小生成树即边权值之和最小的生成树。
求最小生成树通常用Kruskal算法或Prim算法。

下面Prim算法的图解来自别人的博客。 http://www.cnblogs.com/biyeymyhjob/archive/2012/07/30/2615542.html

图例	说明	不可选	可选	已选（Vnew）
	此为原始的加权连通图。每条边一侧的数字代表其权值。	-	-	-
	顶点D被任意选为起始点。顶点A、B、E和F通过单条边与D相连。A是距离D最近的顶点，因此将A及对应边AD以高亮表示。	C, G	A, B, E, F	D
	下一个顶点为距离D或A最近的顶点。B距D为9，距A为7，E为15，F为6。因此，F距D或A最近，因此将顶点F与相应边DF以高亮表示。	C, G	B, E, F	A, D
	算法继续重复上面的步骤。距离A为7的顶点B被高亮表示。	C	B, E, G	A, D, F
	在当前情况下，可以在C、E与G间进行选择。C距B为8，E距B为7，G距F为11。E最近，因此将顶点E与相应边BE高亮表示。	无	C, E, G	A, D, F, B
	这里，可供选择的顶点只有C和G。C距E为5，G距E为9，故选取C，并与边EC一同高亮表示。	无	C, G	A, D, F, B, E
	顶点G是唯一剩下的顶点，它距F为11，距E为9，E最近，故高亮表示G及相应边EG。	无	G	A, D, F, B, E, C
	现在，所有顶点均已被选取，图中绿色部分即为连通图的最小生成树。在此例中，最小生成树的权值之和为39。	无	无	A, D, F, B, E, C, G

 

以下是完整的代码，模板来自kuangbing大神。
因为这道题里有已经修通的路，那么把这些边的权值设为0就可以了。

#include<iostream>
#include<cstring>
using namespace std;
const int MAXN=110;
const int INF=0x3f3f3f3f;
int vis[MAXN];
int lowc[MAXN];
int cost[MAXN][MAXN];
int Prim(int cost[][MAXN],int n)
{
	int ans=0;
	memset(vis,0,sizeof(vis));
	vis[1]=1;                    //把1作为起点
	for(int i=1;i<=n;i++)
		lowc[i]=cost[1][i];      //存放所有点与1点的边的权值，没有边的权值为INF
	for(int i=1;i<n;i++)         //寻找n-1条边
	{	
		int minc=INF;
		int p=-1;
		for(int j=1;j<=n;j++)
			if(!vis[j]&&minc>lowc[j])
			{
				minc=lowc[j];
				p=j;                    
			}
	//	if(minc==INF)        //原图不连通      
	//			return -1;
		ans+=minc;           //最小权值累加
		vis[p]=1;           //p是与上一个点的最小权值点，记录已经访问过
		for(int j=1;j<=n;j++)
			if(!vis[j]&&lowc[j]>cost[p][j])
				lowc[j]=cost[p][j];            //存放所有点与p点的边的权值
	}
	return ans;
}
int main(void)
{
	int n;
	while(scanf("%d",&n)&&n)
	{
	    memset(cost,INF,sizeof(cost));
	    for(int i=1;i<=n*(n-1)/2;i++)
		{
			int a,b,c,d;
			scanf("%d%d%d%d",&a,&b,&c,&d);
			if(d==1)
				cost[a][b]=cost[b][a]=0;
			else
				cost[a][b]=cost[b][a]=c;
		}
		cout<<Prim(cost,n)<<endl;
	}
	return 0;
}