版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/baidu_36797646/article/details/86144590
Solution
简单题。
先dfs一次求出球到的节点顺序,对于
,可以把球逐个放入,在数据结构上二分,找到第一个没有球的节点;再维护每个点到根上的球数,对于
,就是询问这个东西。我用了两个树状数组。
Code
#include<bits/stdc++.h>
using namespace std;
#define LL long long
#define pa pair<int,int>
const int Maxn=100010;
const int inf=2147483647;
int read()
{
int x=0,f=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch>='0'&&ch<='9')x=(x<<3)+(x<<1)+(ch^48),ch=getchar();
return x*f;
}
int n,q,root,a[Maxn],b[Maxn],in[Maxn],out[Maxn],dfn=0,tot=0,cnt=0,fa[Maxn][17],dep[Maxn];
struct Edge{int y,next;}e[Maxn];
int last[Maxn],len=0;
void ins(int x,int y)
{
if(!x){root=y;return;}
int t=++len;e[t].y=y;e[t].next=last[x];last[x]=t;
}
struct Bit
{
int s[Maxn];
int query_sum(int x){int re=0;for(;x;x-=(x&-x))re+=s[x];return re;}
void add(int x,int y){for(;x<=n;x+=(x&-x))s[x]+=y;}
int query_pos(int x)
{
int re=0,v=0;
for(int i=16;i>=0;i--)
if(re+(1<<i)<n&&v+s[re+(1<<i)]<x)re+=(1<<i),v+=s[re];
return re+1;
}
}T1,T2;
int mn[Maxn];
struct Node
{
int x,id;
Node(int _x,int _id){x=_x,id=_id;}
};
bool cmp(Node a,Node b){return a.x<b.x;}
vector<Node>h[Maxn];
void DFS(int x)
{
mn[x]=x;
for(int i=last[x];i;i=e[i].next)
{
int y=e[i].y;
DFS(y);mn[x]=min(mn[x],mn[y]);
}
}
void dfs(int x)
{
in[x]=++dfn;
for(int i=1;(1<<i)<=dep[x];i++)fa[x][i]=fa[fa[x][i-1]][i-1];
for(int i=last[x];i;i=e[i].next)h[x].push_back(Node(mn[e[i].y],e[i].y));
sort(h[x].begin(),h[x].end(),cmp);
for(int i=0;i<h[x].size();i++)dep[h[x][i].id]=dep[x]+1,fa[h[x][i].id][0]=x,dfs(h[x][i].id);
out[x]=dfn;a[x]=++tot;b[tot]=x;
}
int jump(int x,int y)
{
for(int i=16;i>=0;i--)
if((1<<i)<=y)x=fa[x][i],y-=(1<<i);
return x;
}
int main()
{
n=read(),q=read();
for(int i=1;i<=n;i++)ins(read(),i);
for(int i=1;i<=n;i++)T1.s[i]=i&(-i),T2.s[i]=0;
DFS(root);dep[root]=0;dfs(root);
while(q--)
{
int op=read(),x=read();
if(op==1)
{
x=min(x,n-cnt);cnt+=x;int p;
for(int i=1;i<=x;i++)
{
p=T1.query_pos(1);
T1.add(p,-1);
T2.add(in[b[p]],1);if(out[b[p]]<n)T2.add(out[b[p]]+1,-1);
}
printf("%d\n",b[p]);
}
else
{
cnt--;
int t=T2.query_sum(in[x])-1;
printf("%d\n",t);
int p=jump(x,t);
T1.add(a[p],1);
T2.add(in[p],-1);if(out[p]<n)T2.add(out[p]+1,1);
}
}
}