bzoj 2815 ZJOI2012 灾难 灭绝树 and DAG

题意:求灭绝树下的子树大小
解:

1、拓扑排序

2、****设立虚节点****作为所有刚开始没有父亲节点的父亲

3、我们按拓扑序对于每一个节点找到他灭绝树上的父亲 (他所有原图中的父亲在 灭绝树上的 lca)

#include<bits/stdc++.h>
#define pb push_back
#define en '\n'
using namespace std;
const int maxn = 65536+100;//,maxm = 10005;
int du[maxn],head[3][maxn];
inline int read(){
    int out = 0,flag = 1; char c = getchar();
    while (c < 48 || c > 57){if (c == '-') flag = -1; c = getchar();}
    while (c >= 48 && c <= 57){out = (out << 3) + (out << 1) + c - 48; c = getchar();}
    return out * flag;
}
vector<int>vec,vec_fa[maxn],vec_1[maxn],vec_2[maxn];
bool outq[maxn];
int anc[maxn][33];
int sz[maxn],dep[maxn];
queue<int>q;
void dfs(int x){
    sz[x]+=1;
    for(int i=0;i<vec_2[x].size();i++){
        int y=vec_2[x][i];
        dfs(y);
        sz[x]+=sz[y];
    }
}
int lca(int x,int y){
    if(dep[x]>dep[y])swap(x,y);
    int t=dep[y]-dep[x];
    for(int i=0;i<30;++i){
        if((t>>i)&1){
            y=anc[y][i];
        }
    }
    if(x==y)return x;
    for(int i=29;i>=0;--i){
        if(anc[x][i]==anc[y][i])continue;
        x=anc[x][i],y=anc[y][i];
    }
    return anc[x][0];
}
void Link(int x,int y){
    anc[y][0]=x;
    dep[y]=dep[x]+1;
    vec_2[x].pb(y);
    for(int i=1;i<30;i++){
        anc[y][i]=anc[anc[y][i-1]][i-1];
    }
}
int main(){
    #ifdef local
    freopen("input2.txt","r",stdin);
    #endif // local
    int n;
    cin>>n;
    for(int i=1;i<=n;i++){
        int tem;
        tem=read();
        while(tem){
            du[i]+=1;vec_1[tem].pb(i);vec_fa[i].pb(tem);tem=read();
        }
    }
    dep[n+1]=1;
    for(int i=1;i<=n;i++){
            if(!du[i]){
                q.push(i);
                Link(n+1,i);
            }
    }
    while(q.size()){
        int x=q.front();q.pop();
        for(int i=0;i<vec_1[x].size();i++){
            int y=vec_1[x][i];
            du[y]-=1;
            if(!du[y]){
                int fa=vec_fa[y][0];
                for(int j=1;j<vec_fa[y].size();j++){
                    fa=lca(fa,vec_fa[y][j]);
                }
               Link(fa,y);
                 q.push(y);
            }
        }
    }
        dfs(n+1);
        for(int i=1;i<=n;i++){
            cout<<sz[i]-1<<en;
        }
    return 0;
}

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值