Bad Cowtractors POJ 2377（prim+kruskal）

模板题

机翻:

Bessie被雇来在农民John的N(2≤N<=1，000)个谷仓之间建立一个便宜的互联网网络，这些谷仓被方便地编号为1..N.FJ已经做了一些调查，并找到了M(1≤M<=20000)对谷仓之间的可能连接路线。每个可能的连接路由都有关联的开销C(1≤C≤100，000)。农夫约翰想花最少的钱连接网络；他甚至不想付给贝西。

意识到农夫约翰不会付钱给她，贝西决定做最糟糕的工作。她必须决定要安装的一组连接，以便(I)这些连接的总成本尽可能大，(Ii)所有谷仓连接在一起(使得可以通过已安装连接的路径从任何其他谷仓到达任何谷仓)，以及(Iii)使得在连接之间不存在循环(这是农民John将容易检测到的)。条件(Ii)和(Iii)确保最后一组连接看起来像一棵“树”。

求最大树，比较的时候换一下就行，还有地图的初始化还有找不出树的时候退出条件也变动一下，就没了

prim：

#include<iostream>
#include<algorithm>
#include<cstring>

using namespace std;

const int MAXN=1100;
const int INF=0x3f3f3f3f;

int map[MAXN][MAXN];
int vis[MAXN];
int dis[MAXN];

int n,m;

int prim()
{
	int ans=0;
	memset(vis,0,sizeof(vis));
	for(int i=1;i<=n;i++)
		dis[i]=map[1][i];
	vis[1]=1;
	for(int i=2;i<=n;i++)
	{
		int pos=-1;
		int flag=-INF;
		for(int j=1;j<=n;j++)
		{
			if(!vis[j]&&flag<dis[j])
			{
				flag=dis[j];
				pos=j;
			}
		}
		if(flag==-1)
			return -1;
		vis[pos]=1;
		ans+=dis[pos];
		for(int k=1;k<=n;k++)
		{
			if(!vis[k]&&dis[k]<map[pos][k])
				dis[k]=map[pos][k];
		}
	}
	return ans;
}

int main()
{
	cin>>n>>m;
	int a,b,c;
	memset(map,-1,sizeof(map));
	for(int i=0;i<m;i++)
	{
		cin>>a>>b>>c;
		map[a][b]=map[b][a]=max(map[a][b],c);
	}
	cout<<prim()<<endl;
	return 0;
}

kruskal：不需要初始化地图

#include<iostream>
#include<algorithm>
#include<cstring>

using namespace std;

int n,m;

struct node{
	int cost;
	int from;
	int to;
	bool operator < (const node &a) const
	{
		return cost>a.cost;
	} 
}map[20050];

int par[1050];

int find(int x)
{
	if(x==par[x])
		return x;
	else
		return par[x]=find(par[x]);
}

void unite(int x,int y)
{
	x=find(x);
	y=find(y);
	if(x!=y)
	par[x]=y;
	return;
}

bool same(int x,int y)
{
	return find(x)==find(y);
}

int kruskal()
{
	int ans=0;
	for(int i=0;i<=n;i++)
		par[i]=i;
	for(int i=0;i<m;i++)
	{
		if(!same(map[i].from,map[i].to))
		{
			unite(map[i].from,map[i].to);
			ans+=map[i].cost;
		}
	}
	int cnt=0;
	for(int i=1;i<=n;i++)
		if(par[i]==i)
			cnt++;
	return cnt>1?-1:ans;
}

int main()
{
	cin>>n>>m;
	for(int i=0;i<m;i++)
		cin>>map[i].from>>map[i].to>>map[i].cost;
	sort(map,map+m);
	cout<<kruskal()<<endl;
	return 0;
}