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; }