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.线段树维护区间里面未标记的牛的数量,找到满足条件的区间即可。找到后同时更新一下