bzoj2815 ZJOI2012 灾难

题目链接

先求拓扑序

现在要求一棵树，如果某一个结点消失，那么整棵子树都会消失

一个点能影响的只能是在拓扑序中比它靠前的，于是边往树中加点，就边求一个点在原图中连接的点的lca

为了保证这些点能受到当前要加入点的影响，就按拓扑序加点，最后dfs一遍找出子树大小

#include<algorithm>
#include<iostream>
#include<cstdlib>
#include<cstring>
#include<cstdio>
#include<string>
#include<cmath>
#include<ctime>
#include<queue>
#include<stack>
#include<map>
#include<set>
#define rre(i,r,l) for(int i=(r);i>=(l);i--)
#define re(i,l,r) for(int i=(l);i<=(r);i++)
#define Clear(a,b) memset(a,b,sizeof(a))
#define inout(x) printf("%d",(x))
#define douin(x) scanf("%lf",&x)
#define strin(x) scanf("%s",(x))
#define LLin(x) scanf("%lld",&x)
#define op operator
#define CSC main
typedef unsigned long long ULL;
typedef const int cint;
typedef long long LL;
using namespace std;
void inin(int &ret)
{
    ret=0;int f=0;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=1;ch=getchar();}
    while(ch>='0'&&ch<='9')ret*=10,ret+=ch-'0',ch=getchar();
    ret=f?-ret:ret;
}
int sta[100010],top,n,m;
struct wocao
{
    int head[100010],next[100020],zhi[100020],in[100010],ed,s[100010],fa[100010][20],shen[100010];
    int st[100010],to;
    void add(int a,int b)
    {
        next[++ed]=head[a],head[a]=ed,zhi[ed]=b;
        in[b]++;
    }
    void dfs(int x)
    {
        s[x]=1;
        for(int i=head[x];i;i=next[i])if(zhi[i]!=fa[x][0])
        {
            fa[zhi[i]][0]=x;
            dfs(zhi[i]);
            s[x]+=s[zhi[i]];
        }
    }
    int lca(int a,int b)
    {
        if(a==-1)return b;
        if(shen[a]<shen[b])swap(a,b);
        int temp=shen[a]-shen[b];
        rre(i,19,0)if(temp&(1<<i))a=fa[a][i];
        rre(i,19,0)if(fa[a][i]!=fa[b][i])a=fa[a][i],b=fa[b][i];
        return a==b?a:fa[a][0];
    }
    void topsort()
    {
        re(i,1,n)if(!in[i])st[++to]=i;
        while(to)
        {
            int x=st[to--];
            sta[++top]=x;
            for(int i=head[x];i;i=next[i])
            {
                in[zhi[i]]--;
                if(!in[zhi[i]])st[++to]=zhi[i];
            }
        }
    }
    void nima(int x)
    {
        re(i,1,17)fa[x][i]=fa[fa[x][i-1]][i-1];
    }
}A,B;
void wori()
{
    rre(i,top,1)
    {
        int temp=sta[i],f=-1;
        for(int i=A.head[temp];i;i=A.next[i])
            f=B.lca(f,A.zhi[i]);
        if(f==-1)f=0;
        B.add(f,temp);
        B.shen[temp]=B.shen[f]+1;
        B.fa[temp][0]=f;
        B.nima(temp);
    }
}
int main()
{
    inin(n);
    re(i,1,n)
    {
        int x;
        while(1)
        {
            inin(x);
            if(!x)break;
            A.add(i,x);
        }
    }
    A.topsort();
    wori();
    B.dfs(0);
    re(i,1,n)printf("%d
",B.s[i]-1);
     return 0;
}