题解 P2597 【[ZJOI2012]灾难】
分析
考场上看出是道支配树?不会写也不想写
后来发现只是个弱化版的DAG上支配树
Code
#include<bits/stdc++.h>
using namespace std;
int n,ind[100010],topo[100010],CNT,fa[100010],f[100010][17],dep[100010],ip,SIZE[100010];
vector<int>a[100010],b[100010];
queue<int>q;
void toposort()
{
for(int i=1;i<=n;i++)
if(!ind[i])
q.push(i);
while(!q.empty())
{
int T=q.front();
q.pop();
topo[++CNT]=T;
for(int i=0;i<b[T].size();i++)
{
ind[b[T][i]]--;
if(!ind[b[T][i]])
q.push(b[T][i]);
}
}
}
void RMQ(int x)
{
f[x][0]=fa[x];
for(int i=1;i<=16;i++)
f[x][i]=f[f[x][i-1]][i-1];
}
int LCA(int x,int y)
{
if(dep[x]<dep[y])
swap(x,y);
for(int i=16;i>=0;i--)
if(f[x][i]&&dep[f[x][i]]>=dep[y])
x=f[x][i];
if(x==y)
return x;
for(int i=16;i>=0;i--)
{
if(f[x][i]&&f[y][i]&&f[x][i]!=f[y][i])
{
x=f[x][i];
y=f[y][i];
}
}
return fa[x];
}
void build()
{
dep[n+1]=1;
fa[n+1]=n+1;
for(int i=n;i>=1;i--)
{
int T=topo[i];
if(!b[T].size())
{
a[n+1].push_back(T);
fa[T]=n+1;
f[T][0]=n+1;
dep[T]=2;
continue;
}
int lca=b[T][0];
for(int j=0;j<b[T].size();j++)
{
lca=LCA(lca,b[T][j]);
}
fa[T]=lca;
dep[T]=dep[lca]+1;
a[lca].push_back(T);
RMQ(T);
}
}
void DFS(int x)
{
SIZE[x]=1;
for(int i=0;i<a[x].size();i++)
{
DFS(a[x][i]);
SIZE[x]+=SIZE[a[x][i]];
}
}
int main()
{
//freopen("catas.in","r",stdin);
//freopen("catas.out","w",stdout);
scanf("%d",&n);
for(int i=1;i<=n;i++)
{
while(scanf("%d",&ip)!=EOF&&ip)
{
ind[ip]++;
b[i].push_back(ip);
}
}
toposort();
build();
DFS(n+1);
for(int i=1;i<=n;i++)
{
printf("%d\n",SIZE[i]-1);
}
}