题目描述
https://www.lydsy.com/JudgeOnline/problem.php?id=3133
题解
还是分两个操作来说吧。
先看第一个操作,放球,可以发现,对于祖先节点和后代节点来说,后代节点没有放球,祖先节点就不能放,对于兄弟节点来说,子树编号小的没放满,编号大的就不能放,可以发现这是树的后序遍历。
用堆维护这个顺序,依次放就好了。
第二问更简单,就是找到最靠上的有球的祖先,把那个位置拿走就好了。
代码
#include<iostream>
#include<cstdio>
#include<queue>
#include<algorithm>
#include<vector>
#define N 100009
using namespace std;
int p[][N],deep[N],n,dfn[N],tot,root,dp[N];
bool tag[N];
struct node{
int x;
bool operator <(const node &b)const{
return dfn[x]>dfn[b.x];
}
};
priority_queue<node>q;
vector<int>vec[N];
inline int rd(){
int x=;char c=getchar();bool f=;
while(!isdigit(c)){if(c=='-')f=;c=getchar();}
while(isdigit(c)){x=(x<<)+(x<<)+(c^);c=getchar();}
return f?-x:x;
}
inline bool cmp(int a,int b){return dp[a]<dp[b];}
void predfs(int u){
dp[u]=u;
for(int i=;i<vec[u].size();++i){
int v=vec[u][i];predfs(v);
dp[u]=min(dp[u],dp[v]);
}
}
void dfs(int u){
for(int i=;(<<i)<=deep[u];++i)p[i][u]=p[i-][p[i-][u]];
for(int i=;i<vec[u].size();++i){
int v=vec[u][i];deep[v]=deep[u]+;p[][v]=u;
dfs(v);
}
dfn[u]=++tot;
}
int main(){
n=rd();int qu=rd();int x,opt,fa;
for(int i=;i<=n;++i){fa=rd();if(!fa)root=i;vec[fa].push_back(i);}
predfs(root);
for(int i=;i<=n;++i)sort(vec[i].begin(),vec[i].end(),cmp);
dfs(root);
for(int i=;i<=n;++i)q.push(node{i});
while(qu--){
opt=rd();x=rd();
if(opt==){
int id;
for(int i=;i<=x;++i){
id=q.top().x;q.pop();
tag[id]=;
}
printf("%d\n",id);
}
else{
int now=x;
for(int i=;i>=;--i)if(p[i][now]&&tag[p[i][now]])now=p[i][now];
tag[now]=;q.push(node{now});printf("%d\n",deep[x]-deep[now]);
}
}
return ;
}