ZOJ 3635 线段树

来源：互联网发布：好用的网络机顶盒软件编辑：程序博客网时间：2024/05/18 15:05

线段树维护的是区间有多少个空位置，每次查询第X个空位置在哪，sum[rt]>=X就向左区间找，sum[rt]<X就向又区间找。

#include <cstdio>#include <algorithm>#include <iostream>using namespace std;#define lson l , m , rt << 1#define rson m + 1 , r , rt << 1 | 1const int maxn = 55555;int n,m;int sum[maxn<<2],a[maxn];void pushup(int rt){    sum[rt]=sum[rt<<1]+sum[rt<<1|1];}void build(int l,int r,int rt) {    if (l==r) {sum[rt]=1;return;}    int m=(l+r)>>1;    build(lson);    build(rson);    pushup(rt);}int update(int l,int r,int rt,int cnt){    int ret,m=(l+r)>>1;    if(l==r)    {        sum[rt]=0;        return l;    }    if(cnt<=sum[rt<<1]) ret=update(lson,cnt);    else ret=update(rson,cnt-sum[rt<<1]);    pushup(rt);    return ret;}int main(){    int x;    while(~scanf("%d",&n))    {        build(1,n,1);        for(int i=1;i<=n;i++)        {            scanf("%d",&x);            a[i]=update(1,n,1,x);        }        scanf("%d",&m);        for(int i=1;i<=m;i++)        {            scanf("%d",&x);            if(i==1) printf("%d",a[x]);            else printf(" %d",a[x]);        }        puts("");    }    return 0;}

4 0