bzoj 4811: [Ynoi2017]由乃的OJ 树链剖分+线段树+贪心

题意

给你一个有n个点的树，每个点的包括一个位运算opt和一个权值x，位运算有&,l,^三种，分别用1,2,3表示。每次询问包含三个数x,y,z,初始选定一个数v。然后v依次经过从x到y的所有节点，每经过一个点i，v就变成v opti xi，所以他想问你，最后到y时，希望得到的值尽可能大，求最大值？给定的初始值v必须是在[0,z]之间。每次修改包含三个数x,y,z，意思是把x点的操作修改为y，数值改为z
0 <= n , m <= 100000 , k <= 64

分析

由于是要结果尽量大且位与位之间的运算相互独立，考虑从高位开始逐位贪心。现在的问题是如何求出对于每一位，放0和放1在经过链上的所有操作后，得到的结果是什么。
用树链剖分和线段树来维护树链，对于线段树的每一个节点，记录v0,v1分别表示每一位初始是0时经过这个区间后会变成什么和初始是1时的情况。合并的话比较简单。
把操作结果求出来后从高位到低位贪心就好了。
一遍AC还是很开心的。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#define mp(x,y) std::make_pair(x,y)

typedef unsigned long long llu;
typedef std::pair<int,int> pi;

const int N=100005;

int n,m,k,cnt,last[N],size[N],fa[N],dep[N],pos[N],top[N],tim,bel[N],op[N];
llu bin[70],a[N];
pi stack[N];
struct edge{int to,next;}e[N*2];
struct data{llu v0,v1;};
struct tree{data lef,rig;}t[N*4];

void addedge(int u,int v)
{
    e[++cnt].to=v;e[cnt].next=last[u];last[u]=cnt;
    e[++cnt].to=u;e[cnt].next=last[v];last[v]=cnt;
}

void dfs1(int x)
{
    size[x]=1;dep[x]=dep[fa[x]]+1;
    for (int i=last[x];i;i=e[i].next)
    {
        if (e[i].to==fa[x]) continue;
        fa[e[i].to]=x;
        dfs1(e[i].to);
        size[x]+=size[e[i].to];
    }
}

void dfs2(int x,int chain)
{
    pos[x]=++tim;bel[tim]=x;top[x]=chain;int k=0;
    for (int i=last[x];i;i=e[i].next)
        if (e[i].to!=fa[x]&&size[e[i].to]>size[k]) k=e[i].to;
    if (!k) return;
    dfs2(k,chain);
    for (int i=last[x];i;i=e[i].next)
        if (e[i].to!=fa[x]&&e[i].to!=k) dfs2(e[i].to,e[i].to);
}

data merge(data x,data y)
{
    data ans;
    ans.v0=(x.v0&y.v1)|((x.v0^(bin[k]-1))&y.v0);
    ans.v1=(x.v1&y.v1)|((x.v1^(bin[k]-1))&y.v0);
    return ans;
}

void build(int d,int l,int r)
{
    if (l==r)
    {
        if (op[bel[l]]==1) t[d].lef=t[d].rig=(data){0,a[bel[l]]};
        else if (op[bel[l]]==2) t[d].lef=t[d].rig=(data){a[bel[l]],bin[k]-1};
        else t[d].lef=t[d].rig=(data){a[bel[l]],(bin[k]-1)^a[bel[l]]};
        return;
    }
    int mid=(l+r)/2;
    build(d*2,l,mid);build(d*2+1,mid+1,r);
    t[d].lef=merge(t[d*2].lef,t[d*2+1].lef);
    t[d].rig=merge(t[d*2+1].rig,t[d*2].rig);
}

void modify(int d,int l,int r,int x)
{
    if (l==r)
    {
        if (op[bel[l]]==1) t[d].lef=t[d].rig=(data){0,a[bel[l]]};
        else if (op[bel[l]]==2) t[d].lef=t[d].rig=(data){a[bel[l]],bin[k]-1};
        else t[d].lef=t[d].rig=(data){a[bel[l]],(bin[k]-1)^a[bel[l]]};
        return;
    }
    int mid=(l+r)/2;
    if (x<=mid) modify(d*2,l,mid,x);
    else modify(d*2+1,mid+1,r,x);
    t[d].lef=merge(t[d*2].lef,t[d*2+1].lef);
    t[d].rig=merge(t[d*2+1].rig,t[d*2].rig);
}

data query(int d,int l,int r,int x,int y,int ty)
{
    if (x<=l&&r<=y) return (!ty)?t[d].lef:t[d].rig;
    int mid=(l+r)/2;data ans=(data){0,bin[k]-1};
    if (!ty)
    {
        if (x<=mid) ans=merge(ans,query(d*2,l,mid,x,y,ty));
        if (y>mid) ans=merge(ans,query(d*2+1,mid+1,r,x,y,ty));
    }
    else
    {
        if (y>mid) ans=merge(ans,query(d*2+1,mid+1,r,x,y,ty));
        if (x<=mid) ans=merge(ans,query(d*2,l,mid,x,y,ty));
    }
    return ans;
}

int get_lca(int x,int y)
{
    while (top[x]!=top[y])
    {
        if (dep[top[x]]<dep[top[y]]) std::swap(x,y);
        x=fa[top[x]];
    }
    return dep[x]<dep[y]?x:y;
}

llu solve(int x,int y,llu z)
{
    int lca=get_lca(x,y);data w=(data){0,bin[k]-1};
    while (top[x]!=top[lca])
    {
        w=merge(w,query(1,1,n,pos[top[x]],pos[x],1));
        x=fa[top[x]];
    }
    w=merge(w,query(1,1,n,pos[lca],pos[x],1));
    int h=0;
    while (top[y]!=top[lca])
    {
        stack[++h]=mp(pos[top[y]],pos[y]);
        y=fa[top[y]];
    }
    if (y!=lca) w=merge(w,query(1,1,n,pos[lca]+1,pos[y],0));
    while (h) w=merge(w,query(1,1,n,stack[h].first,stack[h].second,0)),h--;
    llu ans=0;
    for (int i=k-1;i>=0;i--)
        if (w.v0&bin[i]) ans+=bin[i];
        else if ((w.v1&bin[i])&&bin[i]<=z) z-=bin[i],ans+=bin[i];
    return ans;
}

int main()
{
    scanf("%d%d%d",&n,&m,&k);
    bin[0]=1;
    for (int i=1;i<=k;i++) bin[i]=bin[i-1]*2;
    for (int i=1;i<=n;i++) scanf("%d%llu",&op[i],&a[i]);
    for (int i=1;i<n;i++)
    {
        int x,y;scanf("%d%d",&x,&y);
        addedge(x,y);
    }
    dfs1(1);dfs2(1,1);
    build(1,1,n);
    while (m--)
    {
        int q,x,y;llu z;scanf("%d%d%d%llu",&q,&x,&y,&z);
        if (q==1) printf("%llu\n",solve(x,y,z));
        else op[x]=y,a[x]=z,modify(1,1,n,pos[x]);
    }
}