谜一样的牛(树状数组)

题目传送门

思路

我现在有一组数据
5
0 1 2 1 0
那么从后往前看
第五个数也就是代表了所有的数中，从小到大排序排名第1的数字，然后把这个数删掉
第四个数也就是代表了剩下的所有的数中，从小到大排序排名第2的数字，然后把这个数删掉
第三个数也就是代表了剩下的所有的数中，从小到大排序排名第3的数字，然后把这个数删掉
第二个数也就是代表了剩下的所有的数中，从小到大排序排名第2的数字，然后把这个数删掉
第一个数也就是代表了剩下的所有的数中，从小到大排序排名第1的数字，然后把这个数删掉

那么可以用树状数组来直接搞一个前缀和，
因为a[1]=1 a[2]=2 a[3]=3 a[4]=4 a[5]=5
这样直接二分来查找这个位次的数字就可以
但是有一个问题就是我用的哪两个二分的板子？
应该是用查找最小值的板子？
为什么？
因为查找最小值，是刚好是前缀和数组恰好变换的时刻，因此这一点处一定是1.

绿色的点恰好就是我们要找的点。
因此用最小值的那个板子。
下面是ac代码

#include<iostream>
using namespace std;
const int N=21021000;
int n;
int a[N];
int tree[N];
int ans[N];
int lowbit(int x){
	return x&(-x);
}
void add(int i,int x){
	for(;i<=n;i+=lowbit(i)){
		tree[i]+=x;
	} 
}
int query(int x){
	int res=0;
	for(int i=x;i;i-=lowbit(i)){
		res+=tree[i];
	}
	return res;
}
int bsearch(int l,int r,int ask){
	int mid;
	while(l<r){
		mid=l+r+1>>1;
		if(query(mid)<=ask)l=mid;
		else r=mid-1;
	}
	return l;
}
int main(){
	cin>>n;
	for(int i=2;i<=n;i++)cin>>a[i];
	for(int i=1;i<=n;i++)add(i,1);
	for(int i=n;i;i--){
		ans[i]=bsearch(1,n,a[i]+1);
		add(ans[i],-1);
	}
	for(int i=1;i<=n;i++)cout<<ans[i]<<endl;
}