Description
给出一个n个点m条边的图,求最大生成树
Input
第一行输入两个整数n和m表示点数和边数,之后m行每行三个整数u,v,c表示u点和v点之间有一条权值为c的边相连(2<=n<=1000,1<=m<=20000,1<=c<=1e5)
Output
输出最大生成树权值和
Sample Input
5 8
1 2 3
1 3 7
2 3 10
2 4 4
2 5 8
3 4 6
3 5 2
4 5 17
Sample Output
42
Solution
把边从大到小排序再用Kruscal去求即为最大生成树
Code
#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
#define maxn 1111
#define maxm 22222
struct node
{
int u,v,w;
}edge[maxm];
int T,n,m,fa[maxn];
int cmp(node a,node b)
{
return a.w>b.w;
}
int find(int x)
{
if(fa[x]==x) return x;
return fa[x]=find(fa[x]);
}
int kruskal(int n,int m)
{
for(int i=1;i<=n;i++)fa[i]=i;
int ans=0,cnt=0;
sort(edge,edge+m,cmp);
for(int k=0;k<m;k++)
{
int x=find(edge[k].u),y=find(edge[k].v);
if(x!=y)
{
cnt++;
fa[x]=y;
ans+=edge[k].w;
if(cnt==n-1)return ans;
}
}
return -1;
}
int main()
{
while(~scanf("%d%d",&n,&m))
{
for(int i=0;i<m;i++)scanf("%d%d%d",&edge[i].u,&edge[i].v,&edge[i].w);
printf("%d\n",kruskal(n,m));
}
return 0;
}