BZOJ 2588 Count on a tree 主席樹

題意:
給定一棵樹,多次詢問x到y的路徑上的第k小的點權。
強制在線。
解析:
COT1
顯然我們可以開權值線段樹來維護這個事情,但是又因爲涉及到多版本線段樹,所以容易考慮到可持久化。
對於每一個點來說,在這個點的權值線段樹即爲他的父節點的線段樹版本加上該點的點權的更新。
因爲不涉及到修改,所以我們直接搞一個主席樹即可。
然後就是很普通的詢問拆分過程了。
即u這個點的主席樹+v這個點的主席樹-lca(u,v)這個點的主席樹-fa[lca(u,v)]這個點的主席樹後產生的主席樹上查第k小即可。

最後一個詢問後面不輸出換行,否則PE!

代碼:

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define N 100100
using namespace std;
typedef long long ll;
int n,m;
ll val[N],a[N];
int head[N],cnt;
int fa[N][21],deep[N];
int root[N];
struct node
{
    int from,to,next;
}edge[N<<1];
struct Seg
{
    int l,r,siz;
}seg[N*30];
void init()
{
    deep[1]=1;
    memset(head,-1,sizeof(head));
    cnt=1;
}
void edgeadd(int from,int to)
{
    edge[cnt].from=from,edge[cnt].to=to;
    edge[cnt].next=head[from],head[from]=cnt++;
}
void dfs(int now,int f)
{
    for(int i=head[now];i!=-1;i=edge[i].next)
    {
        int to=edge[i].to;
        if(to==f)continue;
        fa[to][0]=now,deep[to]=deep[now]+1;
        dfs(to,now);
    }
}
int lca(int x,int y)
{
    if(deep[x]<deep[y])swap(x,y);
    for(int i=20;i>=0;i--)
        if(deep[fa[x][i]]>=deep[y])
            x=fa[x][i];
    if(x==y)return x;
    for(int i=20;i>=0;i--)
        if(deep[fa[x][i]]==deep[fa[y][i]]&&fa[x][i]!=fa[y][i])
            x=fa[x][i],y=fa[y][i];
    return fa[x][0];
}
int size;
void pushup(int rt)
{
    seg[rt].siz=seg[seg[rt].l].siz+seg[seg[rt].r].siz;
}
void insert(int y,int &x,int l,int r,int v)
{
    x=++size;
    seg[x]=seg[y];
    seg[x].siz++;
    if(l==r)return ;
    int mid=(l+r)>>1;
    if(v<=mid)
        insert(seg[y].l,seg[x].l,l,mid,v);
    else insert(seg[y].r,seg[x].r,mid+1,r,v);
    pushup(x);
}
void dfs2(int now)
{
    insert(root[fa[now][0]],root[now],1,n,val[now]);
    for(int i=head[now];i!=-1;i=edge[i].next)
    {
        int to=edge[i].to;
        if(to==fa[now][0])continue;
        dfs2(to);
    }
}
int query(int a,int b,int c,int d,int l,int r,int k)
{
    if(l==r)return l;
    int mid=(l+r)>>1;
    int Size=seg[seg[a].l].siz+seg[seg[b].l].siz-seg[seg[c].l].siz-seg[seg[d].l].siz;
    if(Size>=k)
        return query(seg[a].l,seg[b].l,seg[c].l,seg[d].l,l,mid,k);
    else return query(seg[a].r,seg[b].r,seg[c].r,seg[d].r,mid+1,r,k-Size);
}
int main()
{
    init();
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
        scanf("%lld",&val[i]),a[i]=val[i];
    sort(a+1,a+n+1);
    int tot=unique(a+1,a+n+1)-a-1;
    for(int i=1;i<=n;i++)
        val[i]=lower_bound(a+1,a+tot+1,val[i])-a;
    for(int i=1;i<n;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        edgeadd(x,y),edgeadd(y,x);
    }
    dfs(1,0);
    for(int i=1;i<=20;i++)
        for(int j=1;j<=n;j++)
            fa[j][i]=fa[fa[j][i-1]][i-1];
    int ans=0;
    dfs2(1);
    for(int i=1;i<=m;i++)
    {
        int u,v,k;
        scanf("%d%d%d",&u,&v,&k);
        u^=ans;
        int ff=lca(u,v);
        int no=query(root[u],root[v],root[ff],root[fa[ff][0]],1,n,k);
        printf("%lld",a[no]);
        ans=a[no];
        if(i!=m)puts("");
    }
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章