bzoj 1146 [CTSC2008]网络管理Network

    很久之前写过 count on the tree。

    然后一直不懂树状数组是怎么套上这个主席树的。

    看了两小时发现它套的就是个权值线段树，

    看不出来可持久化在哪里。    

    因为动态开点所以空间nlog2n。

    树状数组维护dfs序，每个节点挂个线段树。

    为了省空间拿原树建了个主席树。

    

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#define N 80005
using namespace std;
int n,q;
int t[N];
int head[N],ver[N*2],nxt[N*2],tot;
int li[N*2],now[N*2],cnt;
void add(int a,int b)
{
    tot++;nxt[tot]=head[a];head[a]=tot;ver[tot]=b;return ;
}
struct node 
{
    int k,a,b;
}qq[N];
int zz,st[N],ed[N];
int dep[N];
int fa[N][18];
void lca()
{
    for(int i=1;i<=17;i++)
    for(int j=1;j<=n;j++)
    fa[j][i]=fa[fa[j][i-1]][i-1];
}
int lca(int x,int y)
{
    if(dep[x]<dep[y])swap(x,y);
    for(int i=17;i>=0;i--)if(dep[fa[x][i]]>=dep[y])x=fa[x][i];
    if(x==y)return x;
    for(int i=17;i>=0;i--)
    {
        if(fa[x][i]!=fa[y][i])
        {
            x=fa[x][i];y=fa[y][i];
        }
    }
    return fa[x][0];
}
void dfs(int x,int f)
{
    st[x]=++zz;
    for(int i=head[x];i;i=nxt[i])
    {
        if(ver[i]==f)continue;
        fa[ver[i]][0]=x;
        dep[ver[i]]=dep[x]+1;
        dfs(ver[i],x);
    }
    ed[x]=zz;
}
int root[N],rot[N];
struct Node
{
    int l,r,sum;
}a[N*140];
int num;

