kruskal模板

Prim

#include<bits/stdc++.h>
using namespace std;
int n,m,cnt,ans;
bool b[5010];
struct edge
{
	int x,w;
	bool operator<(const edge &a)const
	{
		return w>a.w;
	}
};
vector<edge> v[5010];
priority_queue<edge> q;
int main()
{
	cin>>n>>m;
	for(int i=1,x,y,w;i<=m;i++)
	{
		cin>>x>>y>>w;
		v[y].push_back(edge{x,w});
		v[x].push_back(edge{y,w});
	}
	b[1]=1;
	cnt=1;
	for(int i=0;i<v[1].size();i++)q.push(edge{v[1][i].x,v[1][i].w});
	while(cnt<n)
	{
		if(q.empty())
		{
			cout<<"orz";
			return 0;
		}
		edge t=q.top();
		q.pop();
		if(b[t.x])continue;
		b[t.x]=1;
		cnt++;
		ans+=t.w;
		for(int i=0;i<v[t.x].size();i++)
		{
			if(!b[v[t.x][i].x])q.push(edge{v[t.x][i].x,v[t.x][i].w});
		}
	}
	cout<<ans;
	return 0;
}

Kruskal

#include<bits/stdc++.h>
using namespace std;
int n,m,ans,f[5010],cnt;
struct edge
{
	int x,y,w;
}a[200010];
void init()
{
    for(int i=1;i<=n;i++)f[i]=i;
}
int find(int x)
{
    if(f[x]==x)return x;
    else return f[x]=find(f[x]);
}
void c(int x)
{
    int fx=find(a[x].x),fy=find(a[x].y);
    if(fx==fy)return ;
    cnt--;
    f[fx]=fy;
    ans+=a[x].w;
}
bool cmp(edge a,edge b)
{
	return a.w<b.w;
}
int main()
{
	cin>>n>>m;
	init();
	for(int i=1;i<=m;i++)cin>>a[i].x>>a[i].y>>a[i].w;
	sort(a+1,a+m+1,cmp);
	cnt=n;
	for(int i=1;i<=m;i++)c(i);
	if(cnt>1)cout<<"orz";
	else cout<<ans;
	return 0;
}
上一篇:数据结构与算法--树(1)


下一篇:一维数组作为函数参数