BZOJ 4129: Haruna’s Breakfast [树上莫队 分块]

传送门

题意：

单点修改，求一条链的mex

---

分块维护权值，$O(1)$修改$O(S)$求mex......

带修改树上莫队

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <cmath>
using namespace std;
typedef long long ll;
const int N=5e4+;
inline int read(){
    char c=getchar();int x=,f=;
    while(c<''||c>''){if(c=='-')f=-;c=getchar();}
    while(c>=''&&c<=''){x=x*+c-'';c=getchar();}
    return x*f;
}
 
int n,Q,a[N],t[N],op,x,y;
struct edge{int v,ne;} e[N<<];
int cnt,h[N];
inline void ins(int u,int v){
    e[++cnt]=(edge){v,h[u]}; h[u]=cnt;
    e[++cnt]=(edge){u,h[v]}; h[v]=cnt;
}
int block, pos[N], m, st[N], top;
int fa[N][], deep[N];
void dfs(int u){
    for(int i=; (<<i)<=deep[u]; i++)
        fa[u][i]=fa[ fa[u][i-] ][i-];
    int bot=top;
    for(int i=h[u];i;i=e[i].ne) if(e[i].v!=fa[u][]){
        fa[e[i].v][]=u; deep[e[i].v]=deep[u]+;
        dfs(e[i].v);
        if(top-bot>=block){
            m++;
            while(top!=bot) pos[st[top--]]=m;
        }
    }
    st[++top]=u;
}
inline int lca(int x,int y){
    if(deep[x]<deep[y]) swap(x, y);
    int bin=deep[x]-deep[y];
    for(int i=;i<;i++) if((<<i)&bin) x=fa[x][i];
    if(x==y) return x;
    for(int i=;i>=;i--)
        if(fa[x][i]!=fa[y][i]) x=fa[x][i], y=fa[y][i];
    return x==y ? x : fa[x][];
}
 
struct cmeow{int u, next, last;}cq[N];
struct meow{
    int u, v, tim, id;
    bool operator <(const meow &a) const {
        return pos[u]==pos[a.u] ? (pos[v]==pos[a.v] ? tim<a.tim : pos[v]<pos[a.v]) : pos[u]<pos[a.u];
    }
}q[N];
int p, tim, u, v, cur, ans[N], vis[N];
 
struct Block{
    int n,pos[N],m,block;
    struct _blo{int l,r;} b[];
    void ini(){
        block=sqrt(n); m=(n-)/block+;
        for(int i=;i<=n;i++) pos[i]=(i-)/block+;
        for(int i=;i<=m;i++) b[i].l=(i-)*block+, b[i].r=i*block;
        b[m].r=n;
    }
    int cou[N], sum[];
    inline void add(int v) { if(v<=n) sum[pos[v]]+= (++cou[v])==; }
    inline void del(int v) { if(v<=n) sum[pos[v]]-= (--cou[v])==; }
    inline int mex(){
        for(int i=;i<=m;i++) if(sum[i] != b[i].r-b[i].l+)
            for(int j=b[i].l; j<=b[i].r; j++) if(!cou[j]) return j;
        return -;
    }
}B;
inline void cha(int u,int d){
    if(vis[u]) B.del(a[u]), B.add(d);
    a[u]=d;
}
inline void Xor(int u){
    if(vis[u]) B.del(a[u]), vis[u]=;
    else B.add(a[u]), vis[u]=;
}
inline void move(int x,int y){
    if(deep[x]<deep[y]) swap(x, y);
    while(deep[x]>deep[y]) Xor(x), x=fa[x][];
    while(x!=y) Xor(x), Xor(y), x=fa[x][], y=fa[y][];
}
 
void modui(){
    u=v=;
    for(int i=;i<=p;i++){
        while(cur<q[i].tim) cur++, cha(cq[cur].u, cq[cur].next);
        while(cur>q[i].tim) cha(cq[cur].u, cq[cur].last), cur--;
        if(u!=q[i].u) move(u, q[i].u), u=q[i].u;
        if(v!=q[i].v) move(v, q[i].v), v=q[i].v;
        int anc=lca(u, v);
        Xor(anc); ans[q[i].id]=B.mex()-; Xor(anc);
    }
}
 
int main(){
    freopen("in","r",stdin);
    n=read(); Q=read();
    for(int i=;i<=n;i++) a[i]=t[i]=read()+;
    for(int i=;i<n ;i++) ins(read(), read() );
    block=pow(n, 0.45);
    dfs();
    while(top) pos[st[top--]]=m;
    for(int i=;i<=Q;i++){
        op=read(); x=read(); y=read();
        if(op) p++, q[p]=(meow){x, y, tim, p};
        else tim++, cq[tim]=(cmeow){x, y+, t[x]}, t[x]=y+;
    } 
 
    B.n = n+; B.ini();
    sort(q+, q++p);
    modui();
    for(int i=;i<=p;i++) printf("%d\n",ans[i]);
}