题解 P2597 【[ZJOI2012]灾难】

luogu

分析

考场上看出是道支配树？不会写也不想写
后来发现只是个弱化版的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);
	}
}