[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