考研路茫茫——空调教室
Time Limit: 4000/2000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)Total Submission(s): 3626 Accepted Submission(s): 1107
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一下就行了。= =
Ac代码:
#include<stdio.h>
#include<string.h>
#include<vector>
#include<algorithm>
#include<map>
using namespace std;
struct node
{
int from,to,next;
}e[1500000];
int size[150000];
int a[150000];
vector<int>mp[105000];
int head[150000];
int n,m,sig,cnt,tt,cont;
int dfn[150050];
int low[150050];
int vis[150050];
int stack[150050];
int color[150050];
int sum[150050];
void add(int from,int to)
{
e[cont].to=to;
e[cont].next=head[from];
head[from]=cont++;
}
void Tarjan(int u,int from)
{
vis[u]=1;
dfn[u]=low[u]=cnt++;
stack[++tt]=u;
int flag=0;
for(int i=0;i<mp[u].size();i++)
{
int v=mp[u][i];
if(vis[v]==0)
{
Tarjan(v,u);
low[u]=min(low[u],low[v]);
}
else if(v==from)
{
if(flag)low[u]=min(low[u],dfn[v]);
flag++;
}
else low[u]=min(low[u],low[v]);
}
if(low[u]==dfn[u])
{
sig++;
do
{
color[stack[tt]]=sig;
vis[stack[tt]]=-1;
}
while(stack[tt--]!=u);
}
}
int Dfs(int u,int from)
{
size[u]=a[u];
for(int i=head[u];i!=-1;i=e[i].next)
{
int v=e[i].to;
if(v==from)continue;
Dfs(v,u);
size[u]+=size[v];
}
}
void Slove()
{
tt=-1;
cnt=1;
sig=0;
memset(dfn,0,sizeof(dfn));
memset(low,0,sizeof(low));
memset(vis,0,sizeof(vis));
memset(stack,0,sizeof(stack));
memset(color,0,sizeof(color));
for(int i=1;i<=n;i++)
{
if(vis[i]==0)
{
Tarjan(i,-1);
}
}
if(sig==1)
{
printf("impossible\n");
return ;
}
cont=0;
memset(size,0,sizeof(size));
memset(head,-1,sizeof(head));
map<pair<int,int >,int>s;
for(int i=1;i<=n;i++)
{
for(int j=0;j<mp[i].size();j++)
{
int u=i,v=mp[i][j];
if(color[u]!=color[v])
{
if(s[make_pair(color[u],color[v])]==1)continue;
s[make_pair(color[u],color[v])]=1;
s[make_pair(color[v],color[u])]=1;
add(color[u],color[v]);
add(color[v],color[u]);
}
}
}
memset(sum,0,sizeof(sum));
for(int i=1;i<=n;i++)
{
sum[color[i]]+=a[i];
}
for(int i=1;i<=n;i++)a[i]=sum[i];
Dfs(color[1],-1);
int tot=size[color[1]];
int output=0x3f3f3f3f;
for(int i=1;i<=sig;i++)
{
output=min(output,abs(tot-2*size[i]));
}
printf("%d\n",output);
}
int main()
{
while(~scanf("%d%d",&n,&m))
{
for(int i=1;i<=n;i++)scanf("%d",&a[i]),mp[i].clear();
for(int i=0;i<m;i++)
{
int x,y;
scanf("%d%d",&x,&y);
x++;y++;
mp[x].push_back(y);
mp[y].push_back(x);
}
Slove();
}
}