[BZOJ2588]Count on a tree（可持久化权值线段树|主席树）

题目描述

传送门

题解

在树上建维护当前节点到根的路径的权值线段树，然后查询的时候为sum[a]+sum[b]-sum[lca(a,b)]-sum[father[lca(a,b)]]。

代码

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

const int max_n=1e5+5;
const int sz=17;
const int max_e=max_n*2;
const int max_tree=5*max_n*sz;

int n,m,x,y,u,t,k,old,fold,ans,size;
int mi[sz+5],val[max_n],p[max_n],num[max_n],hash[max_n];
int h[max_n],father[max_n],root[max_n],f[max_n][sz+5];
int sum[max_tree],ls[max_tree],rs[max_tree];
int tot,point[max_n],next[max_e],v[max_e];

inline int cmp(int a,int b) {return val[a]<val[b];}
inline void addedge(int x,int y){++tot;next[tot]=point[x];point[x]=tot;v[tot]=y;}
inline void dfs(int x,int fa,int dep){
    h[x]=dep; father[x]=fa;
    for (int i=1;i<sz;++i){
        if (h[x]-mi[i]<1) break;
        f[x][i]=f[f[x][i-1]][i-1];
    }
    for (int i=point[x];i;i=next[i])
      if (v[i]!=fa){
        f[v[i]][0]=x;
        dfs(v[i],x,dep+1);
      }
}
inline int lca(int x,int y){
    if (h[x]<h[y]) swap(x,y); int k=h[x]-h[y];
    for (int i=0;i<sz;++i)
      if ((k>>i)&1) x=f[x][i];
    if (x==y) return x;
    for (int i=sz-1;i>=0;--i)
      if (f[x][i]!=f[y][i])
        x=f[x][i],y=f[y][i];
    return f[x][0];
}
inline void update(int &now,int l,int r,int x,int v){
    int mid=(l+r)>>1;
    sum[++size]=sum[now]+v; ls[size]=ls[now]; rs[size]=rs[now]; now=size;
    if (l==r) return;
    if (x<=mid) update(ls[now],l,mid,x,v);
    else update(rs[now],mid+1,r,x,v);
}
inline void build(int x,int fa){
    for (int i=point[x];i;i=next[i])
      if (v[i]!=fa){
        root[v[i]]=root[x];
        update(root[v[i]],1,n,num[v[i]],1);
        build(v[i],x);
      }
}
inline int query(int a,int b,int c,int d,int l,int r,int k){
    int mid=(l+r)>>1;
    if (l==r) return l;
    int t=sum[ls[a]]+sum[ls[b]]-sum[ls[c]]-sum[ls[d]];
    if (t>=k) return query(ls[a],ls[b],ls[c],ls[d],l,mid,k);
    else return query(rs[a],rs[b],rs[c],rs[d],mid+1,r,k-t);
}
int main(){
    mi[0]=1; for (int i=1;i<sz;++i) mi[i]=mi[i-1]*2;
    scanf("%d%d",&n,&m);
    for (int i=1;i<=n;++i) scanf("%d",&val[i]),p[i]=i;
    sort(p+1,p+n+1,cmp);
    for (int i=1;i<=n;++i)
      num[p[i]]=i,hash[i]=val[p[i]];
    for (int i=1;i<n;++i)
        scanf("%d%d",&x,&y),addedge(x,y),addedge(y,x);
    dfs(1,0,1);
    update(root[1],1,n,num[1],1);
    build(1,0);
    for (int i=1;i<=m;++i){
        scanf("%d%d%d",&u,&t,&k);
        u^=ans;
        old=lca(u,t); fold=father[old];
        ans=hash[query(root[u],root[t],root[old],root[fold],1,n,k)];
        printf("%d",ans);
        if (i!=m) printf("\n");
    }
}

总结

这两天光写Tarjan的单向边竟然忘了双向加边了！！！MDZZ
RE应该都是因为答案错了一异或就鬼畜了T_T