POJ2182 线段树维护区间可用值

题意

给你n 下面n - 1个数表示从第2个到第n个——第i个cow前面有多少只编号比它小的cows

问你这n只cows的排列。

思路

线段树维护区间可以用的编号有几个。

从后先前枚举 ans[i]是每次前面有a[i]个可用的答案

找到这一位的答案后 删除已经使用的编号更新线段树

Accode

#include<cstdio>
#include<iostream>
#include<algorithm>

using namespace std;
const int MaxN = 8e4 + 5;
const int inf = 0x3f3f3f3f;

int n;
int a[MaxN],ans[MaxN];

struct NODE{
	int l,r,w;
}tree[MaxN];

void build(int k,int ll,int rr){
	tree[k].l = ll;
	tree[k].r = rr;
	tree[k].w = rr - ll + 1;
	if(ll == rr){
		return ;
	}
	int m = (ll + rr) / 2;
	build(2 * k,ll,m);
	build(2 * k + 1,m + 1,rr);
}

int ask_p(int k,int x){
	if(tree[k].l == tree[k].r){
		return tree[k].l;
	}
	if(tree[k * 2].w >= x) return ask_p(k * 2,x);
	else return ask_p(k * 2 + 1,x - tree[k * 2].w);//已经继承前半部分
}

void chan_p(int k,int x){
	if(tree[k].l == tree[k].r){
		tree[k].w = 0;
		return ;
	}
	int m = (tree[k].l + tree[k].r) / 2;
	if(x <= m) chan_p(k * 2,x);
	else chan_p(k * 2 + 1,x);
	tree[k].w = tree[k * 2].w + tree[k * 2 + 1].w;
}

int main()
{
	scanf("%d",&n);
	for(int i = 2;i <= n; i++) scanf("%d",&a[i]);
	a[1] = 0;
	build(1,1,n);//区间内有 how many 可用
	for(int i = n;i >= 1; i--){
		ans[i] = ask_p(1,a[i] + 1);//统计到前面需要 x 个 这意味到哪
		chan_p(1,ans[i]);
//		cout << tree[1].w << "#"<<ans[i] <<endl;
	}
	for(int i = 1;i <= n; i++) printf("%d\n",ans[i]);
}