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;
}