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