hdu3140 Money Matters 并查集

 题意是有几个人分布在不同的城市，只有在同城市的人才能交换使收支综合等于0。因为n在[0,10000]，所以不能保存路径然后DFS了，数组太大。用并查集，将每个人分到自己的城市，若每一个城市里的人的money总和都等于0，则POSSIBLE，否则相反。

 

 

 #include<iostream> using namespace std; const int Max=10001; int set[Max],money[Max],sum[Max]; int find(int i) { int root=i; while(root!=set[root]) //找到根节点，即城市 { root=set[root]; } while(i!=root) //路径压缩修改查找的路径 { i=set[i]; set[i]=root; } return root; } void merge(int a,int b) //合并相交集合 { int p1,p2; p1=find(a); p2=find(b); if(p1<p2) set[p2]=p1; //用较小的元素代表该集合 else set[p1]=p2; } bool run() { int n,m; scanf("%d%d",&n,&m); int i; for(i=0;i<n;i++) scanf("%d",&money[i]); for(i=0;i<n;i++) set[i]=i; //初始化 for(i=0;i<m;i++) { int a,b; scanf("%d%d",&a,&b); merge(a,b); } fill(&sum[0],&sum[Max-1]+1,0); for(i=0;i<n;i++) { sum[find(i)]+=money[i]; //每个集合的总和 } bool flag=true; for(i=0;i<n;i++) { if(sum[i]!=0) { flag=false; break; } } if(flag==false) printf("IMPOSSIBLE/n"); else printf("POSSIBLE/n"); return true; } int main() { while(run()); return 0; }