→题目链接←
裸的最大生成树
写了两种算法,算是模板吧
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;
}