hdu2422考研路茫茫——空调教室 tarjan+树形dp

//给一个无向图，其每个顶点都有权值，求去掉一条边，将这个图分为两部分
//问这两部分的所有顶点和的绝对值的最小值
//用tarjan缩点 ， 缩点后为一棵树
//然后用树形dp求出其最小的绝对值
// ans = min(ans , (int)(abs((double)(sum - 2*dp[v])))) ;
//其中dp[u] 表示以u点为根节点的子树的学生数
#include<cstdio>
#include<cstring>
#include<iostream>
#include<cmath>
#include<vector>
using namespace std ;
const int maxn  = 10010 ;
const int maxm = 20010 ;
const int inf  = 0x7fffffff ;
struct Edge
{
    int v , u;
    int next ;
}edge[maxm<<1];
int head[maxn] , dfn[maxn] , low[maxn] , vis[maxm<<1];
int stack[maxn] , isstack[maxn] ,value[maxn] ,dp[maxn];
int nedge , top , num , step ,sum;
int n , m ;int belong[maxn] , stu[maxn];
void addedge(int u , int v)
{
    edge[nedge].v = v ;
    edge[nedge].next = head[u] ;
    head[u] = nedge++ ;
}
vector<int> vec[maxn] ;
void init()
{
    memset(vis , 0  ,sizeof(vis)) ;
    memset(dfn , 0  ,sizeof(dfn)) ;
    memset(low , 0 , sizeof(low)) ;
    memset(isstack , 0 , sizeof(isstack)) ;
    memset(value , 0 , sizeof(value));
    memset(head ,-1 , sizeof(head)) ;
    memset(dp , 0 , sizeof(dp)) ;
    for(int i = 0 ;i <= n;i++)
    vec[i].clear() ;
    top = num = step = nedge = sum = 0 ;
}
void tarjan(int u)
{
    isstack[u] = 1 ;
    stack[++top] = u ;
    dfn[u] = low[u] = ++step;
    for(int i = head[u];i != -1 ; i = edge[i].next)
    {
        if(vis[i])continue ;
        vis[i] = vis[i^1] = 1 ;
        int v = edge[i].v ;
        if(!dfn[v])
        {
            tarjan(v) ;
            low[u] = min(low[u] , low[v]) ;
        }
        else if(isstack[v])
        low[u] = min(low[u] , dfn[v]) ;
    }
    if(dfn[u] == low[u])
    {
        num++ ;
        int v ;
        while(u!=v)
        {
            v = stack[top--] ;
            belong[v] = num ;
            isstack[v] = 0 ;
            value[num] += stu[v] ;
        }
    }
}
void build_tree()
{
    for(int i = 1;i <= n;i++)
      for(int j = head[i] ; j != -1 ;j = edge[j].next)
      {
          int u = belong[i] ;
          int v = belong[edge[j].v] ;
          if(u == v)continue ;
          vec[u].push_back(v);
      }
}
void dfs(const int u , int &ans)
{
    vis[u] = 1;
    for(int i = 0;i < vec[u].size();i++)
    {
        int v = vec[u][i] ;
        if(vis[v])continue ;
        dfs(v , ans) ;
        ans = min(ans , (int)(abs((double)(sum - 2*dp[v])))) ;
        dp[u] += dp[v] ;
    }
    dp[u] += value[u];
}
int main()
{
    //freopen("in.txt" , "r" , stdin) ;
    while(~scanf("%d%d" , &n , &m))
    {
        int a , b ;
        init() ;
        for(int i = 1;i <= n;i++)
        {
            scanf("%d" , &stu[i]) ;
            sum += stu[i] ;
        }
        for(int i = 1;i <= m;i++)
        {
            scanf("%d%d" , &a , &b) ;
            addedge(a+1 , b+1) ;
            addedge(b+1 , a+1) ;
        }
        int ans = inf ;
        tarjan(1) ;
        build_tree() ;
        if(num <= 1)puts("impossible") ;
        else
        {
            memset(vis ,  0 , sizeof(vis)) ;
            dfs(1 , ans) ;
            printf("%d\n" , ans) ;
        }
    }
    return 0 ;
}