zoj 3635 Cinema in Akiba

zoj 3635 Cinema in Akiba

#include<cstdio>#include<cstring>using namespace std;const int maxn=50005;int sum[maxn<<2];int ans[maxn];#define lson l,m,rt<<1#define rson m+1,r,rt<<1|1void pushup(int rt){ sum[rt]=sum[rt<<1]+sum[rt<<1|1]; return;}void update(int p,int l,int r,int rt){ if(l==r) {sum[rt]=0;return;} int m=(l+r)>>1; if(p<=m) update(p,lson); else update(p,rson); pushup(rt);}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); return;}int query(int w,int l,int r,int rt){ if(l==r) return l; int m=(l+r)>>1; if(sum[rt<<1]>=w) return query(w,lson); if(sum[rt<<1]<w) return query(w-sum[rt<<1],rson);}int main(){ int n; while(scanf("%d",&n)!=EOF) { memset(ans,0,sizeof(ans)); build(1,n,1); int t,tt; for(int i=1;i<=n;i++) { scanf("%d",&t); ans[i]=query(t,1,n,1); update(ans[i],1,n,1); } scanf("%d",&t); int flag=0; for(int i=0;i<t;i++) { scanf("%d",&tt); if(flag) printf(" "); printf("%d",ans[tt]); flag=1; } printf("n"); } return 0;}