HDU-4031-Attack(树状数组)

#include <stdio.h>

int n,node[20005],ls[20005],rs[20005],pos[20005],c[20005];

int lowbit(int x)
{
    return x & -x;
}

int sum(int x)
{
    int res=0;

    while(x>0)
    {
        res+=node[x];

        x-=lowbit(x);
    }

    return res;
}

void add(int x,int val)
{
    while(x<=n)
    {
        node[x]+=val;

        x+=lowbit(x);
    }
}

int main()
{
    int T,q,t,i,a,b,cnt,cases=1;
    char s[10];

    scanf("%d",&T);

    while(T--)
    {
        scanf("%d%d%d",&n,&q,&t);

        cnt=0;

        for(i=0;i<=n;i++) c[i]=node[i]=pos[i]=0;

        printf("Case %d:\n",cases++);

        while(q--)
        {
            scanf("%s",s);

            if(s[0]=='A')
            {
                scanf("%d%d",&a,&b);

                ls[cnt]=a;
                rs[cnt]=b;

                cnt++;

                add(a,1);
                add(b+1,-1);
            }
            else
            {
                scanf("%d",&a);

                for(i=pos[a];i<cnt;i++)
                {
                    if(a>=ls[i] && a<=rs[i] && i>=pos[a])
                    {
                        pos[a]=i+t;
                        c[a]++;

                        i+=t-1;
                    }
                }

                printf("%d\n",sum(a)-c[a]);
            }
        }
    }
}