模板裸题
n个点,m条边,求最大生成树。
#include <stdio.h>
#include <string.h>
#include <algorithm>
const int MAXN=1110;
const int MAXM=22000;
int F[MAXN];
struct Edge
{
int u,v,w;
} edge[MAXM];
int tol;
void addedge(int u,int v,int w)
{
edge[tol].u=u;
edge[tol].v=v;
edge[tol++].w=w;
}
bool cmp(Edge a,Edge b)
{
return a.w>b.w;
}
int find(int x)
{
if(F[x]==-1)return x;
else return F[x]=find(F[x]);
}
int Kruskal(int n)
{
memset(F,-1,sizeof(F));
std::sort(edge,edge+tol,cmp);
int cnt=0;
int ans=0;
for(int i=0; i<tol; i++)
{
int u=edge[i].u;
int v=edge[i].v;
int w=edge[i].w;
int t1=find(u);
int t2=find(v);
if(t1!=t2)
{
ans+=w;
F[t1]=t2;
cnt++;
}
if(cnt==n-1)break;
}
if(cnt<n-1)return -1;
else return ans;
}
int main()
{
int n,m,u,v,w;
scanf("%d %d",&n,&m);
for(int i = 0; i < m; ++i)
{
scanf("%d %d %d",&u,&v,&w);
addedge(u,v,w);
}
printf("%d\n",Kruskal(n));
return 0;
}