bzoj 3390: [Usaco2004 Dec]Bad Cowtractors牛的报复

→题目链接←

裸的最大生成树

写了两种算法，算是模板吧

kruskal:

#include<iostream>
#include<cstdio>
#include<vector>
#include<algorithm>
#include<queue>

#define ll long long

using namespace std;

struct node{
	int s,t,len;
	friend bool operator < (node a,node b){
		return a.len>b.len;
	}
}e[20020];

int n,m;
int ans=0;
int fa[1010];

int getroot(int x){
	if(fa[x]==x)return x;
	return fa[x]=getroot(fa[x]);
}

int main(){
	scanf("%d%d",&n,&m);
	for(int i=1; i<=n; i++)fa[i]=i;
	for(int i=0; i<m; i++){
		scanf("%d%d%d",&e[i].s,&e[i].t,&e[i].len);
	}
	sort(e,e+m);
	int sum=1;
	for(int i=0; i<m; i++){
		e[i].s=getroot(e[i].s);
		e[i].t=getroot(e[i].t);
		if(e[i].s==e[i].t)continue;
		ans+=e[i].len;
		sum++;
		if(sum==n)break;
		fa[e[i].s]=e[i].t;
	}
	if(sum==n)printf("%d\n",ans);
	else printf("-1\n");
	
	return 0;
}

prim:

#include<iostream>
#include<cstdio>
#include<vector>
#include<algorithm>
#include<queue>

#define ll long long

using namespace std;

struct node{
	int to,len;
	node(int x,int y){to=x,len=y;}
	friend bool operator < (node a,node b){
		return a.len<b.len;
	}
};

int n,m;
priority_queue<node>q;
int ans=0;
bool vis[1010];
vector<node>v[1010];

void link(int x,int y,int z){
	v[x].push_back(node(y,z));
	v[y].push_back(node(x,z));
}

int main(){
	scanf("%d%d",&n,&m);
	for(int i=0; i<m; i++){
		int x,y,z;
		scanf("%d%d%d",&x,&y,&z);
		link(x,y,z);
	}
	vis[1]=true;
	for(int i=0; i<v[1].size(); i++)q.push(node(v[1][i].to,v[1][i].len));
	int sum=1;
	while(!q.empty()){
		node t=q.top();
		q.pop();
		if(vis[t.to])continue;
		sum++;
		ans+=t.len;
		vis[t.to]=true;
		if(sum==n)break;
		for(int i=0; i<v[t.to].size(); i++)if(!vis[v[t.to][i].to])q.push(node(v[t.to][i].to,v[t.to][i].len));
	}
	if(sum==n)printf("%d\n",ans);
	else printf("-1\n");
	
	return 0;
}