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\n",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++;
}
"%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--)
{
"%d%d%d",&u,&v,&w1);
add(u,v,w1);
}
sizeof(edge[0]),cmp);
Kuscal();
}
return 0;
}