2588: Spoj 10628. Count on a tree 主席树+LCA

最新推荐文章于 2019-04-05 19:52:41 发布

ws_fqk

最新推荐文章于 2019-04-05 19:52:41 发布

阅读量467

点赞数

分类专栏： My Code

本文链接：https://blog.csdn.net/Phenix_2015/article/details/50515913

版权

My Code 专栏收录该内容

349 篇文章 0 订阅

订阅专栏

在树上按照dfs序建主席树。转化为线性的问题。。
ans(x,y)=sum[x]+sum[y]-sum[lca(x,y)]-sum[fa[lca(x,y)][0]]，画个图便知道QAQ。。无修改良心。

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
int n,m,lastans,ind,cnt,tot,size;
int v[100005],root[100005],num[100005],dfn[100005],deep[100005],hash[100005],pos[100005],head[100005];
int next[200005],list[200005];
int fa[100005][17];
int l[2000005],r[2000005],sum[2000005];
inline int read()
{
    int a=0,f=1; char c=getchar();
    while (c<'0'||c>'9') {if (c=='-') f=-1; c=getchar();}
    while (c>='0'&&c<='9') {a=a*10+c-'0'; c=getchar();}
    return a*f;
}
inline void insert(int x,int y)
{
    next[++cnt]=head[x];
    head[x]=cnt;
    list[cnt]=y;
}
inline int find(int x)
{
    int l=1,r=tot;
    while (l<=r)
    {
        int mid=l+r>>1;
        if (hash[mid]<x) l=mid+1; 
        else if (hash[mid]==x) return mid;
        else r=mid-1;
    }
}
void dfs(int x)
{
    dfn[++ind]=x; pos[x]=ind;
    for (int i=1;(1<<i)<=deep[x];i++)
        fa[x][i]=fa[fa[x][i-1]][i-1];
    for (int i=head[x];i;i=next[i])
        if (list[i]!=fa[x][0])
        {
            fa[list[i]][0]=x;
            deep[list[i]]=deep[x]+1;
            dfs(list[i]);
        }
}
inline int lca(int x,int y)
{
    if (deep[x]<deep[y]) swap(x,y);
    int t=deep[x]-deep[y];
    for (int i=0;(1<<i)<=t;i++)
        if (t&(1<<i)) x=fa[x][i];
    for (int i=16;~i;i--) 
        if (fa[x][i]!=fa[y][i]) x=fa[x][i],y=fa[y][i];
    return x==y?x:fa[x][0];
}
void update(int ll,int rr,int x,int &y,int v)
{
    y=++size;
    sum[y]=sum[x]+1;
    if (ll==rr) return;
    l[y]=l[x]; r[y]=r[x];
    int mid=ll+rr>>1;
    if (v<=mid) update(ll,mid,l[x],l[y],v);
    else update(mid+1,rr,r[x],r[y],v);
}
inline int query(int x,int y,int k)
{
    int a=x,b=y,c=lca(x,y),d=fa[c][0];
    a=root[pos[a]]; b=root[pos[b]]; c=root[pos[c]]; d=root[pos[d]];
    int ll=1,rr=tot;
    while (ll!=rr)
    {
        int mid=ll+rr>>1;
        int t=sum[l[a]]+sum[l[b]]-sum[l[c]]-sum[l[d]];
        if (t>=k) rr=mid,a=l[a],b=l[b],c=l[c],d=l[d];
        else k-=t,ll=mid+1,a=r[a],b=r[b],c=r[c],d=r[d];
    }
    return hash[ll];
}
int main()
{
    n=read(); m=read();
    for (int i=1;i<=n;i++)
        v[i]=num[i]=read();
    sort(num+1,num+n+1);
    hash[++tot]=num[1];
    for (int i=2;i<=n;i++)
        if (num[i]!=num[i-1]) hash[++tot]=num[i];
    for (int i=1;i<=n;i++) v[i]=find(v[i]);
    for (int i=1;i<n;i++)
    {
        int u=read(),v=read();
        insert(u,v); insert(v,u);
    }
    dfs(1);
    for (int i=1;i<=n;i++)
    {
        int t=dfn[i];
        update(1,tot,root[pos[fa[t][0]]],root[i],v[t]);
    }
    for (int i=1;i<=m;i++)
    {
        int x=read(),y=read(),k=read();
        x^=lastans;
        lastans=query(x,y,k);
        printf("%d",lastans);
        if (i!=m) puts("");
    }
    return 0;
}