UVa 1513 / UVALive 5902 Movie collection (树状数组)

//数状数组模板题

AC代码：

#include<stdio.h>
#include<string.h>
#define N 200005
int a[N];
int b[N];
int n,k;
int lower_bit(int i)
{
    return i&(-i);
}
void add(int i,int v)
{
    while(i<=n+k)
    {
        a[i]+=v;
        i+=lower_bit(i);
    }
}
int sum(int i)
{
    int s=0;
    while(i>0)
    {
        s+=a[i];
        i-=lower_bit(i);
    }
    return s;
}
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        memset(a,0,sizeof(a));
        scanf("%d%d",&n,&k);
        int i;
        for(i=1;i<=n;i++)
        {
            b[i]=n+1-i;
            add(i,1);
        }
        int x;
        int h=n;
        for(i=1;i<=k;i++)
        {
            scanf("%d",&x);
            int q=sum(h)-sum(b[x]);
            if(i==k)
            {
                printf("%d\n",q);
            }
            else
            {
                printf("%d ",q);
            }
            add(b[x],-1);
            b[x]=++h;
            add(b[x],1);
        }
    }
    return 0;
}