#HDU 2242 考研路茫茫——空调教室 (tarjan + dp + 处理重边)

考研路茫茫——空调教室

Time Limit: 4000/2000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 4867    Accepted Submission(s): 1474

 

Problem Description

众所周知，HDU的考研教室是没有空调的，于是就苦了不少不去图书馆的考研仔们。Lele也是其中一个。而某教室旁边又摆着两个未装上的空调，更是引起人们无限YY。

一个炎热的下午，Lele照例在教室睡觉的时候，竟然做起了空调教室的美梦。

Lele梦到学校某天终于大发慈悲给某个教室安上了一个空调。而且建造了了M条通气管道，让整个教学楼的全部教室都直接或间接和空调教室连通上，构成了教室群，于是，全部教室都能吹到空调了。

不仅仅这样，学校发现教室人数越来越多，单单一个空调已经不能满足大家的需求。于是，学校决定封闭掉一条通气管道，把全部教室分成两个连通的教室群，再在那个没有空调的教室群里添置一个空调。

当然，为了让效果更好，学校想让这两个教室群里的学生人数尽量平衡。于是学校找到了你，问你封闭哪条通气管道，使得两个教室群的人数尽量平衡，并且输出人数差值的绝对值。

 

 

Input

本题目包含多组数据，请处理到文件结束。
每组测试第一行包含两个整数N和M(0<N<=10000,0<M<20000)。其中N表示教室的数目(教室编号从0到N-1)，M表示通气管道的数目。
第二行有N个整数Vi(0<=Vi<=1000)，分别代表每个教室的人数。
接下来有M行，每行两个整数Ai，Bi(0<=Ai,Bi<N)，表示教室Ai和教室Bi之间建了一个通气管道。

 

 

Output

对于每组数据，请在一行里面输出所求的差值。
如果不管封闭哪条管道都不能把教室分成两个教室群，就输出"impossible"。

 

 

Sample Input

4 3 1 1 1 1 0 1 1 2 2 3 4 3 1 2 3 5 0 1 1 2 2 3

 

 

Sample Output

0 1

题目大意 ： N个教室之间以M条边连接， 每个教室有一定的人数， 现在要将所有的教室分成两个教室群， 使两者的人数差值最小， 如果无法分成2个教室群， 输出 "impossible"

思路 ： 判断是否可以分成两个教教室群很简单， 直接缩点然后看新点数是否为1， 若为1则不满足， 缩点过程中记录新的点集的人数， 遍历原图， 如果两个点不在一个连通块当中， 则新建一条边， 最后搜索遍历新图，记录每个点的孩子数量， 当一个点的所有孩子访问完以后， 将孩子总数与总人数 - 孩子总数进行相减并取绝对值， 最后该值最小值就是答案， 注意有重边， 可以用链式前向星来存

Accepted code

#include<bits/stdc++.h>
#include<unordered_map>
using namespace std;

#define sc scanf
#define ls rt << 1
#define rs ls | 1
#define Min(x, y) x = min(x, y)
#define Max(x, y) x = max(x, y)
#define ALL(x) (x).begin(),(x).end()
#define SZ(x) ((int)(x).size())
#define MEM(x, b) memset(x, b, sizeof(x))
#define lowbit(x) ((x) & -(x))
#define P2(x) ((x) * (x))

typedef long long ll;
const int MOD = 1e9 + 7;
const int MAXN = 1e5 + 100;
const int INF = 0x3f3f3f3f;
inline ll fpow(ll a, ll b){ ll r = 1, t = a; while (b){ if (b & 1)r = (r*t) % MOD; b >>= 1; t = (t*t) % MOD; }return r; }

struct Edge
{
	int v, f, next;
}e[MAXN << 1];
vector <int> edge[MAXN << 1];
int dfn[MAXN], low[MAXN], n, m, tot;
int val[MAXN], suo[MAXN], dp[MAXN], scnt;
int head[MAXN], num[MAXN], sum, cnt, min_;
stack <int> st;
void init() {
	for (int i = 0; i <= n; i++) edge[i].clear(), dfn[i] = low[i] = suo[i] = num[i] = dp[i] = 0;
	tot = scnt = sum = cnt = 0; min_ = INF; 
	MEM(e, 0); MEM(head, -1);
	while (!st.empty()) st.pop();
}
void add(int from, int to) {
	e[cnt].v = to;
	e[cnt].next = head[from];
	head[from] = cnt++;
}
void tarjan(int x) {
	dfn[x] = low[x] = ++tot;
	st.push(x);
	for (int i = head[x]; i != -1; i = e[i].next) {
		int vi = e[i].v;
		if (e[i].f) continue;
		e[i].f = e[i ^ 1].f = 1;
		if (!dfn[vi]) {
			tarjan(vi);
			Min(low[x], low[vi]);
		}
		else Min(low[x], dfn[vi]);
	}
	if (dfn[x] == low[x]) {
		scnt++; int k;
		do {
			k = st.top(); st.pop();
			dp[scnt] += val[k];
			suo[k] = scnt;
		} while (k != x);
	}
}
void dfs(int x, int fa) {
	num[x] = dp[x];
	for (int i = 0; i < SZ(edge[x]); i++) {
		int vi = edge[x][i];
		if (vi == fa) continue;
		dfs(vi, x);
		num[x] += num[vi];  // 每个点孩子数量
	} 
	int s = num[x], y = sum - num[x];
	Min(min_, abs(s - y));  // 差值
}

int main()
{
	while (~sc("%d %d", &n, &m)) {
		init();
		for (int i = 0; i < n; i++) sc("%d", &val[i]), sum += val[i];
		for (int i = 0; i < m; i++) {
			int ui, vi;
			sc("%d %d", &ui, &vi);
			add(ui, vi); add(vi, ui);
		}
		for (int i = 0; i < n; i++) {
			if (!dfn[i]) tarjan(i);
		}
		for (int i = 0; i < n; i++) {
			for (int j = head[i]; j != -1; j = e[j].next) {
				int ui = suo[i], vi = suo[e[j].v];
				if (ui != vi) edge[ui].push_back(vi);  // 新图
			}
		}
		if (scnt == 1) { printf("impossible\n"); continue; } // 一个连通块
		dfs(1, 1);
		printf("%d\n", min_);
	}
	return 0;
}