P3366 【模板】最小生成树(朴素Prim做法

【模板】最小生成树

题目描述

如题，给出一个无向图，求出最小生成树，如果该图不连通，则输出 orz。

输入格式

第一行包含两个整数 $N,M$，表示该图共有 $N$ 个结点和 $M$ 条无向边。

接下来 $M$ 行每行包含三个整数 $X_i,Y_i,Z_i$，表示有一条长度为 $Z_i$ 的无向边连接结点 $X_i,Y_i$。

输出格式

如果该图连通，则输出一个整数表示最小生成树的各边的长度之和。如果该图不连通则输出 orz。

样例 #1

样例输入 #1

4 5
1 2 2
1 3 2
1 4 3
2 3 4
3 4 3

样例输出 #1

7

提示

数据规模：

对于 $20\%$ 的数据，$N\le 5$，$M\le 20$。

对于 $40\%$ 的数据，$N\le 50$，$M\le 2500$。

对于 $70\%$ 的数据，$N\le 500$，$M\le 10^4$。

对于 $100\%$ 的数据：$1\le N\le 5000$，$1\le M\le 2\times 10^5$，$1\le Z_i\le 10^4$。

样例解释：

所以最小生成树的总边权为 $2+2+3=7$。

#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=5500;
const int M=400010;
int d[N],st[N];
int h[N],w[M],e[M],ne[M],idx;
int n,m,res;
void add(int a,int b,int c){
	e[idx]=b;
	w[idx]=c;
	ne[idx]=h[a];
	h[a]=idx++;
}

bool Prime(){
	d[1]=0;
	for(int i=1;i<=n;i++){
		int t=0;
		
		for(int j=1;j<=n;j++)
			if(!st[j]&&d[t]>d[j])t=j;
	//	printf("t==%d\n",t);
	//	printf("d[%d]=%d\n",t,d[t]);	
		if(d[t]==0x3f3f3f3f)return false;
		st[t]=true;
		res+=d[t];
		for(int k=h[t];~k;k=ne[k]){
			int j=e[k],ww=w[k];
			d[j]=min(d[j],ww); 
		//	printf("d[%d]=%d\n",j,d[j]);
		}
		
	}
	return true;
}

int main(){
	memset(h,-1,sizeof h);
	memset(d,0x3f,sizeof d);
	cin>>n>>m;
	while(m--){
		int x,y,z;cin>>x>>y>>z;
		add(x,y,z);
		add(y,x,z);
	}
	
	if(Prime())cout<<res;
	else cout<<"orz";
	
	
}