支配树模板

支配树模板

只是一个板子，甚至没怎么写注释

#include <cstdio>
#include <vector>
const int N=2e5+10;
std::vector <int> dew[N],bee[N],yuyuyu[N];
int dfn[N],ha[N],fa[N],dfsclock,n,m;
void dfs(int now)
{
    ha[dfn[now]=++dfsclock]=now;
    for(int v,i=0;i<dew[now].size();i++)
        if(!dfn[v=dew[now][i]])
            fa[v]=now,dfs(v);
}
int siz[N],semi[N],idom[N];
void dfs0(int now)
{
    siz[now]=1;
    for(int v,i=0;i<dew[now].size();i++)
        dfs0(v=dew[now][i]),siz[now]+=siz[v];
}
int f[N],mi[N];
int Find(int now)
{
    if(f[now]==now) return now;
    int par=f[now];f[now]=Find(now);
    if(dfn[semi[mi[now]]]>dfn[semi[mi[par]]]) mi[now]=mi[par];
    return f[now];
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int u,v,i=1;i<=m;i++)
    {
        scanf("%d%d",&u,&v);
        dew[u].push_back(v);
        bee[v].push_back(u);
    }
    dfs(1);
    for(int i=1;i<=n;i++) mi[i]=semi[i]=f[i]=i,dew[i].clear();
    for(int k=n;k>1;k--)
    {
        int now=ha[k];
        for(int v,i=0;i<bee[now].size();i++)
        {
            v=bee[now][i];
            Find(v);//不判没影响
            semi[now]=dfn[semi[now]]<dfn[semi[mi[v]]]?semi[now]:semi[mi[v]];
        }
        yuyuyu[semi[now]].push_back(now);
        f[now]=fa[now],now=fa[now];
        for(int v,i=yuyuyu[now].size()-1;~i;i--)
        {
            Find(v=yuyuyu[now][i]);
            if(now==semi[mi[v]]) idom[v]=now;
            else idom[v]=mi[v];//等会从上面再更新
        }
        yuyuyu[now].clear();
    }
    for(int now,i=2;i<=n;i++)
    {
        now=ha[i];
        if(idom[now]!=semi[now]) idom[now]=idom[idom[now]];
        dew[idom[now]].push_back(now);
    }
    dfs0(1);
    for(int i=1;i<=n;i++) printf("%d ",siz[i]);
    return 0;
}