poj3228 Gold Transportation 并查集

poj3228 Gold Transportation


思路是并查集。
并查集是在学习最小生成树kruskal算法的时候学习的。
印象比较深。
kruskal是从边的角度进行的得一种贪心算法。
现在这道题其实很类似最小生成树kruskal算法，但是并它要简单得多。
因为我们只需要满足最大值最小。而不需要满足总体最小。


具体为：将1~n个town抽象为单个节点。
然后对m条边进行从小到大排序。
然后从小到大读取每条边的两个节点，并查集处理之。
直到所有gold可以被运输完。
一次AC。

Source Code


Problem: 3228		User: 64162451
Memory: 380K		Time: 16MS
Language: C++		Result: Accepted
Source Code


#include<iostream>
#include<algorithm>
#define MAX(a,b) (a>b?a:b)
#define MIN(a,b) (a<b?a:b)
using namespace std;
const int N=205;
const int M=20005;
int n,m;
int gold[N];
int sto[N];
int father[N];
int sum[N];//sum数组是每个节点的权值，有正有负，正的为storage，负的为gold。
int sumofgold;
//其实用并查集本质上也是kruskal最小生成树的贪心思想吧。
struct Edge
{
	int u,v,w;
}edge[M];
bool cmp(const Edge & a,const Edge & b)
{
	return a.w<b.w;
}
int find(int i)
{
	while(father[i]>0)
	{
		i=father[i];
	}
	return i;
}
void Union(int a,int b)//将a添加到b下面
{


	father[b]+=father[a];
	father[a]=b;


}
void solve()
{
	int i=1;
	for(;i<=m&&sumofgold>0;i++)
	{
		int u=edge[i].u;
		int v=edge[i].v;
		int a=find(u);
		int b=find(v);
		if(a!=b)
		{
			Union(a,b);
			if(sum[b]*sum[a]<0)//判断两个是否异号。异号时说明有抵消(gold和sto连通)
			{
				if(sum[a]+sum[b]>0)//说明抵消完全
					sumofgold+=MIN(sum[a],sum[b]);
				else
					sumofgold-=MAX(sum[a],sum[b]);
			}
			sum[b]+=sum[a];
		}
	}
	if(sumofgold>0)
		printf("No Solution\n");
	else
		printf("%d\n",edge[i-1].w);
}
int main()
{
	while(scanf("%d",&n),n!=0)
	{
		sumofgold=0;
		memset(edge,0,sizeof(edge));
		memset(sum,0,sizeof(sum));
		memset(gold,0,sizeof(gold));
		memset(sto,0,sizeof(sto));
		memset(father,-1,sizeof(father));
		for(int i=1;i<=n;i++)
		{
			scanf("%d",&gold[i]);
			sum[i]-=gold[i];//sum初始化
		}
		for(int i=1;i<=n;i++)
		{
			scanf("%d",&sto[i]);
			sum[i]+=sto[i];//sum初始化，如果gold的地点和storage的地点重合，刚好抵消。
		}
		for(int i=1;i<=n;i++)
		{
			if(sum[i]<0)
			{
				sumofgold-=sum[i];//记录gold的总和，在从小到大搜索边的时候当sumofgold==0时停止。
			}
		}
		scanf("%d",&m);
		for(int i=1;i<=m;i++)
		{
			scanf("%d%d%d",&edge[i].u,&edge[i].v,&edge[i].w);
		}
		sort(edge+1,edge+m+1,cmp);
		solve();
	}
	return 0;
}