队内训练1 反向建DAG图拓扑+在线跑lca建树+dfs子树大小
题目链接link.
浙江省选题恐怖如斯 究极缝合怪罢了
题意:给你n个点,代表n个生物。接下来n行,每行若干个生物,代表第i个生物能吃掉这些生物。如果某个消费者的所有食物都灭绝了,它会跟着灭绝。我们把一个生物灭绝时会跟着他一起没了的生物的数量定义为灾难值。现在要求出每个生物的灾难值。
这个题思路很妙啊,老实说代码实现起来也不是那么简单毕竟缝合的东西这么多不好debug是吧
建图思路呢,就是如果a吃b,就连一条有向边从b指向a。也就是说这个图的入度为零的点应该是某个生产者。
然后跑拓扑排序得到一个序列。这个序列可以保证,我能吃的所有的食物都出现在我的前面。
最后,考虑怎么建树。我们可以把生物i的所有食物的lca算出来,然后将生物i与lca相连。这样建树就可以保证当一个节点上的生物灭绝时,他子树的所有生物都会随之灭绝。因为这是一边算lca一边往树里面加点一边更新,所以不能用预处理的方法了,只能每插入一个点logn查询lca,记得更新新加进去的点的信息。
AC代码: 位运算少打个括号wa了两天wa成傻逼了
#include<iostream>
#include<cstdio>
#include<vector>
#include<queue>
using namespace std;
struct qaq
{
int fa;
vector<int> son;
int sum;
}tree[70000];
int n;
vector<int>edge[70000];
vector<int>father[70000];
vector<int>ans;
queue<int> q;
int in[70000];
int fa[70000][20];
int dep[70000];
int lg(int x)
{
int ans=0;
while(x)
{
x=x/2;
ans++;
}
return ans;
}
int up(int x,int d)
{
int ret=x;
for(int i=0;(1<<i)<=n;i++)
if(((1<<i)&d)!=0) ret=fa[ret][i];
return ret;
}
int lca(int x,int y)
{
if(dep[x]<dep[y]) swap(x,y);
x=up(x,dep[x]-dep[y]);
if(x==y) return x;
int maxd=lg(dep[x]);
for(int i=maxd;i>=0;i--)
{
if(fa[x][i]!=fa[y][i])
x=fa[x][i],y=fa[y][i];
}
return fa[x][0];
}
int dfs(int x)
{
if(tree[x].son.size()==0)
{
tree[x].sum=1;
//cout<<"x="<<x<<endl;
return 1;
}
int ans=1;
for(auto i:tree[x].son)
{
ans+=dfs(i);
}
tree[x].sum=ans;
//cout<<"tree["<<x<<"]="<<ans<<endl;
return ans;
}
int main( )
{
scanf("%d",&n);
for(int i=1;i<=n;i++)
{
while(1)
{
int x;
scanf("%d",&x);
if(x==0) break;
//输入的x是i的父亲们(他i吃谁x)
//边应该是由x指向i的
edge[x].push_back(i);
father[i].push_back(x);
in[i]++;
}
}
for(int i=1;i<=n;i++)
{
if(in[i]==0)
q.push(i);
}
while(!q.empty())
{
int xx=q.front();
q.pop();
ans.push_back(xx);
for(auto i:edge[xx])
{
in[i]--;
if(in[i]==0)
q.push(i);
}
}
int summ=0;
tree[0].fa=0;
dep[0]=0;
fa[0][0]=0;
for(auto i: ans)
{
if( father[i].size()==0 )
{
tree[0].son.push_back(i);
tree[i].fa=0;
dep[i]=1;
fa[i][0]=0;
}
else if( father[i].size( )==1 )
{
int haha=father[i][0];
dep[i]=dep[haha]+1;
fa[i][0]=haha;
for(int j=1;(1<<j)<=n;j++)
{
fa[i][j]=fa[fa[i][j-1]][j-1];
//cout<<"fa["<<i<<"]["<<j<<"]="<<fa[i][j]<<endl;
}
tree[haha].son.push_back(i);
tree[i].fa=haha;
}
else
{
int lcaa=lca(father[i][0],father[i][1]);
for(int j=2;j<father[i].size();j++)
{
lcaa=lca(lcaa,father[i][j]);
if(lcaa==0) break;
}
dep[i]=dep[lcaa]+1;
fa[i][0]=lcaa;
for(int j=1;(1<<j)<=n;j++)
{
fa[i][j]=fa[fa[i][j-1]][j-1];
}
tree[lcaa].son.push_back(i);
tree[i].fa=lcaa;
}
}
int xi=dfs(0);
//cout<<"xi="<<xi<<endl;
for(int i=1;i<=n;i++)
printf("%d\n",tree[i].sum-1);
return 0;
}