hdu-6162 Ch’s gift 主席树

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

树上建主席树，和这题很像spoj count on a tree 只不过改一改查询函数和节点存的值即可

离线把所有点都离散化了。。。。后来发现根本不必要，在线查询，每次二分找一下就好了，因为离散的点多了，线段树自然会大一下，数组开多MLE，开少RE，交题二分刚好卡到28×1e5过了。。。。。

#include<iostream>
#include<stdio.h>
#include<algorithm>
#include<vector>
#include<string.h>
using namespace std;
const int maxn=100010;
int n,m;
int v[maxn],b[3*maxn],t[maxn],tsize,tot=0;
struct charitree
{
    int l,r;
    long long tsize;
}a[maxn*28];
vector<int>w[maxn];

int fa[maxn],dep[maxn],up[maxn][21];
int build(int l,int r)
{
    int root=tot++;
    a[root].tsize=0;
    if(l==r)
        return root;
    int m=(l+r)>>1;
    a[root].l=build(l,m);
    a[root].r=build(m+1,r);
    return root;
}
int update(int root,int x,int k)
{
    int now=tot++;
    int tmp=now;
    a[now].tsize=a[root].tsize+k;
    int l=1,r=tsize;
    while(l<r)
    {
        int m=(l+r)>>1;
        if(x<=m)
        {
            a[now].l=tot++;
            a[now].r=a[root].r;
            now=a[now].l;
            r=m;
            root=a[root].l;
        }
        else
        {
            a[now].l=a[root].l;
            a[now].r=tot++;
            now=a[now].r;
            l=m+1;
            root=a[root].r;
        }
        a[now].tsize=a[root].tsize+k;     
    }
    return tmp;
}

int init()
{
    memset(up,0,sizeof(up));
    for(int i=1;i<=n;i++)
        up[i][0]=fa[i];
    for(int j=1;j<=16;j++)
    {
        for(int i=1;i<=n;i++)
        {
            up[i][j]=up[up[i][j-1]][j-1];
        }
    }
}
int lca(int lx,int rx)
{
    if(dep[lx]<dep[rx])
        swap(lx,rx);
    int a=rx,b=lx;
    int c=dep[b]-dep[a];
    for(int i=0;i<=16;i++)
    {
        if(c&(1<<i))
        b=up[b][i];
    }
    if(a==b)
        return a;
    for(int i=16;i>=0;i--)
    {
        if(up[a][i]!=up[b][i])
        {
            a=up[a][i];
            b=up[b][i];
        }
    }
    return fa[a];
}
long long query(int lx,int rx,int le,int ri,int l,int r)
{
    if(l>=le&&r<=ri)
    {    
        return a[rx].tsize-a[lx].tsize;
    }
    int m=(l+r)>>1;
    long long sum=0;
    if(m>=le)
    sum+=query(a[lx].l,a[rx].l,le,ri,l,m);
    if(m<ri)
    sum+=query(a[lx].r,a[rx].r,le,ri,m+1,r);
    return sum;
}
long long ask(int lx,int rx,int le,int ri)
{
    int l=1,r=tsize;
    int mx=lca(lx,rx);
    int f=fa[mx];
    long long ans=query(t[mx],t[lx],le,ri,1,tsize);
    ans+=query(t[f],t[rx],le,ri,1,tsize);
    return ans;
}
int len=0;
void hash1()
{
    sort(b+1,b+1+len);
    tsize=unique(b+1,b+1+len)-b-1;
}
int find_index(int x)
{
    return lower_bound(b+1,b+1+tsize,x)-b;
}
void dfs(int x,int ff)
{
    for(int i=0;i<w[x].size();i++)
    {
        if(w[x][i]==ff)
            continue;
        dep[w[x][i]]=dep[x]+1;
        fa[w[x][i]]=x;
        t[w[x][i]]=update(t[x],find_index(v[w[x][i]]),v[w[x][i]]);
        dfs(w[x][i],x);
    }
    return ;
}

struct node
{
    int u,uu,l,r;
}ques[100005];
long long ans[maxn];

int main()
{
    int i,j,l,r,k,u,uu;
    while(cin>>n>>m)
    {
        tot=0;
        len=0;
        memset(a,0,sizeof(a));
        memset(up,0,sizeof(up));
        for(i=1;i<=n;i++)
        scanf("%d",&v[i]),b[++len]=v[i];
        for(i=1;i<n;i++)
        {
            scanf("%d %d",&l,&r);
            w[l].push_back(r);
            w[r].push_back(l);
        }
        
        for(i=0;i<m;i++)
        {
            scanf("%d %d %d %d",&ques[i].u,&ques[i].uu,&ques[i].l,&ques[i].r);
            b[++len]=ques[i].l;
            b[++len]=ques[i].r;
        }
        
        hash1();
        t[0]=build(1,tsize);
        int root=1;
        w[0].push_back(root);
        dep[0]=0;
        dfs(0,0);
        init();
        
        for(i=0;i<m;i++)
        ans[i]=ask(ques[i].u,ques[i].uu,find_index(ques[i].l),find_index(ques[i].r));
        
        for(i=0;i<m;i++)
        {
            printf("%lld",ans[i]);
            if(i==m-1)
            printf("\n");
            else
            printf(" ");
        }
        for(i=0;i<=n;i++)
        w[i].clear();    
    }
    return 0;
}