http://acm.sdut.edu.cn/sdutoj/problem.php?action=showproblem&problemid=2896
最小生成树:n个顶点n-1条边
本题因为有50000个点,所以只能用Kuscal
#include <stdio.h> #include <stdlib.h> #include <string.h> int bin[50001]; struct node { int x,y,z; }q[200001]; int cmp(const void *a,const void *b) { return (*(struct node *)a).z>(*(struct node *)b).z?1:-1; } int findx(int x) { int r=x; while(r!=bin[r]) { r=bin[r]; } int j=x,k; while(j!=r) { k=bin[j]; bin[j]=r; j=k; } return r; } int main() { int n,m; while(scanf("%d%d",&n,&m)!=EOF) { for(int i=0;i<=n;i++) bin[i]=i; for(int i=0;i<m;i++) { scanf("%d%d%d",&q[i].x,&q[i].y,&q[i].z); } qsort(q,m,sizeof(q[0]),cmp); int x,y,l=0,sum=0; for(int i=0;i<m;i++) { x=findx(q[i].x); y=findx(q[i].y); if(x!=y) { bin[x]=y; sum=sum+q[i].z; l++; } if(l==n-1) break; } printf("%d ",sum); } return 0; }
- #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 ",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;
- }