Lost Cows poj2182（線段樹）

https://vjudge.net/contest/370568#problem/S

給出n頭牛前面有多少頭比他編號少的數目，求出原來的牛的編號，

樣例意思就是有5個數，第二個數前面有1個比他小的，第三個數前面有兩個比他小的，以此類推。

最後一個數是確定的，爲k+1.

#include<iostream>
using namespace std;
int sum[100005];
int a[100005],ql,b[100005];
void build(int o,int l,int r){//建樹並且每個值都賦值爲1 
	if(l==r){
		sum[o]=1;
		return;
	}
	int m=(l+r)>>1;
	build(o*2,l,m);
	build(o*2+1,m+1,r);
	sum[o]=sum[o*2]+sum[o*2+1];
}
int qu(int o,int l,int r){//不斷修改k+1的那個值，不斷減去區間和，表示前面還有幾個 
	int m=(l+r)>>1;
	if(l==r)
		return r;
	else if(sum[o*2]>=ql){
		return qu(o*2,l,m);
	}
	else{
		ql-=sum[o*2];
		return qu(o*2+1,m+1,r);
	}
}
void up(int o,int l,int r){//修改區間 
	if(l==r){
		sum[o]--;
		return;
	}
	int m=(l+r)>>1;
	if(ql>m)
		up(o*2+1,m+1,r);
	if(ql<=m)
		up(o*2,l,m);
	sum[o]=sum[o*2]+sum[o*2+1];
}
int main(){
	int n;
	scanf("%d",&n);
	for(int i=1;i<n;i++)
		scanf("%d",&a[i]);
	a[0]=0;
	build(1,1,n);
	for(int i=n-1;i>=0;i--){
		ql=a[i]+1;
		b[i]=qu(1,1,n);
		ql=b[i];
		up(1,1,n);
	}
	for(int i=0;i<n;i++)
		printf("%d\n",b[i]);
}

從後面開始遍歷，碰到的a表示此牛在剩餘的牛中排名a+1.線段樹維護區間裏面未標記的牛的數量，找到滿足條件的區間即可。找到後同時更新一下