#2815. [ZJOI2012]灾难

因为是一个有向无环图,一个生物灭绝的条件是它的食物都灭绝.

那么我们考虑按topotopo序重构这张图,因为一定是一个森林/树,我们考虑增加一个超级源点uu.然后且当它的祖先节点灭绝它就会灭绝,考虑把它**到它所有食物的lcalca下面,然后灭绝它的毁灭值就是1子树大小-1啦.

树的lcalca的倍增和rmqrmq的倍增是有一定的区别的,rmqrmq包含自己,而树不包含自己节点.

code:code:

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;

const int N=1e5+5;
const int M=17;
const int mod=998244353;

vector<int>g[N],G[N];
int d[N],n,dep[N];
int sx[N],id=0;
int f[N][M];
int ans[N];

void dfs(int u)
{
	ans[u]=1;
	for(int v:G[u])
	{
		dfs(v);
		ans[u]+=ans[v];
	}
}


int LCA(int u,int v)
{
	if(dep[u]<dep[v])	swap(u,v);
	for(int i=16;i>=0;i--)
	{
		if(dep[f[u][i]]>=dep[v])	u=f[u][i];
	}
	if(u==v)	return u;
	for(int i=16;i>=0;i--)
	{
		if(f[u][i]!=f[v][i])	u=f[u][i],v=f[v][i];
	}
	return f[u][0];
}

void pre(int u)
{
	for(int i=1;i<=16;i++)
	{
		f[u][i]=f[f[u][i-1]][i-1];
	}
}

void topo()
{
	queue<int>Q;
	for(int i=1;i<=n;i++)
		if(!d[i])	Q.push(i),sx[++id]=i;
	while((int)Q.size())
	{
		int x=Q.front();
		Q.pop();
		for(int u:g[x])
		{
			d[u]--;
			if(d[u]==0)	sx[++id]=u,Q.push(u);
		}
	}
	dep[n+1]=1;
	for(int i=id;i>=1;i--)
	{
		if((int)g[sx[i]].size()==0)
		{
			f[sx[i]][0]=n+1;
			dep[sx[i]]=2;
			G[n+1].push_back(sx[i]);
			continue;
		}
		int lca=g[sx[i]][0];
		for(int u:g[sx[i]])
		{
			lca=LCA(lca,u);
		}
		G[lca].push_back(sx[i]);
		dep[sx[i]]=dep[lca]+1;
		f[sx[i]][0]=lca;
		pre(sx[i]);
	}
}

void run()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
	{
		int x;
		while(true)
		{
			scanf("%d",&x);
			if(x==0)	break;
			g[i].push_back(x);
			d[x]++;
		}
	}
	topo();
	dfs(n+1);
	for(int i=1;i<=n;i++)
		ans[i]--,printf("%d\n",ans[i]);
}

int main()
{
	int T=1;
	//scanf("%d",&T);
	while(T--)	run();
	return 0;
}
lpt的小屋 文章被收录于专栏

我想要一份甜甜的爱情

全部评论
插入??
点赞 回复 分享
发布于 2022-04-14 17:33

相关推荐

评论
点赞
收藏
分享
牛客网
牛客企业服务