HDU 2242 考研路茫茫——空调教室

题目链接

http://acm.hdu.edu.cn/showproblem.php?pid=2242

思路

断掉一条在一个边双连通分量中的边，不会使图变得不连通，首先想到的是边双连通分量缩点，得到的缩点图是一颗树，剩下的边都是桥，不能一条一条的删除，然后计算差值。断掉任意一条边后，变成森林含两个子树，用dp[u]表示以u为根结点的子树上所有节点权值之和。那么另外一颗树上所有节点权值之和为sum-dp[u]。其差值为abs(sum-2*dp[u]) 所以需要计算缩点图上每个结点的dp[]值，差值最小的即答案.

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<vector>
#include<string>
#include<queue>
#include<stack>
#include<set>
#include<map>
#define ll long long
using namespace std;
const int INF = ( 2e9 ) + 2;
const ll maxn = 1e4+10;
const ll maxm = 2*2e4+10;
int w[maxn],c[maxn];
int dfn[maxn],low[maxn],Stack[maxn],Belong[maxn],dp[maxn];
bool Instack[maxn],vis[maxn];
int index,top,scc,ans;
struct edge
{
    int u,v,next;
} e1[maxm],e2[maxm];
int head1[maxn],head2[maxn];
int tot1,tot2;
void add1(int u,int v)
{
    e1[tot1].u=u;
    e1[tot1].v=v;
    e1[tot1].next=head1[u];
    head1[u]=tot1++;
}
void add2(int u,int v)
{
    e2[tot2].u=u;
    e2[tot2].v=v;
    e2[tot2].next=head2[u];
    head2[u]=tot2++;
}
void tarjan(int u,int fa)
{
    dfn[u]=low[u]=++index;
    Stack[top++]=u;
    Instack[u]=true;
    int f=1;
    for(int i=head1[u];i!=-1;i=e1[i].next)
    {
        int v=e1[i].v;
        if(f&&v==fa)
        {
            f=0;
            continue;
        }
        if(!dfn[v])
        {
            tarjan(v,u);
            low[u]=min(low[u],low[v]);
        }
        else if(Instack[v])
            low[u]=min(low[u],low[v]);
    }
    if(dfn[u]==low[u])
    {
        scc++;
        int x;
        do
        {
            x=Stack[--top];
            Belong[x]=scc;
            Instack[x]=false;
            c[scc]+=w[x];
        }while(x!=u);
    }
}
// 断掉任意一边，均会形成子树  dp[u] 代表以u为根结点的子树上全部结点权值之和 
void dfs(int u,int sum)
{
    vis[u]=1;
    dp[u]=c[u];
    for(int i=head2[u];i!=-1;i=e2[i].next)
    {
        int v=e2[i].v;
        if(!vis[v])
        {
            dfs(v,sum);
            dp[u]+=dp[v];
        }
    }
    ans=min(ans,abs(sum-2*dp[u]));
}
void solve(int n,int sum)
{
    memset(dfn,0,sizeof(dfn));
    memset(c,0,sizeof(c));
    memset(vis,0,sizeof(vis));
    index=top=scc=0;
    tarjan(0,-1);

    if(scc==1)
    {
        printf("impossible\n");
        return;
    }
    for(int i=0; i<tot1; i+=2)// 因为是无向图 
    {
        int u=e1[i].u,v=e1[i].v;
        if(Belong[u]!=Belong[v])
        {
            add2(Belong[u],Belong[v]);
            add2(Belong[v],Belong[u]);
        }
    }
    ans=INF;
    dfs(1,sum);
    printf("%d\n",ans);
}
int main()
{
    int n,m;
    while(~scanf("%d%d",&n,&m))
    {
        memset(head1,-1,sizeof(head1));
        memset(head2,-1,sizeof(head2));
        tot1=tot2=0;
        int u,v,sum=0;
        for(int i=0; i<n; i++)
        {
            scanf("%d",&w[i]);
            sum+=w[i];
        }

        for(int i=0; i<m; i++)
        {
            scanf("%d%d",&u,&v);
            add1(u,v);
            add1(v,u);
        }
        solve(n,sum);
    }
}