http://acm.sdut.edu.cn/sdutoj/problem.php?action=showproblem&problemid=2896
最小生成树:n个顶点n-1条边
本题因为有50000个点,所以只能用Kuscal
- #include <iostream>
- #include <stdio.h>
- #include <string.h>
- #include <stdlib.h>
- #include <math.h>
- using namespace std;
- int n,m,tt;
- struct node
- {
- int x,y,w;
- }edge[200001];
- int bin[50002];
- void add(int u,int v,int w1)
- {
- edge[tt].x=u;
- edge[tt].y=v;
- edge[tt].w=w1;
- tt++;
- }
- int cmp(const void *a,const void *b)
- {
- return (*(struct node *)a).w-(*(struct node *)b).w;
- }
- int findx(int x)
- {
- int r=x;
- while(r!=bin[r])
- r=bin[r];
- int j,k;
- j=x;
- while(j!=r)
- {
- k=bin[j];
- bin[j]=r;
- j=k;
- }
- return r;
- }
- void merge(int fx,int fy)
- {
- // int fx=findx(x); //节约时间
- //int fy=findx(y); //节约时间
- if(fx!=fy)
- bin[fx]=fy;
- }
- void Kuscal()
- {
- int sum=0;
- int ll=1;//最小生成树n-1条边
- int i=0;
- while(ll<n)
- {
- if(findx(edge[i].x)!=findx(edge[i].y))
- {
- merge(edge[i].x,edge[i].y);
- sum=sum+edge[i].w;
- ll++;
- }
- i++;
- }
- printf("%d\n",sum);
- }
- int main()
- {
- int u,v,w1;
- while(scanf("%d%d",&n,&m)!=EOF)
- {
- tt=0;
- for(int i=1;i<=n;i++)
- bin[i]=i;
- while(m--)
- {
- scanf("%d%d%d",&u,&v,&w1);
- add(u,v,w1);
- }
- qsort(edge,tt,sizeof(edge[0]),cmp);
- Kuscal();
- }
- return 0;
- }
时间: 2024-12-28 06:15:01