[SP16580]QTREE7

luogu
vjudge

题意

一棵树,每个点初始有个点权和颜色(输入会给你)
0 u :询问所有u,v路径上的最大点权,要满足u,v路径上所有点的颜色都相同
1 u :反转u的颜色
2 u w :把u的点权改成w
(color_iin [0,1],w_iin [−10^9,10^9],n,m≤10^5)

sol

和(QTree 6)很像啊。丢个链接就跑
只要额外对每个点维护一个虚子树的最大权值即可。
开一个(multiset)写起来比较爽。
然后就做完了

code

我不会说我因为快读打错了debug了一个晚上的

#include<cstdio>
#include<algorithm>
#include<vector>
#include<set>
using namespace std;
int gi()
{
	int x=0,w=1;char ch=getchar();
	while ((ch<'0'||ch>'9')&&ch!='-') ch=getchar();
	if (ch=='-') w=0,ch=getchar();
	while (ch>='0'&&ch<='9') x=(x<<3)+(x<<1)+ch-'0',ch=getchar();
	return w?x:-x;
}
const int N = 1e5+5;
int n,m,col[N],w[N],fa[N];
vector<int>G[N];
struct LCT{
	int fa[N],ch[2][N],mx[N];
	multiset<int>S[N];
	bool son(int x){return x==ch[1][fa[x]];}
	bool isroot(int x)
		{
			return x!=ch[0][fa[x]]&&x!=ch[1][fa[x]];
		}
	void pushup(int x)
		{
			mx[x]=max(max(mx[ch[0][x]],mx[ch[1][x]]),w[x]);
			if (!S[x].empty()) mx[x]=max(mx[x],*S[x].rbegin());
		}
	void rotate(int x)
		{
			int y=fa[x],z=fa[y],c=son(x);
			ch[c][y]=ch[c^1][x];if (ch[c][y]) fa[ch[c][y]]=y;
			fa[x]=z;if (!isroot(y)) ch[son(y)][z]=x;
			ch[c^1][x]=y;fa[y]=x;pushup(y);
		}
	void splay(int x)
		{
			for (int y=fa[x];!isroot(x);rotate(x),y=fa[x])
				if (!isroot(y)) son(x)^son(y)?rotate(x):rotate(y);
			pushup(x);
		}
	void access(int x)
		{
			for (int y=0;x;y=x,x=fa[x])
			{
				splay(x);if (ch[1][x]) S[x].insert(mx[ch[1][x]]);
				ch[1][x]=y;if (ch[1][x]) S[x].erase(mx[ch[1][x]]);
				pushup(x);
			}
		}
	int findroot(int x)
		{
			access(x);splay(x);
			while (ch[0][x]) x=ch[0][x];
			splay(x);return x;
		}
	void link(int x,int y)
		{
			if (!y) return;
			access(y);splay(y);splay(x);
			fa[x]=y;S[y].insert(mx[x]);pushup(y);
		}
	void cut(int x,int y)
		{
			if (!y) return;
			access(x);splay(x);
			ch[0][x]=fa[ch[0][x]]=0;pushup(x);
		}
}T[2];
void dfs(int u,int f)
{
	for (int i=0,sz=G[u].size();i<sz;++i)
	{
		int v=G[u][i];if (v==f) continue;
		T[col[v]].link(v,u);fa[v]=u;
		dfs(v,u);
	}
}
int main()
{
	n=gi();
	for (int i=1;i<n;++i)
	{
		int u=gi(),v=gi();
		G[u].push_back(v);G[v].push_back(u);
	}
	for (int i=1;i<=n;++i) col[i]=gi();
	for (int i=1;i<=n;++i) w[i]=gi();
	T[0].mx[0]=T[1].mx[0]=-2147483647;
	dfs(1,0);m=gi();
	while (m--)
	{
		int opt=gi(),u=gi();
		if (!opt){
			T[col[u]].access(u);int ff=T[col[u]].findroot(u);
			if (col[ff]==col[u]) printf("%d
",T[col[u]].mx[ff]);
			else printf("%d
",T[col[u]].mx[T[col[u]].ch[1][T[col[u]].findroot(u)]]);
		}
		else if (opt==1) T[col[u]].cut(u,fa[u]),col[u]^=1,T[col[u]].link(u,fa[u]);
		else{
			T[col[u]].access(u);T[col[u]].splay(u);
			w[u]=gi();T[col[u]].pushup(u);
		}
	}
	return 0;
}