- 某地区经过对城镇交通状况的调查,得到现有城镇间快速道路的统计数据,并提出“畅通工程”的目标:使整个地区任何两个城镇间都可以实现快速交通(但不一定有直接的快速道路相连,只要互相间接通过快速路可达即可)。现得到城镇道路统计表,表中列出了任意两城镇间修建快速路的费用,以及该道路是否已经修通的状态。现请你编写程序,计算出全地区畅通需要的最低成本。
输入格式:
输入的第一行给出村庄数目N (1≤N≤100);随后的N(N−1)/2行对应村庄间道路的成本及修建状态:每行给出4个正整数,分别是两个村庄的编号(从1编号到N),此两村庄间道路的成本,以及修建状态 — 1表示已建,0表示未建。
输出格式:
输出全省畅通需要的最低成本。
输入样例:
4
1 2 1 1
1 3 4 0
1 4 1 1
2 3 3 0
2 4 2 1
3 4 5 0
输出样例:
3
(1)分析
由于是稠密图,所以采用Prim算法。二维数组数组存储图,数组存储路费,visit[]记录该点是否已经加入最小生成树中。首先以第一个结点作为最小生成树的初始结点,然后找出与最小生成树中各结点权重最小边,且该结点不在最小生成树中(确保无环路),然后加入到最小生成树中,更新没有在最小生成树中的结点和最小生成树中各结点权重边。循环上述步骤直到生成最小生成树,过程中最小生成树中各结点权重最小边的权重和就是最低成本。
时间复杂度:O(n²)
(2)代码及运行
#include <bits/stdc++.h>
using namespace std;
int main()
{
int valmap[110][110]; //map数组存储图
int value[110]; //数组存储路费
int visit[110]= {0};//表示该点是否已经加入最小生成树中
int n,i,j;
cin>>n;
memset(value,1,sizeof(value));
int m=n*(n-1)/2;
for(i=1; i<=m; i++)
{
int x1,x2,x3,x4;
cin>>x1>>x2>>x3>>x4;
if(!x4)
{
valmap[x1][x2]=x3;
valmap[x2][x1]=x3;
}
else
{
valmap[x1][x2]=0;
valmap[x2][x1]=0;
}
}
for(i=1; i<=n; i++)
{
valmap[i][i]=0;
}
int start=1;
visit[start]=1;
for(i=1; i<=n; i++)//初始化
{
value[i]=valmap[start][i];
}
int num=0;
for(i=1; i<=n-1; i++)//最多n-1条路
{
int minval=99999;
for(j=1; j<=n; j++)
{
if(!visit[j] && (value[j]<minval))
{
minval=value[j];
start=j;
}
}
if(minval==99999)break;
num+=minval;
visit[start]=1;
for(j=1; j<=n; j++)
{
if (!visit[j] && valmap[start][j]<value[j])
{
value[j]=valmap[start][j];
}
}
}
cout<<num;
return 0;
}