BZOJ1146: [CTSC2008]网络管理Network

这题贼墨迹，写了一天多。

复杂度O（nlog^4n）

用到了我会的最高端的数据结构

树链剖分套线段树套平衡树

做个树链剖分这样就可以用线段树了

做个线段树这样就可以用平衡树了

然后一层一层向下搜

二分答案

或许这就是高端的暴力吧

#include<bits/stdc++.h>
using namespace std;
const int N=100005;
struct que
{
    int l,r,k;
}q[N];
int n,m,cnt,sec,idx,ra,mx;
int id[N],bel[N],head[N],son[N],size[N],d[N],f[N],rt[N*4],a[N];
vector<int>v;
inline int get(int x){return lower_bound(v.begin(),v.end(),x)-v.begin()+1;}
struct node
{
    int to,nex;
}e[N*2];
void adde(int x,int y)
{
    e[++cnt].to=y;e[cnt].nex=head[x];head[x]=cnt;
}
struct treap
{
    int l,r,s,v,rnd,w;
}t[N*50];
void dfs1(int x)
{
    size[x]=1;
    for(int i=head[x];i;i=e[i].nex)
    {
        int y=e[i].to;
        if(y==f[x])continue;
        d[y]=d[x]+1;f[y]=x;dfs1(y);
        size[x]+=size[y];
        if(size[y]>size[son[x]])son[x]=y;
    }
}
void dfs2(int x,int chain)
{
    id[x]=++idx;bel[x]=chain;
    if(son[x])dfs2(son[x],chain);
    for(int i=head[x];i;i=e[i].nex)
    {
        int y=e[i].to;
        if(y==f[x]||y==son[x])continue;
        dfs2(y,y);
    }
}
int lca(int x,int y)
{
    while(bel[x]!=bel[y])
    {
        if(d[bel[x]]<d[bel[y]])swap(x,y);
        x=f[bel[x]];
    }
    return d[x]<d[y]?x:y;
}//树链剖分

void update(int k){t[k].s=t[t[k].l].s+t[t[k].r].s+t[k].w;}
void lturn(int &k){int rs=t[k].r;t[k].r=t[rs].l;t[rs].l=k;t[rs].s=t[k].s;update(k);k=rs;} 
void rturn(int &k){int ls=t[k].l;t[k].l=t[ls].r;t[ls].r=k;t[ls].s=t[k].s;update(k);k=ls;}
void insert(int &k,int num)
{
    if(!k)
    {
        k=++sec;
        t[k].w=t[k].s=1;
        t[k].v=num;
        t[k].rnd=rand();
        return;
    }
    t[k].s++;
    if(t[k].v==num)t[k].w++;
    else
    {
        if(t[k].v>num){
            insert(t[k].l,num);
            if(t[t[k].l].rnd>t[k].rnd)rturn(k);
        }
        else{
            insert(t[k].r,num);
            if(t[t[k].r].rnd>t[k].rnd)lturn(k);
        }
    }
}
void del(int &k,int num)
{
    if(!k)return;
    if(t[k].v==num)
    {
        if(t[k].w>1)t[k].s--,t[k].w--;
        else if(t[k].l*t[k].r==0)k=t[k].l+t[k].r;
        else
        {
            if(t[t[k].l].rnd>t[t[k].r].rnd){rturn(k);del(k,num);}
            else {lturn(k);del(k,num);}
        }
        return;
    }
    t[k].s--;
    if(t[k].v>=num)del(t[k].l,num);
    else del(t[k].r,num);
}
void getrank(int k,int num)
{
    if(!k)return;
    if(t[k].v==num){ra+=t[t[k].r].s;return;}
    if(t[k].v>num){
        ra+=t[t[k].r].s+t[k].w;getrank(t[k].l,num);
    }
    else getrank(t[k].r,num);
}//平衡树

void add(int p,int l,int r,int L,int R,int num)
{
    if(l==L&&r==R){getrank(rt[p],num);return;}
    int mid=l+r>>1;
    if(R<=mid)add(p<<1,l,mid,L,R,num);
    else if(L>mid)add(p<<1|1,mid+1,r,L,R,num);
    else add(p<<1,l,mid,L,mid,num),add(p<<1|1,mid+1,r,mid+1,R,num);
}
void change(int p,int l,int r,int pos,int x,int y)
{
    del(rt[p],x);insert(rt[p],y);
    if(l==r)return;
    int mid=(l+r)>>1;
    if(mid>=pos)change(p<<1,l,mid,pos,x,y);
    else change(p<<1|1,mid+1,r,pos,x,y);
}
void query(int x,int y,int num)
{
    while(bel[x]!=bel[y])
    {
        add(1,1,n,id[bel[x]],id[x],num);
        x=f[bel[x]];
    }
    add(1,1,n,id[y],id[x],num);
}
void solve(int x,int y,int k)
{
    int lc=lca(x,y);
    ra=0;
    query(x,lc,0);
    query(y,lc,0);ra--;
    if(ra<k){puts("invalid request!");return;}
    int l=1,r=mx,ans;
    while(l<=r)
    {
        int mid=l+r>>1;
        ra=0;query(x,lc,mid);query(y,lc,mid);
        if(a[lc]>mid)ra--;
        if(ra<k)ans=mid,r=mid-1;
        else l=mid+1;
    }
    printf("%d
",v[ans-1]); 
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;++i)
    {
        scanf("%d",&a[i]);v.push_back(a[i]);
    }int x,y;
    for(int i=1;i<n;++i)
    {
        scanf("%d%d",&x,&y);
        adde(x,y);adde(y,x);
    }
    for(int i=1;i<=m;++i)
    {
        scanf("%d%d%d",&q[i].k,&q[i].l,&q[i].r);
        if(!q[i].k)
        {
            v.push_back(q[i].r);
        }
    }
    dfs1(1);dfs2(1,1);
    sort(v.begin(),v.end());v.erase(unique(v.begin(),v.end()),v.end());mx=v.size();
    for(int i=1;i<=n;++i)a[i]=get(a[i]);
    for(int i=1;i<=m;++i)if(!q[i].k)q[i].r=get(q[i].r);
    for(int i=1;i<=n;++i)change(1,1,n,id[i],0,a[i]);
    for(int i=1;i<=m;++i)
    {
        if(!q[i].k)
        {
            change(1,1,n,id[q[i].l],a[q[i].l],q[i].r);a[q[i].l]=q[i].r;
        }
        else
        {
            solve(q[i].l,q[i].r,q[i].k);
        }
    }
    return 0;
}