7-10 公路村村通(最小生成树)

代码:

#include<iostream>
#include<vector>
#include<queue>
using namespace std;
const int N=1010;
struct node{
	int nxt,w;
	bool operator <(const node &n1)const{
		if(w==n1.w) return nxt<n1.nxt;
		return w>n1.w;
	}
};
vector<node>e[N];
int n,m,cont,ans;
int vst[N]; 
void prim(){
	priority_queue<node>pq;
	for(auto x:e[1]){
		pq.push(x);
	} vst[1]=1; cont=1;
	while(pq.size()){
		node u=pq.top(); pq.pop();
		if(vst[u.nxt]) continue;
		vst[u.nxt]=1; ans+=u.w; cont++;
		for(auto x:e[u.nxt]){
		   if(vst[x.nxt]) continue;
		   pq.push(x);
	    } 
	}
}
int main(){
	cin>>n>>m;
	for(int i=0;i<m;i++){
		int x,y,num; scanf("%d%d%d",&x,&y,&num);
		e[x].push_back({y,num});
		e[y].push_back({x,num});
	}
	if(m<n){
		cout<<"-1"; return 0;
	}
	prim();
	if(cont!=n) cout<<"-1";
	else cout<<ans;
	return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值