zoj 3635 线段树 区间求位置 单点更新

zoj 3635   http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemId=4803

/*

题意：输入一个n，表示有 n 个空位，接下来就是 n 个数，
第 i 个数 ai 表示第 i 个人要在坐在第 ai 个空位上；在输入一个 m 表示有 m 次访问，
每次访问输入一个整数 bi 表示第 bi 个人坐的位置标号。
分析：采用线段树，线段树里存放的是这一段里有多少个空位，从下往上更新，时间复杂度O(nlogn)。

*/

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<memory.h>
using namespace std;
#define lson rt<<1,l,mid
#define rson rt<<1|1,mid+1,r
const int maxn=50002;
int sum[maxn<<2],pos[maxn<<2];
void pushUP(int rt)
{
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
//void pushDOWN(int rt)
void build(int rt,int l,int r)
{
    if(l==r)
    {
        sum[rt]=1;
        return;
    }
    int mid=(l+r)>>1;
    build(lson);
    build(rson);
    pushUP(rt);
}
void updata(int rt,int l,int r,int ans,int num,int idx)
{
    if(l==r)
    {
        sum[rt]=0;
        pos[idx]=l;
       /// pushUP(rt);
        return;
    }
    int mid=(l+r)>>1;
    if(sum[rt<<1]+ans<num)
        updata(rson,ans+sum[rt<<1],num,idx);
    else updata(lson,ans,num,idx);
    pushUP(rt);
}
int main()
{
    int i,j,n,m,bi,idx;
    while(scanf("%d",&n)!=EOF)
    {
        memset(sum,0,sizeof(sum));
        memset(pos,0,sizeof(pos));
        build(1,1,n);
        for(i=1; i<=n; i++)
        {
            scanf("%d",&bi);
            updata(1,1,n,0,bi,i);
        }
        scanf("%d",&m);
        if(m>=1)
        {
            scanf("%d",&idx);
            printf("%d",pos[idx]);
        }
        for(i=2; i<=m; i++)
        {
            scanf("%d",&idx);
            printf(" %d",pos[idx]);
        }
        printf("\n");
    }
    return 0;
}