BZOJ2588 - Count on a tree

原题链接

题意简述

给出一棵带点权的n(n≤105)个节点的树，Q(Q≤105)次询问路径(u,v)上第k小的点权值。

分析

记节点到根的点权和为dst，则(u,v)上的点权和为dst[u]+dst[v]−dst[lca(u,v)]−dst[fa[lca(u,v)]]。相似的，对于每个节点，我们用权值线段树记录该节点到根的路径上的点权，这样就可以得到路径(u,v)上的点权了。考虑到每个节点的线段树相当于在其父节点的线段树上进行单点+1，我们可以用可持久化线段树来解决这道题。

实现

与普通的可持久化线段树差别不大，只是在查询的时候要多传几个参数。

代码

//Count on a tree
#include <cstdio>
#include <algorithm>
using namespace std;
inline char gc()
{
    static char now[1<<16],*S,*T;
    if(S==T) {T=(S=now)+fread(now,1,1<<16,stdin); if(S==T) return EOF;}
    return *S++;
}
inline int read()
{
    int x=0,f=1; char ch=gc();
    while(ch<'0'||'9'<ch) {if(ch=='-') f=-1; ch=gc();}
    while('0'<=ch&&ch<='9') x=x*10+ch-'0',ch=gc();
    return x*f;
}
int const N=1e5+10;
int n,m,v[N];
int num,map[N];
struct rec{int v,id;} a0[N];
bool cmpV(rec x,rec y) {return x.v<y.v;}
int cnt,h[N];
struct edge{int v,nxt; edge(int u=0,int v1=0){v=v1,nxt=h[u],h[u]=cnt;}} ed[N<<1];
#define s sg[s0]
int sgCnt,rt[N];
struct seg{int cnt; int L,R;} sg[N*20];
void update(int s0) {s.cnt=sg[s.L].cnt+sg[s.R].cnt;}
void ins(int &s0,int fr,int to,int v)
{
    sg[++sgCnt]=s; s0=sgCnt;
    if(fr==to) {s.cnt++; return;}
    int mid=fr+to>>1;
    if(v<=mid) ins(s.L,fr,mid,v);
    else ins(s.R,mid+1,to,v);
    update(s0);
}
int query(int s1,int s2,int s3,int s4,int fr,int to,int k)
{
    if(fr==to) return fr;
    int cntL=sg[sg[s1].L].cnt+sg[sg[s2].L].cnt-sg[sg[s3].L].cnt-sg[sg[s4].L].cnt;
    int mid=fr+to>>1;
    if(k<=cntL) return query(sg[s1].L,sg[s2].L,sg[s3].L,sg[s4].L,fr,mid,k);
    else return query(sg[s1].R,sg[s2].R,sg[s3].R,sg[s4].R,mid+1,to,k-cntL);
}
int fa[N][20],dpt[N];
void trform(int u)
{
    ins(rt[u]=rt[fa[u][0]],1,num,v[u]);
    for(int i=1;i<=17;i++) fa[u][i]=fa[fa[u][i-1]][i-1];
    for(int i=h[u];i;i=ed[i].nxt)
    {
        int v=ed[i].v;
        if(v!=fa[u][0]) fa[v][0]=u,dpt[v]=dpt[u]+1,trform(v);
    }
}
int lca(int u,int v)
{
    if(dpt[u]<dpt[v]) swap(u,v);
    for(int i=17;i>=0;i--) if(dpt[fa[u][i]]>=dpt[v]) u=fa[u][i];
    for(int i=17;i>=0;i--) if(fa[u][i]!=fa[v][i]) u=fa[u][i],v=fa[v][i];
    return u==v?u:fa[u][0];
}
int main()
{
    n=read(); m=read();
    for(int i=1;i<=n;i++) a0[i].v=read(),a0[i].id=i;
    sort(a0+1,a0+n+1,cmpV);
    num=0;
    for(int i=1;i<=n;i++)
    {
        if(a0[i-1].v!=a0[i].v) map[++num]=a0[i].v;
        v[a0[i].id]=num;
    }
    cnt=0;
    for(int i=1;i<=n-1;i++)
    {
        int u=read(),v=read();
        ed[++cnt]=edge(u,v); ed[++cnt]=edge(v,u);
    }
    rt[0]=sgCnt=0;
    fa[1][0]=0,dpt[1]=1,trform(1);
    int ans=0;
    for(int owo=1;owo<=m;owo++)
    {
        int u=read()^ans,v=read(),k=read(); int w=lca(u,v);
        printf("%d",ans=map[query(rt[u],rt[v],rt[w],rt[fa[w][0]],1,num,k)]);
        if(owo!=m) printf("\n");
    }
    return 0;
}

注意

本题强制在线，查询时仅需要将u异或上lastans即可。非强制在线的原题来自SPOJ.com - Problem COT。
最后一次查询不要输出\n，否则会PE。