考研路茫茫——空调教室
Time Limit: 4000/2000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)Total Submission(s): 1767 Accepted Submission(s): 511
Problem Description
众所周知,HDU的考研教室是没有空调的,于是就苦了不少不去图书馆的考研仔们。Lele也是其中一个。而某教室旁边又摆着两个未装上的空调,更是引起人们无限YY。
一个炎热的下午,Lele照例在教室睡觉的时候,竟然做起了空调教室的美梦。
Lele梦到学校某天终于大发慈悲给某个教室安上了一个空调。而且建造了了M条通气管道,让整个教学楼的全部教室都直接或间接和空调教室连通上,构成了教室群,于是,全部教室都能吹到空调了。
不仅仅这样,学校发现教室人数越来越多,单单一个空调已经不能满足大家的需求。于是,学校决定封闭掉一条通气管道,把全部教室分成两个连通的教室群,再在那个没有空调的教室群里添置一个空调。
当然,为了让效果更好,学校想让这两个教室群里的学生人数尽量平衡。于是学校找到了你,问你封闭哪条通气管道,使得两个教室群的人数尽量平衡,并且输出人数差值的绝对值。
一个炎热的下午,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之间建了一个通气管道。
每组测试第一行包含两个整数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"。
如果不管封闭哪条管道都不能把教室分成两个教室群,就输出"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
思路: DFS会生成一颗搜索树,在回溯的时候记录每个节点所有subtree的孩子的和,每次与到割边的时候,看看是否可以跟新所求的直。
还有考虑重边的情况。
#include <iostream>
#include <cstdio>
#include <cstring>
#include <vector>
#include <algorithm>
#include <cmath>
using namespace std;
const int maxn = 10000 + 10;
const int INF = 0x7fffffff;
vector<int> G[maxn];
int pre[maxn];
int low[maxn];
int vertex[maxn]; //record the number of every node.
int dfs_clock;
int total; // the sum of total number.
int result;
int gn, gm;
int dfs(int u, int fa)
{
int lowu = pre[u] = ++dfs_clock;
for(int i = 0; i < (int)G[u].size(); i++) {
int v = G[u][i];
if(!pre[v]) { // neibor have not been visited.
int lowv = dfs(v, u); // backing.
lowu = min(lowv, lowu);
if(lowv > pre[u]) { // bridge from u to v.
int tmp = total - vertex[v];
int dis = max(tmp-vertex[v], vertex[v]-tmp);
result = min(dis, result);
}
}
else if(pre[v] < pre[u] ){ // neibor node have been visited.
int cnt = 0;
for(int t = 0; t < (int)G[u].size(); t++) {
if(G[u][t] == v) cnt++;
}
if(cnt >= 2 || fa != v) { // lead to WA.
lowu = min(lowu, pre[v]);
}
}
}
low[u] = lowu;
vertex[fa] += vertex[u];
return lowu;
}
void init() {
dfs_clock = 0;
total = 0;
result = INF;
memset(pre, 0, sizeof(pre));
memset(low, 0, sizeof(low)); //add.
for(int i = 0; i < maxn; i++) {
G[i].clear();
}
}
int main()
{
int u, v;
while(scanf("%d%d", &gn, &gm) != EOF) {
init();
for(int i = 0; i < gn; i++) {
scanf("%d", &vertex[i]);
total += vertex[i];
}
for(int i = 0; i < gm; i++) {
scanf("%d%d", &u, &v);
G[u].push_back(v);
G[v].push_back(u);
}
dfs(0, -1);
if(result == INF) {
printf("impossible\n");
} else
printf("%d\n", result);
}
return 0;
}
/**
duplicat edges.
5 7
2 2 2 6 7
0 1
0 2
1 2
1 4
2 3
2 3
1 4
**/