【bzoj3307】雨天的尾巴 线段树+树链剖分

考虑如果在序列上的话，每次只需要在区间左端点+1，右端点-1，最后用线段树一起统计即可。

在树上的话，就用树链剖分转换成序列，每一次操作，对应修改O(log n)条重链上的标记，最后用线段树来统计。

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<algorithm>
#include<iostream>
#include<vector>
#include<map>
#define maxn 200100

using namespace std;

struct yts
{
	int l,r;
	int mx,tag;
}t[4*maxn];

int head[maxn],st[maxn],to[maxn],next[maxn],seq[maxn],rank[maxn],fa[19][maxn],size[maxn],dd[maxn];
int n,m,num,tot,T,cnt;
int e[maxn],pos[maxn],ans[maxn],dep[maxn];
bool vis[maxn];
vector<pair<int,int> > v[maxn];
map<int,int> mp;

void addedge(int x,int y)
{
	num++;to[num]=y;next[num]=st[x];st[x]=num;
}

void dfs(int x)
{
	size[x]=1;e[++tot]=x;
	for (int p=st[x];p;p=next[p])
	  if (to[p]!=fa[0][x])
	  {
	  	fa[0][to[p]]=x;
	  	dep[to[p]]=dep[x]+1;
	  	dfs(to[p]);
	  	size[x]+=size[to[p]];
	  	if (size[to[p]]>size[dd[x]]) dd[x]=to[p];
	  }
}

int go_up(int x,int d)
{
	for (int i=0;i<=17;i++)
	  if (d&(1<<i)) x=fa[i][x];
	return x;
}

int LCA(int x,int y)
{
	if (dep[x]>dep[y]) x=go_up(x,dep[x]-dep[y]);
	else y=go_up(y,dep[y]-dep[x]);
	if (x==y) return x;
	for (int i=17;i>=0;i--)
	  if (fa[i][x]!=fa[i][y]) x=fa[i][x],y=fa[i][y];
	return fa[0][x];
}

void add(int x,int a,int b)
{
	v[x].push_back(make_pair(a,b));
}

void update(int i)
{
	t[i].mx=0;t[i].tag=0;
	if (t[i<<1].mx>t[i].mx) t[i].mx=t[i<<1].mx,t[i].tag=t[i<<1].tag;
	if (t[i<<1|1].mx>t[i].mx) t[i].mx=t[i<<1|1].mx,t[i].tag=t[i<<1|1].tag;
	else if (t[i<<1|1].mx==t[i].mx && t[i].tag>t[i<<1|1].tag) t[i].tag=t[i<<1|1].tag; 
}

void build(int i,int l,int r)
{
	t[i].l=l;t[i].r=r;t[i].mx=0;
	if (l==r) {t[i].tag=pos[l];return;}
	int mid=(l+r)/2;
	build(i*2,l,mid);build(i*2+1,mid+1,r);
}

void modify(int i,int x,int d)
{
	if (t[i].l==t[i].r) {t[i].mx+=d;return;}
	int mid=(t[i].l+t[i].r)/2;
	if (x<=mid) modify(i*2,x,d);
	if (mid<x) modify(i*2+1,x,d);
	update(i);
}

int main()
{
	scanf("%d%d",&n,&T);
	for (int i=1;i<n;i++)
	{
		int x,y;
		scanf("%d%d",&x,&y);
		addedge(x,y);addedge(y,x);
	}
	dfs(1);
	int qwer=0;
	for (int i=1;i<=n;i++)
	  if (!vis[e[i]])
	  {
	  	int k=e[i];
	  	while (k)
	  	{
	  		seq[++qwer]=k;vis[k]=1;head[k]=e[i];k=dd[k];
	  	}
	  }
	for (int i=1;i<=n;i++) rank[seq[i]]=i;
	for (int j=1;j<=17;j++)
	  for (int i=1;i<=n;i++)
	    fa[j][i]=fa[j-1][fa[j-1][i]];
	while (T--)
	{
		int x,y,z;
		scanf("%d%d%d",&x,&y,&z);
		if (dep[x]<dep[y]) swap(x,y);
		if (!mp[z]) mp[z]=++cnt,pos[cnt]=z;
		int lca=LCA(x,y);
		z=mp[z];
		while (dep[head[x]]>dep[lca])
		{
			add(rank[head[x]],z,1);add(rank[x]+1,z,-1);x=fa[0][head[x]];
		}
		add(rank[lca],z,1);add(rank[x]+1,z,-1);
		if (y!=lca)
		{
			int w=go_up(y,dep[y]-dep[lca]-1);
			while (dep[head[y]]>dep[w])
			{
				add(rank[head[y]],z,1);add(rank[y]+1,z,-1);y=fa[0][head[y]];
			}
			add(rank[w],z,1);add(rank[y]+1,z,-1);
		}
	}
	build(1,1,cnt);
	for (int i=1;i<=n;i++)
	{
		for (int j=0;j<v[i].size();j++) modify(1,v[i][j].first,v[i][j].second);
		ans[seq[i]]=t[1].tag;
	}
	for (int i=1;i<=n;i++) printf("%d\n",ans[i]);
	return 0;
}