2016夏季练习——线段树

来源：POJ2182

解释在注释里面

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
const int MAXN = 8000+10;
struct Tree{
	int l,r;
	int len;
	Tree():len(0){}
};
int n;
int a[MAXN];
int ans[MAXN];
Tree tree[MAXN*4];
void build(int rt,int left,int right){
	tree[rt].l=left;
	tree[rt].r=right;
	tree[rt].len=right-left+1;
	if(left==right) return ;
	int mid=(left+right)>>1;
	build(rt<<1,left,mid);
	build(rt<<1|1,mid+1,right);
}
int query(int rt,int tot){
	tree[rt].len--;
	//关键就是这一步！！！
	//我们访问这个节点，也就意味着这个节点所表示的区间中有我需要的那个值
	//那么我取走那个值的时候，对应后面的数就不能把这个节点数算在里面
	//cout<<rt<<" "<<tree[rt].l<<" "<<tree[rt].r<<" "<<tree[rt].len<<" "<<tot<<endl;
	if(tree[rt].l==tree[rt].r) {
		//cout<<"leaves!!!"<<endl;
		return tree[rt].l;
	}
	//下面是访问策略
	//根据上面的分析可以知道
	//结构中len的使用其实不能叫做len
	//表示的是当前区间中小于r但是还没有被使用过的数字
	//这样的话下面的这个就是显然的了
	if(tot>tree[rt<<1].len) {
		//cout<<"right"<<endl;
		return query(rt<<1|1,tot-tree[rt<<1].len);
	}
	else {
		//cout<<"left"<<endl;
		return query(rt<<1,tot);
	}
}
int main(){
	scanf("%d",&n);
	build(1,1,n);
	memset(a,0,sizeof(a));
	/*for(int i=1;i<=2*n-1;i++)
		cout<<i<<" "<<tree[i].l<<" "<<tree[i].r<<" "<<tree[i].len<<endl;*/
	for(int i=2;i<=n;i++){
		scanf("%d",a+i);
	}

	//注意，我们最后一个数其实有着自己的性质：因为前面的所有数都已经出来了
	//所以我们最后一个一定是a[i]+1
	//这样，我们继续向前的时候，我们将遇到一个问题：
	//我已经用过的这个数应该是怎么办？
	//我们去query函数中看这个问题。
	for(int i=n;i>0;i--){
		//cout<<"n="<<i<<": a[i]="<<a[i]<<endl;
		ans[i]=query(1,a[i]+1);
	}
	for(int i=1;i<=n;i++){
		cout<<ans[i]<<endl;
	}
	return 0;
}