【堆+倍增】[BalticOI 2013]Ball Machine 发球机器

题目链接

题意：给一棵树，在根节点放一些球，选择编号小的空节点滚下去。从一个位置上拿走球，上面的球也会滚下来。

先来模拟一遍样例，看看球是怎么滚的：

初始节点均为空，放第一个球，落到2节点，然后落到编号较小的3节点，最后落到编号较小的5节点。依次类推，可得出在根节点放8个球（放满树）的顺序（如图蓝色序号）

发现这个落球顺序是固定的。例如如果放3个球必然落到5,8,6三个节点，而不会落到别的节点。也就是说，当有若干个空节点时，来一个球，必然落到优先级更高（按落球顺序从小到大）的节点。考虑用优先队列维护这些空节点，来一个球取出堆顶打上标记。

当拿出一个球时，上面的球顺次往下落，即这个球掉下去它上面的球会落下来补上这个空位，直到这个点原来上面没有球来补它的空位，那么这个位置就由有球节点变成了空节点。也就是要找这个拿走球的节点到根节点路径上深度最浅的有球节点，可以倍增来找。

#include <bits/stdc++.h>
using namespace std;
int n,Q,x,rt,top,timber,opt,dfn[100005],f[100005][21],s[100005],dep[100005];
bool vis[100005];vector<int> e[100005];
struct node{int x;};
bool operator<(const node &xx,const node &yy){return dfn[xx.x]>dfn[yy.x];}
priority_queue<node> q;
inline bool cmp(int x,int y){return s[x]<s[y];}
void dfs(int u){
	s[u]=u;
	for(int i=0;i<e[u].size();i++){
		int v=e[u][i];dfs(v);
		s[u]=min(s[u],s[v]);
	}
}
void DFS(int u){
	for(int i=1;i<=19;i++)f[u][i]=f[f[u][i-1]][i-1];
	for(int i=0;i<e[u].size();i++){
		int v=e[u][i];
		dep[v]=dep[u]+1;f[v][0]=u;
		DFS(v);
	}
	dfn[u]=++timber;
}
int main(){
	scanf("%d%d",&n,&Q);
	for(int i=1;i<=n;i++){
		scanf("%d",&x);if(x==0)rt=i;
		e[x].push_back(i);
	}
	dep[rt]=1;dfs(rt);
	for(int i=1;i<=n;i++)sort(e[i].begin(),e[i].end(),cmp);
	DFS(rt);
	for(int i=1;i<=n;i++)q.push(node{i});
	while(Q--){
		scanf("%d%d",&opt,&x);
		if(opt==1){
			for(int i=1;i<=x;i++){
				top=q.top().x;q.pop();vis[top]=1;
			}
			printf("%d\n",top);
		}else{
			top=x;
			for(int i=19;i>=0;i--)
			    if(vis[f[top][i]])top=f[top][i];	
			vis[top]=0;
			q.push((node){top});
			printf("%d\n",dep[x]-dep[top]);
		}
	}
	return 0;
}