【支配树】HDU - 6604 - Blow up the city

题目链接http://acm.hdu.edu.cn/showproblem.php?pid=6604

---

题意

在一个DAG图上，有两个起始点，它们向终点运输物资。没有出边的点是终点。如果摧毁一个点，使得某一个起始点无法到达所有的终点，就算胜利。
有q个询问，每次询问两个点作为初始点，有几种胜利方案。

---

题解

所有边反向，新建一个超级源点连向所有入度为零的点。题目就变成了源点到两个点上的必经点的个数。用支配树可以计算，然后再用LCA进行容斥即可。
因为本身图是DAG，所以可以先求出拓扑序列，然后扫一遍，求出所有到该点的LCA来建支配树。

---

#include<bits/stdc++.h>
using namespace std;
const int N=6e5+7;
struct Edge{
    int v[N],nxt[N],p[N],edn;
    void init(int n){
        for(int i=1;i<=n;i++) p[i]=-1;
        edn=-1;
    }
    void add(int from,int to){
        v[++edn]=to;nxt[edn]=p[from];p[from]=edn;
    }
}e,fe;
int du[N],d[N];
int fa[N][22];
int t,n,m,k;
bool vis[N];
int lca(int u,int v){
    if(d[u]<d[v]) swap(u,v);
    for(int i=20;i>=0;i--) if(d[fa[u][i]]>=d[v]) u=fa[u][i];
    if(u==v) return u;
    for(int i=20;i>=0;i--){
        if(fa[u][i]!=fa[v][i]){
            u=fa[u][i],v=fa[v][i];
        }
    }
    return fa[u][0];
}
int tp[N],tot;
int main(){
    scanf("%d",&t);
    d[0]=-1;
    while(t--){
        scanf("%d%d",&n,&m);
        for(int i=1;i<=n+1;i++) fa[i][0]=0,d[i]=0,du[i]=0;
        e.init(n+1);fe.init(n+1);
        for(int i=1,u,v;i<=m;i++){
            scanf("%d%d",&v,&u);
            e.add(u,v);
            fe.add(v,u);
            du[v]++;
        }
        for(int i=1;i<=n;i++){
            if(du[i]==0){
                e.add(n+1,i);
                fe.add(i,n+1);
                du[i]++;
            }
        }
        queue<int>q;
        q.push(n+1);
        tot=0;
        while(!q.empty()){
            int u=q.front();q.pop();
            tp[++tot]=u;
            for(int i=e.p[u];~i;i=e.nxt[i]){
                int v=e.v[i];
                du[v]--;
                if(du[v]==0) q.push(v);
            }
        }
        for(int j=1;j<=n+1;j++){
            int u=tp[j];
            for(int i=fe.p[u];~i;i=fe.nxt[i]){
                int v=fe.v[i];
                if(fa[u][0]==0) fa[u][0]=v;
                else fa[u][0]=lca(fa[u][0],v);
            }
            d[u]=d[fa[u][0]]+1;
            for(int i=1;i<=20;i++){
                fa[u][i]=fa[fa[u][i-1]][i-1];
            }
        }
        scanf("%d",&k);
        for(int i=1,u,v;i<=k;i++){
            scanf("%d%d",&u,&v);
            printf("%d\n",d[u]+d[v]-d[lca(u,v)]);
        }
    }
}