7-5 畅通工程之最低成本建设问题(30 分)
某地区经过对城镇交通状况的调查,得到现有城镇间快速道路的统计数据,并提出“畅通工程”的目标:使整个地区任何两个城镇间都可以实现快速交通(但不一定有直接的快速道路相连,只要互相间接通过快速路可达即可)。现得到城镇道路统计表,表中列出了有可能建设成快速路的若干条道路的成本,求畅通工程需要的最低成本。
输入格式:
输入的第一行给出城镇数目N (1<N≤1000)和候选道路数目M≤3N;随后的M行,每行给出3个正整数,分别是该条道路直接连通的两个城镇的编号(从1编号到N)以及该道路改建的预算成本。
输出格式:
输出畅通工程需要的最低成本。如果输入数据不足以保证畅通,则输出“Impossible”。
输入样例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
输出样例1:
12
输入样例2:
5 4
1 2 1
2 3 2
3 1 3
4 5 4
输出样例2:
Impossible
#include<iostream>
#include<algorithm>
using namespace std;
const int inf=10000000;
const int N=1002;
int G[N][N];
int d[N];
int n,m;
bool vis[N]={false};
int prim()
{
fill(d,d+N,inf);
d[1]=0;
int ans=0;
for(int i=1;i<=n;i++)
{
int u=-1;
int min=inf;
for(int j=1;j<=n;j++)
{
if(vis[j]==false&&d[j]<min)
{
u=j;
min=d[j];
}
}
if(u==-1)
return -1;
vis[u]=true;
ans+=d[u];
for(int v=1;v<=n;v++)
{
if(vis[v]==false&&G[u][v]!=inf&&G[u][v]<d[v])
d[v]=G[u][v];
}
}
return ans;
}
int main()
{
int u,v,c;
cin>>n>>m;
fill(G[0],G[0]+N*N,inf);
for(int i=1;i<=m;i++)
{
cin>>u>>v>>c;
G[u][v]=G[v][u]=c;
}
int ans=prim();
if(ans==-1)
cout<<"Impossible";
else
cout<<ans;
return 0;
}