某地区经过对城镇交通状况的调查,得到现有城镇间快速道路的统计数据,并提出“畅通工程”的目标:使整个地区任何两个城镇间都可以实现快速交通(但不一定有直接的快速道路相连,只要互相间接通过快速路可达即可)。现得到城镇道路统计表,表中列出了任意两城镇间修建快速路的费用,以及该道路是否已经修通的状态。现请你编写程序,计算出全地区畅通需要的最低成本。
输入格式:
输入的第一行给出村庄数目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
结尾无空行
#include <bits/stdc++.h>
#define INF 0x3f3f3f3f
#define MAXN 101
using namespace std;
int n;
int City[MAXN][MAXN];
int dis[MAXN];
bool vis[MAXN]{false};
int prim(){
fill(dis, dis + MAXN, INF);
// 源点
dis[1] = 0;
int sum = 0;
for(int i = 1; i <= n; i++){
int u = -1, Min = INF;
// 找边权最小顶点
for(int j = 1; j <= n; j++){
if(vis[j] == false && dis[j] < Min)
Min = dis[u = j];
}
// 找不到可连接点 图不连通 退出
if(u == -1) break;
sum += dis[u];
vis[u] = true;
// 更新邻接点边权
for(int v = 1; v <= n; v++){
if(vis[v] == false && City[u][v] < dis[v])
dis[v] = City[u][v];
}
}
cout << sum;
}
main(){
fill(City[0], City[0] + MAXN * MAXN, INF);
int x, y, state, cost;
cin >> n;
while(cin >> x >> y >> cost >> state){
// 已建成本为0
if(state == 1)
City[x][y] = City[y][x] = 0;
else
City[x][y] = City[y][x] = cost;
}
prim();
}
fill(City[0], City[0] + MAXN * MAXN, INF);
图是二维数组,地址需要从列开始即 City[0] 而不是 City.