void inrt(int x,int y,int l,int r,int pos)
{
    if(l==r)
    {
        a[x].sum=a[y].sum+1;
        return ;
    }
    int mid=(l+r)>>1;
    if(pos<=mid)
    {
        if(!a[x].l)a[x].l=++num;
        a[x].r=a[y].r;
        inrt(a[x].l,a[y].l,l,mid,pos);
    }
    else
    {
        if(!a[x].r)a[x].r=++num;
        a[x].l=a[y].l;
        inrt(a[x].r,a[y].r,mid+1,r,pos);
    }
    a[x].sum=a[a[x].l].sum+a[a[x].r].sum;
}
void build()
{
    for(int i=1;i<=n;i++)
    {
        if(!rot[i])rot[i]=++num;
        inrt(rot[i],rot[fa[i][0]],1,cnt,now[i]);
    }
}
void insert(int x,int l,int r,int pos,int z)
{
    if(l==r)
    {
        a[x].sum+=z;return ;
    }
    int mid=(l+r)>>1;
    if(pos<=mid)
    {
        if(!a[x].l)a[x].l=++num;
        insert(a[x].l,l,mid,pos,z);
    }
    else
    {
        if(!a[x].r)a[x].r=++num;
        insert(a[x].r,mid+1,r,pos,z);
    }
    a[x].sum=a[a[x].l].sum+a[a[x].r].sum;
}
void ad(int x,int z,int la)
{
    for(int i=x;i<=n;i+=(i&(-i)))
    {
        if(!root[i])root[i]=++num;
        insert(root[i],1,cnt,z,la);
    }
    return ;
}
int no[N][4];
int t1,t2,t3,t4;
int qur(int x,int y)
{
    int as=0;
    for(int i=x;i;i-=(i&(-i)))
    {
        as+=a[a[no[i][y]].l].sum;
    }
    return as;
}
int qur(int l,int r,int kk,int r1,int r2,int r3,int r4)
{
    if(l==r)
    {
        return l;
    }
    int mid=(l+r)>>1;
    int xx=a[a[r3].l].sum+a[a[r4].l].sum-a[a[r1].l].sum-a[a[r2].l].sum;
    xx+=qur(st[t1],0);xx+=qur(st[t2],1);
    xx-=qur(st[t3],2);xx-=qur(st[t4],3);
    if(xx>=kk)
    {
        for(int i=st[t1];i;i-=(i&(-i)))no[i][0]=a[no[i][0]].l;
        for(int i=st[t2];i;i-=(i&(-i)))no[i][1]=a[no[i][1]].l;
        for(int i=st[t3];i;i-=(i&(-i)))no[i][2]=a[no[i][2]].l;
        for(int i=st[t4];i;i-=(i&(-i)))no[i][3]=a[no[i][3]].l;
        return qur(l,mid,kk,a[r1].l,a[r2].l,a[r3].l,a[r4].l);
    }
    else
    {
        for(int i=st[t1];i;i-=(i&(-i)))no[i][0]=a[no[i][0]].r;
        for(int i=st[t2];i;i-=(i&(-i)))no[i][1]=a[no[i][1]].r;
        for(int i=st[t3];i;i-=(i&(-i)))no[i][2]=a[no[i][2]].r;
        for(int i=st[t4];i;i-=(i&(-i)))no[i][3]=a[no[i][3]].r;
        return qur(mid+1,r,kk-xx,a[r1].r,a[r2].r,a[r3].r,a[r4].r);
    }
}
int main()
{
    scanf("%d%d",&n,&q);
    for(int i=1;i<=n;i++)scanf("%d",&t[i]),li[++cnt]=t[i];
    for(int i=1;i<n;i++)
    {
        scanf("%d%d",&t1,&t2);
        add(t1,t2);add(t2,t1);
    }
    for(int i=1;i<=q;i++)
    {
        scanf("%d%d%d",&qq[i].k,&qq[i].a,&qq[i].b);
        if(!qq[i].k)li[++cnt]=qq[i].b;
    }
    sort(li+1,li+cnt+1);
    cnt=unique(li+1,li+cnt+1)-li-1;
    for(int i=1;i<=n;i++)now[i]=lower_bound(li+1,li+cnt+1,t[i])-li;
    dep[1]=1;dfs(1,-1);
    lca();
    build();
    for(int i=1;i<=q;i++)
    {
        if(!qq[i].k)
        {
            ad(st[qq[i].a],now[qq[i].a],-1);
            if(ed[qq[i].a]!=zz)ad(ed[qq[i].a]+1,now[qq[i].a],1);
            now[qq[i].a]=lower_bound(li+1,li+cnt+1,qq[i].b)-li;
            ad(st[qq[i].a],now[qq[i].a],1);
            if(ed[qq[i].a]!=zz)ad(ed[qq[i].a]+1,now[qq[i].a],-1);
        }
        else
        {
            int tmp=lca(qq[i].a,qq[i].b);
            int yy=dep[qq[i].a]+dep[qq[i].b]-dep[tmp]*2+1;
            if(yy<qq[i].k)
            {
                puts("invalid request!");
                continue;
            }
            else
            {
                yy=yy-qq[i].k+1;
                for(int j=st[qq[i].a];j;j-=(j&(-j)))no[j][0]=root[j];
                for(int j=st[qq[i].b];j;j-=(j&(-j)))no[j][1]=root[j];
                for(int j=st[tmp];j;j-=(j&(-j)))no[j][2]=root[j];
                for(int j=st[fa[tmp][0]];j;j-=(j&(-j)))no[j][3]=root[j];
                t1=qq[i].a;t2=qq[i].b;t3=tmp;t4=fa[tmp][0];
                int tp=qur(1,cnt,yy,rot[tmp],rot[fa[tmp][0]],rot[qq[i].a],rot[qq[i].b]);
                printf("%d\n",li[tp]);
            }
        }
    }
    return 0;
}