Problem Description
省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通(但不一定有直接的公路相连,只要能间接通过公路可达即可)。现得到城镇道路统计表,表中列出了任意两城镇间修建道路的费用,以及该道路是否已经修通的状态。现请你编写程序,计算出全省畅通需要的最低成本。
Input
测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( 1< N < 100 );随后的 N(N-1)/2 行对应村庄间道路的成本及修建状态,每行给4个正整数,分别是两个村庄的编号(从1编号到N),此两村庄间道路的成本,以及修建状态:1表示已建,0表示未建。
当N为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;
}