XMU 1613 刘备闯三国之三顾茅庐（一）【并查集】

1613: 刘备闯三国之三顾茅庐（一）

Time Limit: 1000 MS Memory Limit: 128 MB
Submit: 99 Solved: 29
[Submit][Status][Web Board]

Description

刘备（161年－223年6月10日），字玄德，东汉末年幽州涿郡涿县，西汉中山靖王刘胜的后代。刘备一生极具传奇色彩，早年颠沛流离、备尝艰辛最终却凭借自己的谋略终成一方霸主。那么在那个风云激荡的年代，刘备又是如何从一个卖草鞋的小人物一步一步成为蜀汉的开国皇帝呢？让我们一起拨开历史的迷雾，还原一个真实的刘备。

公元207年冬至，当时驻军新野的刘备在徐庶的建议下，到南阳卧龙岗拜访诸葛亮。这是刘备第一次拜访诸葛亮的故事。三国演义记载的事，由于诸葛亮出游去了，导致刘备无功而返。然而据我考古发现，其实诸葛亮是抛出了如下的问题，想考察刘备本人的谋略、以确定其是否是自己要效忠的那个人：

由于战事频发，朝廷派兵往n(2<=n<=10000)个城镇押运粮草，但是沟通不顺畅，导致有些城镇粮草多运了，而有些城镇则少运了。由于战争的影响，一些道路被毁坏，但城镇之间仍有m(0<=m<=50000)条尚未损毁的道路，诸葛亮想知道是否有可能使得利用现有的道路，重新在城镇间运载粮草，使得每个城镇需要的粮草不多也不少？

刘备为人宽厚仁爱，但对此类问题也是束手无策，这时候就是是你表现的时候了。

Input

第一行包含两个整数n（2<=n<=10000) 和m(0<=m<=50000)。

以下n行，每一行包括一个整数，分别表示城镇0到城镇n-1多运的粮草，如果为负数，则表示少运了。保证输入中，这n个数和为0。

以下m行，每一行包含两个整数x,y(0<=x<y<=n-1)，表示城镇x和城镇y存在一条道路。

Output

如果存在一种运载方案，输出POSSIBLE，否则输出IMPOSSIBLE。

Sample Input

5 3
100
-75
-25
-42
42
0 1
1 2
3 4

Sample Output

POSSIBLE

HINT

Source

by cjf

[Submit][Status][Web Board]

题目链接：

　　http://acm.xmu.edu.cn/JudgeOnline/problem.php?id=1613

题目大意：

　　N个点M条边，每个点有一个粮草值，正为多余负为缺少，问能否通过边运输粮草使每个点粮草相等。

题目思路：

　　【并查集】

　　将被边连通的点并查集归为一类。

　　然后统计被归为一类的点的权值和，如果都为0就是可行，否则不可行。

/****************************************************Author : CoolxxxCopyright 2017 by Coolxxx. All rights reserved.BLOG : http://blog.csdn.net/u010568270****************************************************/
#include<bits/stdc++.h>
#pragma comment(linker,"/STACK:1024000000,1024000000")
#define abs(a) ((a)>0?(a):(-(a)))
#define lowbit(a) (a&(-a))
#define sqr(a) ((a)*(a))
#define mem(a,b) memset(a,b,sizeof(a))
const double EPS=1e-8;
const int J=10;
const int MOD=100000007;
const int MAX=0x7f7f7f7f;
const double PI=3.14159265358979323;
const int N=10004;
const int M=100004;
using namespace std;
typedef long long LL;
double anss;
LL aans;
int cas,cass;
int n,m,lll,ans;
int a[N],fa[N],sz[N],sum[N];
int zhao(int aa)
{if(fa[aa]==0 || fa[aa]==aa)return aa;return fa[aa]=zhao(fa[aa]);
}
bool judge()
{int i;for(i=1;i<=n;i++){if(sz[i] && sum[i]!=0)return 0;}return 1;
}
int main()
{#ifndef ONLINE_JUDGEfreopen("1.txt","r",stdin);
//  freopen("2.txt","w",stdout);#endifint i,j,k;int x,y,z;
//  for(scanf("%d",&cass);cass;cass--)
//  for(scanf("%d",&cas),cass=1;cass<=cas;cass++)
//  while(~scanf("%s",s))while(~scanf("%d",&n)){mem(fa,0);mem(sz,0);mem(sum,0);scanf("%d",&m);for(i=1;i<=n;i++)scanf("%d",&a[i]);for(i=1;i<=m;i++){scanf("%d%d",&x,&y);x++;y++;int fx=zhao(x);int fy=zhao(y);fa[fy]=fx;}for(i=1;i<=n;i++){fa[i]=zhao(i);sz[fa[i]]++;sum[fa[i]]+=a[i];}if(judge())puts("POSSIBLE");else puts("IMPOSSIBLE");}return 0;
}
/*
////
*/