[BZOJ4811][YNOI2017]由乃的OJ(树链剖分+线段树)

起床困难综合症那题，只要从高往低贪心，每次暴力跑一边看这一位输入0和1分别得到什么结果即可。

放到序列上且带修改，只要对每位维护一个线段树，每个节点分别记录0和1从左往右和从右往左走完这段区间后变成的数即可。

放到树上，只要树链剖分即可。但这里有一个很大的常数k，实际上我们只需要一个数就可以记录64个二进制位的信息，于是复杂度$O(nlog^2 n)$

小错误毁一生。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define ls (x<<1)
#define rs (ls|1)
#define lson ls,L,mid
#define rson rs,mid+1,R
#define rep(i,l,r) for (int i=(l); i<=(r); i++)
#define For(i,x) for (int i=h[x],k; i; i=nxt[i])
typedef unsigned long long ll;
using namespace std;

const int N=100010;
ll z,I=1;
int n,m,k,u,v,x,y,op,cnt,tim,pos[N],sz[N],dep[N],son[N],fa[N],dfn[N],top[N];
int h[N],to[N<<1],nxt[N<<1];
struct P{ int op; ll x; }p[N];
struct D{ ll s0,s1; }v1[N<<2],v2[N<<2];
struct S{ int l,r; }lp[N],rp[N];

void add(int u,int v){ to[++cnt]=v; nxt[cnt]=h[u]; h[u]=cnt; }

void work(int x,P p){
    if (p.op==1) v1[x]=v2[x]=(D){0,p.x};
    if (p.op==2) v1[x]=v2[x]=(D){p.x,-1};
    if (p.op==3) v1[x]=v2[x]=(D){p.x,p.x^(-1)};
}

D operator +(D a,D b){ return (D){((~a.s0)&b.s0)|(a.s0&b.s1),((~a.s1)&b.s0)|(a.s1&b.s1)}; }
void upd(int x){ v1[x]=v1[ls]+v1[rs]; v2[x]=v2[rs]+v2[ls]; }

void dfs(int x){
    sz[x]=1; dep[x]=dep[fa[x]]+1;
    For(i,x) if ((k=to[i])!=fa[x]){
        fa[k]=x; dfs(k); sz[x]+=sz[k];
        if (sz[k]>sz[son[x]]) son[x]=k;
    }
}

void dfs2(int x,int tp){
    top[x]=tp; dfn[x]=++tim; pos[tim]=x;
    if (son[x]) dfs2(son[x],tp);
    For(i,x) if ((k=to[i])!=fa[x] && k!=son[x]) dfs2(k,k);
}

void build(int x,int L,int R){
    if (L==R){ work(x,p[pos[L]]); return; }
    int mid=(L+R)>>1;
    build(lson); build(rson); upd(x);
}

void mdf(int x,int L,int R,int pos,P p){
    if (L==R){ work(x,p); return; }
    int mid=(L+R)>>1;
    if (pos<=mid) mdf(lson,pos,p); else mdf(rson,pos,p);
    upd(x);
}

D que(int x,int L,int R,int l,int r,int k){
    if (L==l && r==R) return (k==1) ? v1[x] : v2[x];
    int mid=(L+R)>>1;
    if (r<=mid) return que(lson,l,r,k);
    else if (l>mid) return que(rson,l,r,k);
        else{
            D a=que(lson,l,mid,k),b=que(rson,mid+1,r,k);
            return (k==1) ? a+b : b+a;
        }
}

void solve(int x,int y,ll z){
    int ld=0,rd=0,rev=0;
    S *L=lp,*R=rp;
    for (; top[x]!=top[y]; x=fa[top[x]]){
        if (dep[top[x]]<dep[top[y]]) rev^=1,swap(x,y),swap(ld,rd),swap(L,R);
        L[++ld]=(S){dfn[top[x]],dfn[x]};
    }
    if (dep[x]<dep[y]) rev^=1,swap(x,y),swap(ld,rd),swap(L,R);
    L[++ld]=(S){dfn[y],dfn[x]};
    if (rev) swap(ld,rd),swap(L,R);
    D res=(D){0,-1};
    rep(i,1,ld) res=res+que(1,1,n,L[i].l,L[i].r,2);
    for (int i=rd; i; i--) res=res+que(1,1,n,R[i].l,R[i].r,1);
    ll s=0,c=0;
    for (int i=k-1; ~i; i--)
        if (res.s0&(I<<i)) s+=I<<i;
            else if ((res.s1&(I<<i)) && c+(I<<i)<=z) c+=I<<i,s+=I<<i;
    printf("%llu
",s);
}

int main(){
    freopen("bzoj4811.in","r",stdin);
    freopen("bzoj4811.out","w",stdout);
    scanf("%d%d%d",&n,&m,&k);
    rep(i,1,n) scanf("%d%llu",&p[i].op,&p[i].x);
    rep(i,2,n) scanf("%d%d",&u,&v),add(u,v),add(v,u);
    dfs(1); dfs2(1,1); build(1,1,n);
    rep(i,1,m){
        scanf("%d%d%d%llu",&op,&x,&y,&z);
        if (op==1) solve(x,y,z); else mdf(1,1,n,dfn[x],(P){y,z});
    }
    return 0;
}