浙大月赛C题（2012/8）Cinema in Akiba（线段树）

http://acm.zju.edu.cn/onlinejudge/showContestProblem.do?problemId=4791

（1）第一次写浙大的题目，这题让我十分意外，基本的线段树类型（求第x个空位）。

（2）电影院里，一次插入一个人，让他坐在第x个空位shang （每个x均给定)。求最终的位置安排（制定一些人，求他们各自的座号）。

具体代码：

#include<stdio.h>
#define lson l, m, rt<<1
#define rson m+1, r, rt<<1|1
const int maxn=51000;
int n, m;
int pos[maxn], sum[maxn<<2];
void pushup(int rt)
{
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
void build(int l, int r, int rt)
{
    sum[rt]=r-l+1;
    if(l==r)
    {
        return ;
    }
    int m=l+r>>1;
    build(lson);
    build(rson);
}
void insert(int p, int i, int l, int r, int rt)
{
    if(l==r)
    {
        sum[rt]=0;
        pos[i]=l;
        return ;
    }
    int m=l+r>>1;
    if(p<=sum[rt<<1]) insert(p, i, lson);
    else insert(p-sum[rt<<1], i, rson);
    pushup(rt);
}
int main()
{
    while(scanf("%d", &n)!=EOF)
    {
        build(1, n, 1);
        for(int i=1;i<=n;i++)
        {
            int p;
            scanf("%d", &p);
            insert(p, i, 1, n, 1);
        }
        scanf("%d", &m);
        for(int i=1;i<=m;i++)
        {
            int x;
            scanf("%d", &x);
            if(i!=1) printf(" %d", pos[x]);
            else printf("%d", pos[x]);
        }
        printf("\n");
    }
    return 0;
}