树状数组模板

#include<iostream>
using namespace std;

const int max_n=8010;
int tree[max_n];
int a[max_n],b[max_n];
bool used[max_n];
int n;

int lowbit(int x)
{
	return x&-x;
}

void updata(int idx,int v)
{
	while(idx<=n)
	{
		tree[idx]+=v;
		idx+=lowbit(idx);
	}
}

int count(int idx)
{
	int sum=0;
	while(idx>0)
	{
		sum+=tree[idx];
		idx-=lowbit(idx);
	}
	return sum;
}

int find(int x)              //在树状数组中用二分法查找总值为x的叶子节点的下标
{
	int l=1,r=n;
	int mid,t;
	while(l<=r)
	{
		mid=(l+r)>>1;
		t=count(mid);
		if(t<x) l=mid+1;
		else if(t>x) r=mid;
		else                                  //第一次查找到的叶子节点可能是已经使用过得，所以得特判，否则会进入死循环
		{                                     //假如该叶子节点是已经使用过得，则从该节点往后寻找第一个未使用过的叶子节点
			if(used[mid]!=0) r=mid;      
			else 
			{
				l=mid;r=mid-1;
			}
		}
	}
	return l;
}

int main()
{
	int i;
	scanf("%d",&n);
	for(i=2;i<=n;i++)
	{
		scanf("%d",&a[i]);a[i]++;
	}
	a[1]=1;
	for(i=1;i<=n;i++)
		updata(i,1);
	int t;
	memset(b,-1,sizeof(b));
	memset(used,0,sizeof(used));
	for(i=n;i>=1;i--)
	{
		t=find(a[i]);
		updata(t,-1);
		used[t]=1;                   //为使用过的叶子节点标记
		b[i]=t;
	}
	for(i=1;i<=n;i++)
		printf("%d\n",b[i]);
	return 0;
}