[BZOJ]3133: [Baltic2013]ballmachine 树状数组

Solution

简单题。
先dfs一次求出球到的节点顺序，对于$op=1$，可以把球逐个放入，在数据结构上二分，找到第一个没有球的节点；再维护每个点到根上的球数，对于$op=2$，就是询问这个东西。我用了两个树状数组。

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);
		}
	}
}