HDU 3333 Turing Tree

http://acm.hdu.edu.cn/showproblem.php?pid=3333

树状数组+离散化+离线算法

读入所有区间，按右端点排序，每加入一个数时，如果已经加过，就在该数原来位置减去它，在新位置加上它

#include <cstdio>
#include <algorithm>
using namespace std;

typedef __int64 LL;
const int N=30010,Q=100010;
int a[N],n,vis[N];
int dz[N],nd;
LL c[N];
struct extent
{
    int u,v,p;
    LL ans;
}e[Q];
bool cmp1(const extent &a,const extent &b)
{
    return a.v<b.v;
}
bool cmp2(const extent &a,const extent &b)
{
    return a.p<b.p;
}
int bfind(int x,int *a,int l,int r)
{
    while(l<r)
    {
        int m=(l+r)/2;
        if(x==a[m]) return m;
        else if(x<a[m]) r=m;
        else l=m+1;
    }
    return l;
}
int lowbit(int x)
{
    return x&(-x);
}
void add(int x,int p,int maxn)
{
    for(int i=p;i<=maxn;i+=lowbit(i))
     c[i]+=x;
}
LL sum(int p)
{
    LL s=0;
    for(int i=p;i;i-=lowbit(i)) s+=c[i];
    return s;
}
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        int n;
        scanf("%d",&n);
        for(int i=1;i<=n;i++) scanf("%d",&a[i]);
        for(int i=0;i<n;i++) dz[i]=a[i+1];
        sort(dz,dz+n);
        nd=1;
        for(int i=1;i<n;i++)
            if(dz[i]!=dz[i-1]) dz[nd++]=dz[i];
        int q;
        scanf("%d",&q);
        for(int i=0;i<q;i++)
        {
            scanf("%d%d",&e[i].u,&e[i].v);
            e[i].p=i;
        }
        sort(e,e+q,cmp1);
        memset(vis,0,sizeof(vis));
        memset(c,0,sizeof(c));
        int p=0;
        for(int i=1;i<=n;i++)
        {
            add(a[i],i,n);
            int x=bfind(a[i],dz,0,nd-1);
            if(vis[x]) add(-a[i],vis[x],n);
            vis[x]=i;
            while(p<q && e[p].v==i)
            {
                e[p].ans=sum(e[p].v)-sum(e[p].u-1);
                p++;
            }
        }
        sort(e,e+q,cmp2);
        for(int i=0;i<q;i++) printf("%I64d\n",e[i].ans);
    }
    return 0;
}