bzoj1146整体二分+树链剖分+树状数组

其实也没啥好说的

用树状数组可以O(logn)的查询

套一层整体二分就可以做到O(nlngn)

最后用树链剖分让序列上树

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
inline int read()
{
    int x=0,f=1,ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-'){f=-1;}ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
int n,q;
int data[80005];
inline void add(int x,int i)
{
    while(x<=n)
    {
        data[x]+=i;
        x+=x&(-x);
    }
}
inline int query(int x)
{
    int res=0;
    while(x>=1)
    {
        res+=data[x];
        x-=x&(-x);
    }
    return res;
}
int h[80005],next[160005],to[160005],cnt;
inline void addedge(int x,int y)
{
    next[cnt]=h[x];
    to[cnt]=y;
    h[x]=cnt;
    cnt++;
}
int size[80005],hson[80005],dep[80005],f[80005];
inline void bfs(int x,int fa)
{
    int i;
    size[x]=1;
    for(i=h[x];i!=-1;i=next[i])
    {
        if(to[i]==fa)
        {
            continue;
        }
        dep[to[i]]=dep[x]+1;
        f[to[i]]=x;
        bfs(to[i],x);
        size[x]+=size[to[i]];
        if(size[to[i]]>size[hson[x]])
        {
            hson[x]=to[i];
        }
    }
}
int up[80005],first[80005],tim;
inline void spfa(int x,int tt)
{
    tim++;
    first[x]=tim;
    up[x]=tt;
    if(!hson[x])
    {
        return ;
    }
    spfa(hson[x],tt);
    int i;
    for(i=h[x];i!=-1;i=next[i])
    {
        if(to[i]==f[x]||to[i]==hson[x])
        {
            continue;
        }
        spfa(to[i],to[i]);
    }
}
inline int lca(int x,int y)
{
    int res=0;
    while(up[x]!=up[y])
    {
        if(dep[up[x]]<dep[up[y]])
        {
            swap(x,y);
        }
        res+=query(first[x])-query(first[up[x]]-1);
        x=f[up[x]];
    }
    if(dep[x]>dep[y])
    {
        swap(x,y);
    }
    res+=query(first[y])-query(first[x]-1);
    return res;
}
struct stu
{
    int op,l,r,id,x;
};
stu a[240005];
int ans[80005];
int m[240005];
stu q1[240005],q2[240005];
inline void erfen(int l,int r,int sl,int sr)
{
    if(sl>sr)
    {
        return ;
    }
    if(l==r)
    {
        int i;
        for(i=sl;i<=sr;i++)
        {
            if(a[i].op!=2)
            {
                continue;
            }
            ans[a[i].id]=l;
        }
        return ;
    }
    int i,mid=(l+r)>>1,k,cnt1=0,cnt2=0;
    for(i=sl;i<=sr;i++)
    {
        if(a[i].op==0)
        {
            if(a[i].x<=mid)
            {
                add(first[a[i].l],1);
                m[i]=0;
            }
            else
            {
                m[i]=1;
            }
            continue;
        }
        if(a[i].op==1)
        {
            if(a[i].x<=mid)
            {
                add(first[a[i].l],-1);
                m[i]=0;
            }
            else
            {
                m[i]=1;
            }
            continue;
        }
        if(a[i].op==2)
        {
            k=lca(a[i].l,a[i].r);
            if(k<a[i].x)
            {
                a[i].x-=k;
                m[i]=1;
            }
            else
            {
                m[i]=0;
            }
        }
    }
    for(i=sl;i<=sr;i++)
    {
        if(a[i].op==0)
        {
            if(a[i].x<=mid)
            {
                add(first[a[i].l],-1);
            }
        }
        if(a[i].op==1)
        {
            if(a[i].x<=mid)
            {
                add(first[a[i].l],1);
            }
        }
        if(m[i]==0)
        {
            cnt1++;
            q1[cnt1]=a[i];
        }
        else
        {
            cnt2++;
            q2[cnt2]=a[i];
        }
    }
    for(i=1;i<=cnt1;i++)
    {
        a[sl+i-1]=q1[i];
    }
    for(i=1;i<=cnt2;i++)
    {
        a[sl+cnt1+i-1]=q2[i];
    }
    erfen(l,mid,sl,sl+cnt1-1);
    erfen(mid+1,r,sl+cnt1,sr);
}
int p[80005];
int main()
{
    memset(h,-1,sizeof(h));
    memset(ans,0x3f,sizeof(ans));
    n=read(),q=read();
    int i,x,y,tail=0,k,o;
    for(i=1;i<=n;i++)
    {
        x=read();
        tail++;
        a[tail].op=0,a[tail].l=i,a[tail].x=x;
        p[i]=x;
    }
    for(i=1;i<n;i++)
    {
        x=read(),y=read();
        addedge(x,y);
        addedge(y,x);
    }
    dep[1]=1;
    bfs(1,-1);
    spfa(1,1);
    for(i=1;i<=n;i++)
    {
        add(first[i],1);
    }
    for(i=1;i<=q;i++)
    {
        k=read(),x=read(),y=read();
        if(k==0)
        {
            tail++;
            a[tail].op=1;a[tail].l=x,a[tail].x=p[x];
            tail++;
            a[tail].op=0;a[tail].l=x;a[tail].x=y;
            p[x]=y;
        }
        else
        {
            o=lca(x,y);
            if(o<k)
            {
                ans[i]=-1;
                continue;
            }
            k=o-k+1;
            tail++;
            a[tail].op=2,a[tail].l=x,a[tail].r=y,a[tail].x=k;
            a[tail].id=i;
        }
    }
    memset(data,0,sizeof(data));
    erfen(0,100000000,1,tail);
    for(i=1;i<=q;i++)
    {
        if(ans[i]==0x3f3f3f3f)
        {
            continue;
        }
        if(ans[i]==-1)
        {
            puts("invalid request!");
        }
        else
        {
            printf("%d\n",ans[i]);
        }
    }
    return 0;
}