题意:求灭绝树下的子树大小
解:
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;
}