2019西安邀请赛 E-Tree 树链剖分+线段树

题目链接：https://nanti.jisuanke.com/t/39272

题意：n个点的树。每个点有一个权值，3种操作

1 s t,1到s的路径上所有点或上t

2 s t,1到s的路径上所有点与上t

3 s t,1到s的路径上所有点的加上 t 进行尼姆游戏看先手是否能胜出

题解：很明显这个题的是要用树链剖分来做，问题就在于我们怎么处理或和与这两种操作，我们拆成二进制来看，对于或来说，为1的位置，路径所有点这个位置都要变成1，为0的位置不变，对于与来说，为1的位置不变，为0的位置都要变成0，并且这两种操作对于某个位置来说是起到覆盖效果的，那么我们就可以把这两种操作结合起来了，laz[i] = 1 时标记这个位置都要变为1，laz[i] = -1 标记这个位置都要变为0，然后再维护下，区间每个位置1的个数，当个数为奇数时才对尼姆游戏异或的结果有效。

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int N=1e5+10;
struct node{
	int l,r,len;
	int num[32];
	int laz[32];
	int sum;
}tree[N<<2];
struct edge{
	int to,nex;
}e[N*2];
int head[N],len;
int n,m,val[N];
int in[N],out[N],p[N],cnt;
int dep[N],fath[N],f[N];
int son[N],son_id[N];
void addedge(int x,int y)
{
	e[len].to=y;
	e[len].nex=head[x];
	head[x]=len++;
}
void dfs1(int u,int fa)
{
	int to;
	son[u]=1;
	dep[u]=dep[fa]+1;
	for(int i=head[u];i!=-1;i=e[i].nex)
	{
		to=e[i].to;
		if(to==fa) continue;
		dfs1(to,u);
		son[u]+=son[to];
		if(son[to] > son[son_id[u]])
		{
			son_id[u]=to;
		}
	}
}
void dfs2(int u,int fa,int rt)
{
	int to;
	in[u]=++cnt;
	p[cnt]=u;
	fath[u]=rt;
	f[u]=fa;
	if(son_id[u])
	{
		dfs2(son_id[u],u,rt);
	}
	for(int i=head[u];i!=-1;i=e[i].nex)
	{
		to=e[i].to;
		if(to==fa || to==son_id[u]) continue;
		dfs2(to,u,to);
	}
	out[u]=cnt;
}
void pushup(int cur)
{
	for(int i=0;i<30;i++)
		tree[cur].num[i]=tree[cur<<1].num[i]+tree[cur<<1|1].num[i];
}
void pushdown(int cur)
{
	for(int i=0;i<30;i++)
	{
		if(tree[cur].laz[i]==1)
		{
			tree[cur<<1].num[i]=tree[cur<<1].len;
			tree[cur<<1].laz[i]=1;
			
			tree[cur<<1|1].num[i]=tree[cur<<1|1].len;
			tree[cur<<1|1].laz[i]=1;
			
			tree[cur].laz[i]=0;
		}
		if(tree[cur].laz[i]==-1)
		{
			tree[cur<<1].num[i]=0;
			tree[cur<<1].laz[i]=-1;
			
			tree[cur<<1|1].num[i]=0;
			tree[cur<<1|1].laz[i]=-1;
			
			tree[cur].laz[i]=0;
		}
	}
}
void build(int l,int r,int cur)
{
	tree[cur].l=l;
	tree[cur].r=r;
	tree[cur].len=r-l+1;
	if(l==r)
	{
		for(int i=0;i<30;i++)
		{
			if((1<<i) & val[p[l]])
				tree[cur].num[i]=1;
		}
		return;
	}
	int mid=(r+l)>>1;
	build(l,mid,cur<<1);
	build(mid+1,r,cur<<1|1);
	pushup(cur);
}
void update(int pl,int pr,int cur,int op,int v)
{
	if(pl<=tree[cur].l && tree[cur].r<=pr)
	{
		if(op==1)
		{
			for(int i=0;i<30;i++)
			{
				if((1<<i) & v)
				{
					tree[cur].num[i]=tree[cur].len;
					tree[cur].laz[i]=1;
				}
			}
		}
		else
		{
			for(int i=0;i<30;i++)
			{
				if(!((1<<i) & v))
				{
					tree[cur].num[i]=0;
					tree[cur].laz[i]=-1;
				}
			}
		}
		return;
	}
	pushdown(cur);
	if(pl<=tree[cur<<1].r) update(pl,pr,cur<<1,op,v);
	if(pr>=tree[cur<<1|1].l) update(pl,pr,cur<<1|1,op,v);
	pushup(cur);
}
void solve(int x,int y,int v,int op)
{
	int fy;
	while(fath[y]!=x)
	{
		fy=fath[y];
		update(in[fy],in[y],1,op,v);
		y=f[fy];
	}
	update(in[x],in[y],1,op,v);
}
int ans;
void query(int pl,int pr,int cur)
{
	if(pl<=tree[cur].l && tree[cur].r<=pr)
	{
		for(int i=0;i<30;i++)
		{
			if(tree[cur].num[i] & 1)
				ans^=(1<<i);
		}
		return;
	}
	pushdown(cur);
	if(pl<=tree[cur<<1].r) query(pl,pr,cur<<1);
	if(pr>=tree[cur<<1|1].l) query(pl,pr,cur<<1|1);
}
int main()
{
	int x,y;
	int op;
	int fy;
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&val[i]);
		head[i]=-1;
	}
//	cout<<1<<endl;
	for(int i=1;i<n;i++)
	{
		scanf("%d%d",&x,&y);
		addedge(x,y);
		addedge(y,x);
	}
	dfs1(1,0);
	dfs2(1,0,1);
	build(1,n,1);
	while(m--)
	{
		scanf("%d%d%d",&op,&x,&y);
		if(op==1 || op==2)
		{
			solve(1,x,y,op);
		}
		else
		{
			ans=0;
			while(fath[x]!=1)
			{
				fy=fath[x];
				query(in[fy],in[x],1);
				x=f[fy];
			}
			query(1,in[x],1);
			ans^=y;
			printf(ans?"YES\n":"NO\n");
		}
	}
	return 0;
}