7-5 公路村村通分数 30 Prim算法+并查集

现有村落间道路的统计数据表中，列出了有可能建设成标准公路的若干条道路的成本，求使每个村落都有公路连通所需要的最低成本。

输入格式:

输入数据包括城镇数目正整数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

代码长度限制

16 KB

时间限制

400 ms

内存限制

64 MB

C++ (g++)

#include<bits/stdc++.h>
using namespace std;
const int N = 1010;
int n,m;
int G[N][N];
int dt[N],st[N];
int pr[N];
int find_f(int num)
{
    if(num!=pr[num]) pr[num]=find_f(pr[num]);
    return pr[num];
}

void prim()
{
    memset(dt,0x3f,sizeof (dt));
    dt[1]=0;
    int res = 0;
    for(int i = 0;i<n;i++){
    	int t = -1;
    	for(int j = 1;j<=n;j++){
    		if(!st[j]&&(t==-1||dt[j]<dt[t]))
    			t = j;
		}
		st[t]=1;
		res+=dt[t];
		for(int x = 1;x<=n;x++){
			if(G[t][x]<dt[x]&&!st[x])
				dt[x]=G[t][x];
		}
	}
	cout<<res<<endl;
	return; 
}
int main()
{
    cin>>n>>m;
    memset(G, 0x3f, sizeof(G));
    for(int i = 1;i<=n;i++)
        pr[i]=i;
    for(int i = 1;i<=m;i++)
    {
        int a,b,w;cin>>a>>b>>w;
        G[a][b]=G[b][a]=min(w,G[a][b]);
        pr[find_f(b)] = find_f(a);
    }
    int xx = find_f(1);
    for(int i = 1;i<=n;i++){
        if(find_f(i)!=xx){
            cout<<"-1"<<endl;
            return 0;
        }
    }
    prim();
    return 0;
}