7-13 畅通工程之最低成本建设问题

某地区经过对城镇交通状况的调查，得到现有城镇间快速道路的统计数据，并提出“畅通工程”的目标：使整个地区任何两个城镇间都可以实现快速交通（但不一定有直接的快速道路相连，只要互相间接通过快速路可达即可）。现得到城镇道路统计表，表中列出了有可能建设成快速路的若干条道路的成本，求畅通工程需要的最低成本。

输入格式:

输入的第一行给出城镇数目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>
using namespace std;
const int infinite=99999;
int mp[1001][1001];
int lowcost[1001];
int mst[1001];
int Prim(int n){
    int i,j,min=infinite,mid=0,sum=0;
    for(i=2;i<=n;i++){
        lowcost[i]=mp[1][i];
        mst[i]=1;
    }
    mst[1]=0;
    for(i=2;i<=n;i++){
        min=infinite;
        for(j=2;j<=n;j++)
            if(lowcost[j]<min&&lowcost[j]){
                min=lowcost[j];
                mid=j;
            }
        if(min==infinite)
            return -1;
        sum+=min;
        lowcost[mid]=0;
        for(j=2;j<=n;j++)
            if(lowcost[j]>mp[j][mid]){
                lowcost[j]=mp[j][mid];
                mst[j]=mid;
            }
    }
    return sum;
}
int main(){
    int n,m,i=0,j=0,start,end,dis;
    cin>>n>>m;
    for(i=1;i<=n;i++)
        for(j=1;j<=n;j++)
            if(i==j)
                mp[i][j]=0;
            else
                mp[i][j]=infinite;
    for(i=0;i<m;i++){
        cin>>start>>end>>dis;
        mp[start][end]=dis;
        mp[end][start]=dis;
    }
    if(Prim(n)==-1)
        cout<<"Impossible"<<endl;
    else
        cout<<Prim(n)<<endl;
    return 0;
}