现有村落间道路的统计数据表中,列出了有可能建设成标准公路的若干条道路的成本,求使每个村落都有公路连通所需要的最低成本。
输入格式:
输入数据包括城镇数目正整数N(≤1000)和候选道路数目M(≤3N);随后的M行对应M条道路,每行给出3个正整数,分别是该条道路直接连通的两个城镇的编号以及该道路改建的预算成本。为简单起见,城镇从1到N编号。
输出格式:
输出村村通需要的最低成本。如果输入数据不足以保证畅通,则输出−1,表示需要建设更多公路。
输入样例:
6 15
1 2 5
1 3 3
1 4 7
1 5 4
1 6 2
2 3 4
2 4 6
2 5 2
2 6 6
3 4 6
3 5 1
3 6 1
4 5 10
4 6 8
5 6 3
结尾无空行
输出样例:
12
Prim最小生成树算法:
#include <bits/stdc++.h>
#define x first
#define y second
#define INF 0x3f3f3f3f
using namespace std;
typedef long long ll;
int N, M;
int g[1024][1024];
int dis[1024];
int findMinDis()
{
int idx = 0, minDis = INF;
for(int i = 1; i <= N; i++)
{
if(dis[i] != 0 && dis[i] < minDis)
{
minDis = dis[i];
idx = i;
}
}
return idx;
}
int main()
{
// system("chcp 65001");
cin.tie(0);
cout.tie(0);
// freopen("C:/Users/zhaochen/Desktop/input.txt", "r", stdin);
memset(g, INF, sizeof(g));
cin >> N >> M;
for(int i = 0; i < M; i++)
{
int a, b, w;
cin >> a >> b >> w;
g[a][b] = w;
g[b][a] = w;
}
for(int i = 1; i <= N; i++)
{
dis[i] = g[i][1]; // dis[i]为第i个点到最小生成树的距离,为0则表示已访问
}
dis[1] = 0; // 以第一个点为起点
int ans = 0, Vcnt = 1;
while(1)
{
int V = findMinDis();
if(!V)
{
break;
}
ans += dis[V];
dis[V] = 0;
Vcnt++;
for(int i = 1; i <= N; i++)
{
if(dis[i] != 0)
{
dis[i] = min(dis[i], g[i][V]);
}
}
}
if(Vcnt == N)
cout << ans;
else
cout << -1;
return 0;
}