算法竞赛——进阶指南——acwing353. 雨天的尾巴 树上差分+动态开点线段树 +线段树合并

每个点开一个权值线段树（表示每个点每个种类的个数），x-y加上z种类，相当于让x-y区间的z种类+1。
在树上就是 x的z种类+1，y的z种类+1，lca(x,y)的z种类-1，fa[lca(x,y)]的z种类-1.

上述操作用动态开店权值线段树维护单点更新即可。

时空复杂度均为nlogz  但合并时会多一个2倍的常数。（可以通过节点回收消除常数）

最后dfs求子树和时，用线段树合并即可。

如标题算法合并的模板题。

三个算法都弄熟练这题就能手敲出来了。

#include <cstdio>
#include<iostream>
#include<algorithm>
#include<queue>
#include<cstring>
using namespace std;
typedef long long ll;
const int M = 2e5+7;
 
int head[M],cnt;
void init(){cnt=0,memset(head,0,sizeof(head));}
struct EDGE{int to,nxt,w;}ee[M*2];
void add(int x,int y,int w){ee[++cnt].nxt=head[x],ee[cnt].w=w,ee[cnt].to=y,head[x]=cnt;}
 
int d[M];//深度
int dist[M];//点i到根节点的距离
int f[M][21];//点i的  (1<<j)辈祖先 
int n,m;
void bfs()
{
	queue<int>q;
	memset(d,0,sizeof(d));
	q.push(1);d[1]=1;
	while(q.size())
	{
		int x=q.front();q.pop();
		for(int i=head[x];i;i=ee[i].nxt)
		{
			int y=ee[i].to,w=ee[i].w;
			if(d[y])continue;
			d[y]=d[x]+1;
			dist[y]=dist[x]+w;
			f[y][0]=x;
			q.push(y);
		}
	}
	for(int k=1;k<=20;k++)
		for(int i=1;i<=n;i++)
			f[i][k]=f[f[i][k-1]][k-1];
}
int lca(int x,int y)
{
	if(d[x]>d[y])swap(x,y);
	for(int i=20;i>=0;i--)
		if(d[f[y][i]]>=d[x])y=f[y][i];
	if(x==y)return x;
	for(int i=20;i>=0;i--)
		if(f[x][i]!=f[y][i])x=f[x][i],y=f[y][i];
	return f[x][0];
}
int ls[M*80],rs[M*80],mx[M*80],rt[M*80],id[M*80],sz;
void up(int o,int l,int r,int x,int d)
{
	if(l==r)
	{
		mx[o]+=d;
		id[o]=l;
		return ;
	}
	int m=(l+r)/2;
	if(x<=m)
	{
		if(!ls[o])ls[o]=++sz;
		up(ls[o],l,m,x,d);
	}
	else
	{
		if(!rs[o])rs[o]=++sz;
		up(rs[o],m+1,r,x,d);
	}
	if(mx[ls[o]]>=mx[rs[o]])mx[o]=mx[ls[o]],id[o]=id[ls[o]];
	else mx[o]=mx[rs[o]],id[o]=id[rs[o]];
}
int mg(int o,int p,int l,int r)
{
	if(!p)return o;
	if(!o)return p;//返回节点不为空的id
	if(l==r)//到达叶子节点 
	{
		mx[o]+=mx[p];
		return o;
	 } 
	int m=(l+r)/2;
	ls[o]=mg(ls[o],ls[p],l,m);
	rs[o]=mg(rs[o],rs[p],m+1,r);
	if(mx[ls[o]]>=mx[rs[o]])mx[o]=mx[ls[o]],id[o]=id[ls[o]];
	else mx[o]=mx[rs[o]],id[o]=id[rs[o]];
	return o;
}
void dfs(int x,int fa)
{
	for(int i=head[x];i;i=ee[i].nxt)
	{
		int y=ee[i].to;
		if(y==fa)continue;
		dfs(y,x);
		rt[x]=mg(rt[x],rt[y],1,n);
	}
}
struct node{
	int x,y,z;
}p[M];
int li[M],lt;
int main()
{
	int u,v,w,x,y,z;
	scanf("%d%d",&n,&m);
	for(int i=1;i<n;i++)scanf("%d%d",&u,&v),add(u,v,1),add(v,u,1);
	bfs();
	for(int i=1;i<=n;i++)rt[i]=i;
	sz=n;
	for(int i=1;i<=m;i++)
	{
		scanf("%d%d%d",&x,&y,&z);
		p[i]=node{x,y,z};
		li[++lt]=z;
	}
	sort(li+1,li+1+lt);
	lt=unique(li+1,li+1+lt)-(li+1);
	for(int i=1;i<=m;i++)
	{
		x=p[i].x,y=p[i].y,z=lower_bound(li+1,li+1+lt,p[i].z)-li;
	//	cout<<x<<" -  "<<y<<"   "<<z<<endl;
		up(rt[x],1,n,z,1);
		up(rt[y],1,n,z,1);
		up(rt[lca(x,y)],1,n,z,-1);
		up(rt[f[lca(x,y)][0]],1,n,z,-1);
	}
//	for(int i=1;i<=n;i++)cout<<mx[rt[i]]<<" ";
	//cout<<endl;
	dfs(1,0);
	for(int i=1;i<=n;i++)printf("%d\n",li[id[rt[i]]]);
	return 0;
}