主席树+dfs SPOJ BZOJ2588 Count on a tree

这道题我由于智障错误导致一直错。

在树上建主席树，加上lca思想，很简单。

 #include<bits/stdc++.h>
 using namespace std;
 const int N=;
 struct node
 {
     int to,nex;
 }e[N<<];
 struct tree
 {
     int l,r,s;
 }t[];
 int n,m,cnt,cn,mx,pre,head[N],f[N][],a[N],rt[N],d[N];
 vector<int>v;
 inline int get(int x){return lower_bound(v.begin(),v.end(),x)-v.begin()+;}
 void add(int x,int y)
 {
     e[++cnt].to=y;e[cnt].nex=head[x];head[x]=cnt;
 }
 void update(int &x,int l,int r,int p,int num)
 {
     x=++cn;t[x]=t[p];t[x].s++;int mid=l+r>>;
     if(l==r)return ;
     if(mid>=num)update(t[x].l,l,mid,t[p].l,num);
     else update(t[x].r,mid+,r,t[p].r,num);
 }
 void dfs(int x,int fa)
 {
     update(rt[x],,mx,rt[fa],get(a[x]));f[x][]=fa;d[x]=d[fa]+;
     for(int i=;i<=;++i)
     if(d[x]>=(<<i))
     f[x][i]=f[f[x][i-]][i-];
     for(int i=head[x];i;i=e[i].nex)
     {
         int y=e[i].to;
         if(y==fa)continue;
         dfs(y,x);
     }
 }
 int lca(int x,int y)
 {
     if(d[x]<d[y])swap(x,y);
     int tmp=d[x]-d[y];
     for(int i=;i>=;i--)
     if(tmp&(<<i))x=f[x][i];
     for(int i=;i>=;i--)
     {
         if(f[x][i]!=f[y][i])
             x=f[x][i],y=f[y][i];
     }
     return x==y?x:f[x][];
 }
 int query(int l,int r,int rt1,int rt2,int rt3,int rt4,int k)
 {
     if(l==r){
         pre=v[l-];return l;
     }
     int sum=t[t[rt1].l].s+t[t[rt2].l].s-t[t[rt3].l].s-t[t[rt4].l].s;
     int mid=(l+r)>>;
     if(sum>=k)return query(l,mid,t[rt1].l,t[rt2].l,t[rt3].l,t[rt4].l,k);
     else return query(mid+,r,t[rt1].r,t[rt2].r,t[rt3].r,t[rt4].r,k-sum);
 }
 int main()
 {
     //freopen("rand.out","r",stdin);
     //freopen("my.out","w",stdout);
     int x,y,k;
     scanf("%d%d",&n,&m);
     for(int i=;i<=n;++i)
     {
         scanf("%d",&a[i]);v.push_back(a[i]);
     }
     sort(v.begin(),v.end());v.erase(unique(v.begin(),v.end()),v.end());mx=v.size();
     for(int i=;i<n;++i)
     {
         scanf("%d%d",&x,&y);
         add(x,y);add(y,x);
     }
     dfs(,);
     for(int i=;i<=m;++i)
     {
         scanf("%d%d%d",&x,&y,&k);
         int z=lca(pre^x,y);
         printf("%d",v[query(,mx,rt[pre^x],rt[y],rt[z],rt[f[z][]],k)-]);
         if(i<m) puts("");
     }
     return ;
 }

BZOJ最后换行会PE。。。。