HYSBZ-2588- Count on a tree （树上主席树）

Spoj 10628. Count on a tree

Time Limit: 12 Sec   Memory Limit:128 MB
Submit: 7148   Solved: 1739
[ Submit][ Status][ Discuss]

Description

给定一棵N个节点的树，每个点有一个权值，对于M个询问(u,v,k)，你需要回答u xor lastans和v这两个节点间第K小的点权。其中lastans是上一个询问的答案，初始为0，即第一个询问的u是明文。

Input

第一行两个整数N,M。

第二行有N个整数，其中第i个整数表示点i的权值。

后面N-1行每行两个整数(x,y)，表示点x到点y有一条边。

最后M行每行两个整数(u,v,k)，表示一组询问。

Output

M行，表示每个询问的答案。最后一个询问不输出换行符

Sample Input

8 5
105 2 9 3 8 5 7 7
1 2
1 3
1 4
3 5
3 6
3 7
4 8
2 5 1
0 5 2
10 5 3
11 5 4
110 8 2

Sample Output

2
8
9
105
7

HINT

HINT：

N,M<=100000

暴力自重。。。

中文题面就不讲意思了；

思路：这题就是找树上一条链上的第k小值，所以把每个点到根节点这些链建主席树，然后会发现u，v两个节点之间的链，在最后利用主席书计算的时候，主席树上记录的值等于u->root(根)这条链加上v->root减去lca[u,v]->root减去fa[lca[u,v]]->root,画个图想一下就懂了，这题就是在树上建主席树 ，算个入门题把；主席树刚学几天，写的时候写了个错误wa了好多发~~~这题求lca我是用树链剖分求的，倍增法还不会，并且感觉树链剖分求lca很好理解；

AC代码：

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<vector>
#include<cmath>
using namespace std;
const int maxn=1e5+10;
int n,m,arr[maxn];


vector<int> v;
int root[maxn],c;
struct ndoe{int l,r,cnt;}tree[maxn*40];
int getid(int x){return lower_bound(v.begin(),v.end(),x)-v.begin()+1;}
void update(int l,int r,int &x,int y,int pos)
{
    tree[++c]=tree[y];tree[c].cnt++;x=c;
    if(l==r) return;
    int mid=(l+r)>>1;
    if(pos<=mid) update(l,mid,tree[x].l,tree[y].l,pos);
    else update(mid+1,r,tree[x].r,tree[y].r,pos);
}
int query(int l,int r,int x,int y,int lca,int flca,int k)
{
    if(l==r) return v[l-1];
    int tmp=tree[tree[x].l].cnt+tree[tree[y].l].cnt-tree[tree[lca].l].cnt-tree[tree[flca].l].cnt;
    int mid=(l+r)>>1;
    if(tmp>=k) return query(l,mid,tree[x].l,tree[y].l,tree[lca].l,tree[flca].l,k);
    else return query(mid+1,r,tree[x].r,tree[y].r,tree[lca].r,tree[flca].r,k-tmp);
}

int cnt,num,fa[maxn],depth[maxn],head[maxn],son[maxn],pos[maxn],sz[maxn],top[maxn];
struct edge{int to,next;}E[maxn*2];
void add_edge(int u,int v){E[++cnt].to=v;E[cnt].next=head[u];head[u]=cnt;}
void dfs1(int u,int f,int dep)
{
    fa[u]=f;depth[u]=dep;
    son[u]=0;sz[u]=1;
    update(1,n,root[u],root[f],getid(arr[u]));
    for(int i=head[u];i!=-1;i=E[i].next)
    {
        int ff=E[i].to;
        if(ff==f) continue;
        dfs1(ff,u,dep+1);
        sz[u]+=sz[ff];
        if(sz[son[u]]<sz[ff]) son[u]=ff;
    }
}
void dfs2(int u,int tp)
{
    pos[u]=++num;
    top[u]=tp;
    if(son[u]) dfs2(son[u],tp);
    for(int i=head[u];i!=-1;i=E[i].next)
    {
        int ff=E[i].to;
        if(ff==fa[u]||ff==son[u]) continue;
        dfs2(ff,ff);
    }
}
int lca(int u,int v)
{
    while(top[u]!=top[v]){
        if(depth[top[u]]<depth[top[v]]) v=fa[top[v]];
        else u=fa[top[u]];
    }
    return depth[u]<depth[v]?u:v;
}


void init()
{
    c=cnt=num=0;
    for(int i=1;i<=maxn;i++) head[i]=-1;
}
int main()
{
    scanf("%d%d",&n,&m);
    init();
    for(int i=1;i<=n;i++) scanf("%d",&arr[i]),v.push_back(arr[i]);
    sort(v.begin(),v.end()),v.erase(unique(v.begin(),v.end()),v.end());
    int u,v;
    for(int i=1;i<n;i++){
        scanf("%d%d",&u,&v);
        add_edge(u,v);
        add_edge(v,u);
    }
    dfs1(1,0,1);
    dfs2(1,1);
    int ans;
    while(m--){
        int l,r,k;
        scanf("%d%d%d",&l,&r,&k);
        int p=lca(l,r);
        ans=query(1,n,root[l],root[r],root[p],root[fa[p]],k);
        printf("%d\n",ans);
    }
    return 0;
}