题意
给你一个有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]);
}
}