3133: [Baltic2013]ballmachine

题解：

第一步，读懂题，可以看下discuss
那么对每个球求出优先级，开个堆记录当前有什么空位
再开个线段树记录每个点最上面的点的深度，倍增上去修改即可
code：

#include<queue>
#include<vector>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
int n,Q,root,ys[100010],la[100010],z=0,id[100010],s[100010],num=0,Min[100010];
vector<int> vec[100010];
struct trnode{
	int lc,rc,c,fa[20],dep,u;
}tr[200010];int tot=0;
priority_queue<int,vector<int>,greater<int> > q;
void dfs(int x,int fa)
{
	ys[x]=++z;tr[x].fa[0]=fa;tr[x].dep=tr[fa].dep+1;
	for(int i=1;(1<<i)<=tr[x].dep;i++)
		tr[x].fa[i]=tr[tr[x].fa[i-1]].fa[i-1];
	for(int i=0;i<vec[x].size();i++)
	{
		int y=vec[x][i];
		dfs(y,x);
	}
	id[x]=++num;s[num]=x;la[x]=z;
}
int bt(int l,int r)
{
	int x=++tot;
	if(l!=r)
	{
		int mid=(l+r)/2;
		tr[x].lc=bt(l,mid);
		tr[x].rc=bt(mid+1,r);
	}
	return x;
}
void pushdown(int x)
{
	int lc=tr[x].lc,rc=tr[x].rc,c=tr[x].u;tr[x].u=0;
	tr[lc].c+=c;tr[rc].c+=c;tr[lc].u+=c;tr[rc].u+=c;
}
void change(int x,int l,int r,int fl,int fr,int c,int op)
{
	if(l==fl&&r==fr)
	{
		if(op) tr[x].c=c;
		else tr[x].c+=c,tr[x].u+=c;
		return;
	}
	if(tr[x].u) pushdown(x);
	int mid=(l+r)/2;
	if(fr<=mid) change(tr[x].lc,l,mid,fl,fr,c,op);
	else if(fl>mid) change(tr[x].rc,mid+1,r,fl,fr,c,op);
	else change(tr[x].lc,l,mid,fl,mid,c,op),change(tr[x].rc,mid+1,r,mid+1,fr,c,op);
}
int findans(int x,int l,int r,int k)
{
	if(l==r) return tr[x].c;
	if(tr[x].u) pushdown(x);
	int mid=(l+r)/2;
	if(k<=mid) return findans(tr[x].lc,l,mid,k);
	return findans(tr[x].rc,mid+1,r,k);
}
int solve(int x,int y)
{
	for(int i=19;i>=0;i--)
		if((1<<i)<=tr[x].dep-y) x=tr[x].fa[i];
	return x;
}
void pre(int x)
{
	Min[x]=x;
	for(int i=0;i<vec[x].size();i++)
	{
		int y=vec[x][i];pre(y);
		Min[x]=min(Min[x],Min[y]);
	}
}
bool cmp(int a,int b) {return Min[a]<Min[b];}
int main()
{
	scanf("%d %d",&n,&Q);
	for(int i=1;i<=n;i++)
	{
		int fa;scanf("%d",&fa);
		if(fa==0) root=i;
		else vec[fa].push_back(i);
	}
	pre(root);
	for(int i=1;i<=n;i++)
		if(vec[i].size()) sort(vec[i].begin(),vec[i].end(),cmp);
	tr[0].dep=-1;dfs(root,0);
	bt(1,n);
	for(int i=1;i<=n;i++) q.push(i);
	while(Q--)
	{
		int op,c;scanf("%d %d",&op,&c);
		if(op==1)
		{
			int t;
			while(c--)
			{
				t=q.top();q.pop();t=s[t];
				change(1,1,n,ys[t],ys[t],tr[t].dep,1);
				if(ys[t]!=la[t]) change(1,1,n,ys[t]+1,la[t],-1,0);
			}
			printf("%d\n",t);
		}
		else
		{
			int y=findans(1,1,n,ys[c]),x=solve(c,y);
			printf("%d\n",tr[c].dep-tr[x].dep);
			q.push(id[x]);
			change(1,1,n,ys[x],ys[x],0,1);
			if(ys[x]!=la[x]) change(1,1,n,ys[x]+1,la[x],1,0);
		}
	}
}