有一个1-n的序列,第i个数知道在它前面且比它小的数的个数,求该序列。
线段树,点修改,点查询。
更细致一点,是权值线段树,查区间第k大
#include <cstdio>
const int N = 8000 + 10;
int n,sumv[N<<2],a[N],ans[N];
void build(int rt,int L,int R){
if(L == R) {
sumv[rt] = 1;
return ;
}
int mid = (L+R)/2;
build(rt<<1,L,mid);
build(rt<<1|1,mid+1,R);
sumv[rt] = sumv[rt<<1] + sumv[rt<<1|1];
}
void update(int rt,int L,int R,int idx,int pos){
if(L == R) {
sumv[rt]--;
ans[pos] = L;
return ;
}
int mid = (L + R) /2;
if(sumv[rt<<1] > idx) update(rt<<1,L,mid,idx,pos);
else update(rt<<1|1,mid+1,R,idx-sumv[rt<<1],pos);
sumv[rt] = sumv[rt<<1] + sumv[rt<<1|1];
}
int main(){
while(~scanf("%d",&n)){
a[0] = 0;
for(int i = 1;i<n;i++) scanf("%d",&a[i]);
build(1,1,n);
for(int i = n-1;~i;i--){
update(1,1,n,a[i],i);
}
for(int i = 0;i<n;i++) printf("%d\n",ans[i]);
}
return 0;
}