7-10 公路村村通（最小生成树）

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;
}

P3366 【模板】最小生成树

代码：

#include<iostream>
#include<queue>
using namespace std;
struct node{
	int d,nextv;
	bool operator <(const node &n1)const{
		if(d==n1.d) return nextv<n1.nextv;  //从大到小 
		else return d>n1.d;      //从小到大 
	} 
};
int n,m,sum,vtct;
int vt[5100];
int h[5100],ne[400100],v[400100],w[400100];
int idx;
void add(int x,int y,int z){
	ne[++idx]=h[x]; h[x]=idx; v[idx]=y; w[idx]=z;
}
void prim(){
	priority_queue<node>pq;
	vt[1]=1;
	vtct++;
	for(int i=h[1];i;i=ne[i])
	   pq.push({w[i],v[i]});
	while(vtct<n&&!pq.empty()){
		node tmp=pq.top();
		pq.pop();
		if(vt[tmp.nextv]) continue;
		sum+=tmp.d;
		vtct++;
		vt[tmp.nextv]=1;
		for(int i=h[tmp.nextv];i;i=ne[i]){
		    if(vt[v[i]]) continue;
		     pq.push({w[i],v[i]});
		}
	      
	}
}
int main(){
	cin>>n>>m;
	if(m<n-1){
		printf("orz"); return 0;
	}
	while(m--){
		int x,y,z;
		scanf("%d%d%d",&x,&y,&z);
		add(x,y,z),add(y,x,z);
	}
	prim();
	if(vtct==n) printf("%d",sum);
	else printf("orz");
	return 0;
}