poj1287 - Networking (最小生成树）

题目链接：http://poj.org/problem?id=1287

题目大意：给出n个节点，再有m条边，这m条边代表从a节点到b节点电缆的长度，现在要你将所有节点都连起来，并且使长度最小
解题思路：这是个标准的最小生成树的问题，用prim的时候需要注意的是他有重边，取边最小的那条加入图里就可以了，但是kruskal可以忽略这个问题

prim

#include <iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<string.h>
using namespace std;
#define maxn 105
#define inf 0x3f3f3f3f
int n,m;
int maps[maxn][maxn];
bool vis[maxn];
int  dis[maxn];
void prim()
{
	memset(vis,false,sizeof vis);
	vis[1]=1;
	for(int i=2;i<=n;i++)
		dis[i]=maps[1][i];
	dis[1]=0;
	int sum=0;
	for(int i=1;i<=n;i++)
	{
		int pos=-1;
		for(int j=1;j<=n;j++)
		{
			if(!vis[j]&&(pos==-1||dis[j]<dis[pos]))
				pos=j;
		}
		vis[pos]=true;
		sum+=dis[pos];
		for(int j=1;j<=n;j++)
		{
			if(!vis[j]&&dis[j]>maps[pos][j])
				dis[j]=maps[pos][j];
		}
	}
	printf("%d\n",sum);
}
int main()
{
	while(~scanf("%d",&n),n!=0)
	{
		scanf("%d",&m);
		for(int i=1;i<=n;i++)
			for(int j=1;j<=n;j++)
                if(i==j)
                    maps[i][j]=0;
                else
                    maps[i][j]=inf;
		int a,b,c;
		while(m--)
		{
			scanf("%d%d%d",&a,&b,&c);
			maps[b][a]=maps[a][b]=min(maps[a][b],c);
		}
		prim();
	}
	return 0;
}

kruskal：

#include <iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<string.h>
using namespace std;
#define maxn 2500
int n,m,t;
int u[maxn],v[maxn];
int  w[maxn];
int r[maxn],pre[maxn];
int find(int x)
{
	int r=x;
	while(r!=pre[r])
		r=pre[r];
	int i=x,j;
	while(i!=r)
	{
		j=pre[i];
		pre[i]=r;
		i=j;
	}
	return r;
}
void join (int x,int y)
{
	int fx=find(x),fy=find(y);
	if(fx!=fy)
		pre[fx]=fy;
}
bool cmp(int i,int j)
{
	return w[i]<w[j];
}
int tot;
bool check()
{
	int ans=find(1);
	for(int i=2;i<=n;i++)
		if(find(i)!=ans)
		return false;
	return true;
}
void kruskal()
{
	int t=0;
	for(int i=0;i<=n;i++)pre[i]=i;
	for(int i=0;i<tot;i++)r[i]=i;
	sort(r,r+tot,cmp);
	int  sum=0;
	for(int i=0;i<tot;i++)
	{
		int e=r[i],x=find(u[e]),y=find(v[e]);
		if(x!=y)
		{
			sum+=w[e];
			pre[x]=y;
		}
		if(check())
			break;
	}
	printf("%d\n",sum);
}
int main()
{
	while(~scanf("%d",&n),n!=0)
	{
		tot=0;
		scanf("%d",&m);
		int a,b,c;
		for(int i=1;i<=m;i++)
		{
			scanf("%d%d%d",&a,&b,&c);
			u[tot]=a;
			v[tot]=b;
			w[tot]=c;
			tot++;
		}
		kruskal();
	}
	return 0;
}