luoguP4242树上的毒瘤

传送门

模板集合吧，除了码农，没啥难的。。。

和bzoj2243:[SDOI2011]染色十分相像，但是多了点集和查询的区别

然后点集显然可以看出是虚树问题，查询可以用点分治\(O(nlogn)\)，也可以二次换根\(O(n)\)，显然二次换根\(O(n)\)更加优秀

然后码就是了

代码：

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
void read(int &x) {
	char ch; bool ok;
	for(ok=0,ch=getchar(); !isdigit(ch); ch=getchar()) if(ch=='-') ok=1;
	for(x=0; isdigit(ch); x=x*10+ch-'0',ch=getchar()); if(ok) x=-x;
}
#define rg register
const int maxn=1e5+10;
int n,m,c[maxn],dep[maxn],st[maxn],top[maxn],id[maxn],tmp,f[maxn][20];
int nid[maxn],size[maxn],d[maxn],tot,w[maxn],g[maxn],ans[maxn],ssize[maxn];
struct oo{
	int pre[maxn*2],nxt[maxn*2],h[maxn],cnt;
	void add(int x,int y)
	{
		pre[++cnt]=y,nxt[cnt]=h[x],h[x]=cnt;
		pre[++cnt]=x,nxt[cnt]=h[y],h[y]=cnt;
	}
}a,b;
struct segment_tree{int l,r,sum,ls,rs,cov;}s[maxn*4];
void update(int x)
{
	s[x].sum=s[x<<1].sum+s[x<<1|1].sum-(s[x<<1].rs==s[x<<1|1].ls);
	s[x].ls=s[x<<1].ls,s[x].rs=s[x<<1|1].rs;
}
void build(int x,int l,int r)
{
	s[x].l=l,s[x].r=r;int mid=(l+r)>>1;
	if(l==r){s[x].sum=1,s[x].ls=s[x].rs=c[nid[l]];return ;}
	build(x<<1,l,mid),build(x<<1|1,mid+1,r);
	update(x);
}
void dfs(int x)
{
	size[x]=1;
	for(rg int i=1;i<20;i++)
	{
		if(dep[x]<(1<<i))break;
		f[x][i]=f[f[x][i-1]][i-1];
	}
	for(rg int i=a.h[x];i;i=a.nxt[i])
		if(a.pre[i]!=f[x][0])
		{
			f[a.pre[i]][0]=x,dep[a.pre[i]]=dep[x]+1;
			dfs(a.pre[i]),size[x]+=size[a.pre[i]];
		}
}
void dfs1(int x,int f)
{
	top[x]=f,id[x]=++tmp,nid[tmp]=x;int k=0;
	for(int i=a.h[x];i;i=a.nxt[i])
		if(dep[a.pre[i]]>dep[x]&&size[a.pre[i]]>size[k])k=a.pre[i];
	if(!k)return ;dfs1(k,f);
	for(rg int i=a.h[x];i;i=a.nxt[i])
		if(dep[a.pre[i]]>dep[x]&&a.pre[i]!=k)dfs1(a.pre[i],a.pre[i]);
}
int lca(int x,int y)
{
	if(dep[x]>dep[y])swap(x,y);
	int poor=dep[y]-dep[x];
	for(rg int i=19;i>=0;i--)if(poor&(1<<i))y=f[y][i];
	if(x==y)return x;
	for(rg int i=19;i>=0;i--)if(f[x][i]!=f[y][i])x=f[x][i],y=f[y][i];
	return x==y?x:f[x][0];
}
void pushdown(int x)
{
	s[x<<1].sum=s[x<<1|1].sum=1;
	s[x<<1].cov=s[x<<1|1].cov=s[x].cov;
	s[x<<1].ls=s[x<<1].rs=s[x].cov;
	s[x<<1|1].ls=s[x<<1|1].rs=s[x].cov;
	s[x].cov=0;
}
void change(int x,int l,int r,int v)
{
	if(l<=s[x].l&&r>=s[x].r)
	{
		s[x].cov=s[x].ls=s[x].rs=v,s[x].sum=1;
		return ;
	}
	if(s[x].cov)pushdown(x);
	int mid=(s[x].l+s[x].r)>>1;
	if(l<=mid)change(x<<1,l,r,v);
	if(r>mid)change(x<<1|1,l,r,v);
	update(x);
}
void modify(int x,int y,int z)
{
	while(top[x]!=top[y])
	{
		if(dep[top[x]]<dep[top[y]])swap(x,y);
		change(1,id[top[x]],id[x],z),x=f[top[x]][0];
	}
	if(id[x]>id[y])swap(x,y);
	change(1,id[x],id[y],z);
}
bool cmp(int x,int y){return id[x]<id[y];}
segment_tree get(int x,int l,int r)
{
	if(l<=s[x].l&&r>=s[x].r)return s[x];
	int mid=(s[x].l+s[x].r)>>1;
	if(s[x].cov)pushdown(x);
	if(r<=mid)return get(x<<1,l,r);
	else if(l>mid)return get(x<<1|1,l,r);
	else
	{
		segment_tree a=get(x<<1,l,r),b=get(x<<1|1,l,r),c;
		c.sum=a.sum+b.sum-(a.rs==b.ls);
		c.ls=a.ls,c.rs=b.rs;
		return c;
	}
}
int qsum(int x,int y)
{
	int sum=0;segment_tree a,b;
	while(top[x]!=top[y])
	{
		if(dep[top[x]]<dep[top[y]])swap(x,y);
		a=get(1,id[top[x]],id[x]);
		x=f[top[x]][0],b=get(1,id[x],id[x]);
		sum=sum+a.sum-(b.rs==a.ls);
	}
	if(id[x]>id[y])swap(x,y);
	a=get(1,id[x],id[y]),sum+=a.sum;
	return sum;
}
void prepare(int x,int fa)
{
	ssize[x]=0;if(w[x])ssize[x]++;
	for(rg int i=b.h[x];i;i=b.nxt[i])
		if(b.pre[i]!=fa)prepare(b.pre[i],x),ssize[x]+=ssize[b.pre[i]];
}
void solve(int x,int fa,int tot)
{
	if(w[x])ans[w[x]]=g[x];
	for(rg int i=b.h[x];i;i=b.nxt[i])
		if(b.pre[i]!=fa)
		{
			int now=qsum(x,b.pre[i])-1;
			g[b.pre[i]]=g[x]+now*(tot-2*ssize[b.pre[i]]);
			solve(b.pre[i],x,tot);
		}
}
int main()
{
	read(n),read(m);
	for(rg int i=1;i<=n;i++)read(c[i]);
	for(rg int i=1,x,y;i<n;i++)read(x),read(y),a.add(x,y);
	dfs(1),dfs1(1,1),build(1,1,n);
	for(rg int i=1,q,x,y,z;i<=m;i++)
	{
		read(q);
		if(q==1)read(x),read(y),read(z),modify(x,y,z);
		else
		{
			read(z);tot=0;b.cnt=0;memset(b.h,0,sizeof b.h);
			for(rg int j=1;j<=z;j++)read(d[j]),w[d[j]]=j;
			sort(d+1,d+z+1,cmp),st[++tot]=1;
			for(rg int j=1;j<=z;j++)
			{
				int w=0,e=0;
				while(tot&&lca(st[tot],d[j])!=st[tot])
				{
					if(w)b.add(st[tot],w);
					w=st[tot],tot--;
				}
				if(w)e=lca(w,d[j]),b.add(w,e);
				if(e&&e!=st[tot])st[++tot]=e;
				st[++tot]=d[j];
			}
			while(tot>1)b.add(st[tot],st[tot-1]),tot--;
			for(rg int j=1;j<=z;j++)g[1]=g[1]+qsum(1,d[j]);
			prepare(1,0),solve(1,0,z);
			for(rg int j=1;j<=z;j++)printf("%d ",ans[j]);printf("\n");
			for(rg int j=1;j<=z;j++)w[d[j]]=0,g[j]=0;
		}
	}
}