UVA 1513 Movie collection

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#define N 200010
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1

int sum[N<<4],tag[N/2];
int n,q,cnt;

void PushUp(int rt)
{
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}

void build(int l,int r,int rt)
{
    if(l==r)
    {
        if((l>=N-n)&&l<N)
            sum[rt]=1,tag[++cnt]=l;
        return;
    }
    int m=(l+r)>>1;
    build(lson);
    build(rson);
    PushUp(rt);
}

void update(int i,int c,int l,int r,int rt)
{
    if(l==r)
    {
        sum[rt]=c;
        return;
    }
    int m=(l+r)>>1;
    if(tag[i]<=m)
        update(i,c,lson);
    else update(i,c,rson);
    PushUp(rt);
}

int query(int L,int R,int l,int r,int rt)
{
    if(L<=l&&R>=r)
        return sum[rt];
    int m=(l+r)>>1;
    int res=0;
    if(L<=m)
        res+=query(L,R,lson);
    if(R>m)
        res+=query(L,R,rson);
    return res;
}

int main(void)
{
    int T,x;
    scanf("%d",&T);
    while(T--)
    {
        memset(sum,0,sizeof(sum));
        memset(tag,0,sizeof(tag));
        int pre;
        scanf("%d%d",&n,&q);
        cnt=0;
        build(1,N-1,1);
        pre=tag[1];
        while(q--)
        {
            scanf("%d",&x);
            int ans=0;
            if(pre<=tag[x]-1)
                { update(x,0,1,N-1,1);
                    ans=query(pre,tag[x]-1,1,N-1,1),tag[x]=--pre,update(x,1,1,N-1,1);}
            printf("%d%c",ans,!q?'
':' ');
        }
    }
    return 0;
}