hdu2242 考研路茫茫——空调教室 （tarjan缩点+树形dp）

考研路茫茫——空调教室
Time Limit: 4000/2000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 4294 Accepted Submission(s): 1262

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

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

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

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

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

Input
本题目包含多组数据，请处理到文件结束。
每组测试第一行包含两个整数N和M(0

题意

一个有向图，要你去掉其中的一条边，使得剩下的两个图是不连通的，同时满足两个图权值差的绝对值最小。

思路

因为砍掉边之后剩下的两个图不连通，所以砍掉的边肯定是桥，于是通过tarjan求出所有的桥，将图缩成树之后，遍历每个桥，求出桥两边两个子树差的绝对值的最小值。
至于求子树的权值，明显用到树形dp，树的总权值减去 两倍的子树权值即为两个子树权值的差。

代码

#define push_back pb
#define make_pair mk
#include<iostream>
#include<stdlib.h>
#include<stdio.h>
#include<cmath>
#include<map>
#include<algorithm>
#include<string>
#include<string.h>
#include<set>
#include<queue>
#include<bitset>
#include<stack>
#include<functional>
using std::pair;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int,int> PII;
typedef pair<double,int>PDI;
const double PI=acos(-1);
const int maxn=5e4+10;
const int maxm=2e4+10;
const int INF = 0x3f3f3f3f;
using namespace std;
struct edge{int to,next,vis;}e[maxm*2];
int w[maxn],low[maxn],dfn[maxn],index,scccnt,id[maxn],top;
bool instack[maxn];
vector<int>g[maxn];
int head[maxn],cnt;
int bridge[maxn][2],bcnt;
int ww[maxn];
int n,m;
int sumw;
int dp[maxn];
stack<int>S;
void tarjan(int u)
{
    instack[u]=true;
    S.push(u);
    dfn[u]=low[u]=++index;
    for(int i=head[u];~i;i=e[i].next)
    {
       // cout<<"a"<<endl;

        int v=e[i].to;
        if(e[i].vis) continue;
        e[i].vis=e[i^1].vis=1;
       // cout<<u<<" "<<v<<endl;
        if(!dfn[v])
        {
            tarjan(v);
            low[u]=min(low[u],low[v]);
            if(dfn[u]<low[v])
            {
                bridge[bcnt][0]=u;
                bridge[bcnt++][1]=v;
            }
        }
        else if(instack[v]) low[u]=min(low[u],dfn[v]);
    }
    if(dfn[u]==low[u])
    {
        scccnt++;
        while(1)
        {
            int x=S.top();S.pop();
            instack[x]=false;
            id[x]=scccnt;
            ww[scccnt]+=w[x];
            if(x==u) break;
        }
    }
}

void find_scc()
{
    memset(dfn,0,sizeof(dfn));
    top=-1;
    index=scccnt=0;
    bcnt=0;
    for(int i=0;i<n;i++) if(!dfn[i]) tarjan(i);
}

void add(int u,int v)
{
    e[cnt].to=v;
    e[cnt].next=head[u];
    e[cnt].vis=0;
    head[u]=cnt++;
}
int num;
int mi;
int vis[maxn];
void dfs(int u,int fa)
{
    vis[u]=1;
    dp[u]=ww[u];
    for(int i=0;i<g[u].size();i++)
    {
        int v=g[u][i];
        if(v==fa||vis[v]) continue;
        //cout<<u<<" "<<v<<endl;
        dfs(v,u);
        dp[u]+=dp[v];

    }
     mi=min(mi,abs(sumw-2*dp[u]));
}
int main()
{
    int t;
    while(~scanf("%d%d",&n,&m))
    {
        sumw=0;
        memset(head,-1,sizeof(head));
        cnt=0;
        memset(dp,0,sizeof(dp));
        memset(ww,0,sizeof(ww));
        for(int i=0;i<n;i++)
        {
            scanf("%d",&w[i]);
            sumw+=w[i];
        }
        int u,v;
        for(int i=0;i<m;i++)
        {
            scanf("%d%d",&u,&v);
            add(u,v);
            add(v,u);
        }
        find_scc();

        if(scccnt==1)
        {
            printf("impossible\n");
            continue;
        }
        for(int i=1;i<=n;i++) g[i].clear();
        for(int i=0;i<n;i++)
        {
            for(int j=head[i];~j;j=e[j].next)
            {
                int v=e[j].to;
                if(id[v]!=id[i])
                {
                    g[id[v]].pb(id[i]);
                    g[id[i]].pb(id[v]);
                }
            }
        }
        mi=INF;
        memset(vis,0,sizeof(vis));
        dfs(1,-1);
        printf("%d\n",mi);
    }
}