hdu 5029 Relief grain 樹鍊剖分

給一棵樹

離線操作；

a  b  z; 在a到b路徑上的點加，種類為 z 的糧食；

輸出：每個點的糧食個數最多的種類數。//詳見測試用例

樹鍊剖分後，線段樹為糧食的種類，在用标記數組标記下段的加值，搞搞……

#pragma comment(linker, "/STACK:200000000")
#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<stdlib.h>
#include<iostream>
#include<vector>
using namespace std;
#define ll __int64
const ll N=100010;

vector<int> plu[100010];
vector<int> sub[100010];
struct Edge
{
	int to,next;
}edge[N*2];
int head[N],tot;
int top[N];//top[v]表示v所在的重鍊的頂端節點
int fa[N]; //父親節點
int deep[N];//深度
int num[N];//num[v]表示以v為根的子樹的節點數
int p[N];//p[v]表示v與其父親節點的連邊線上段樹中的位置
int fp[N];//和p數組相反
int son[N];//重兒子
int pos;
void init()
{
	tot=0;
	memset(head,-1,sizeof(head));
	pos=1;
	memset(son,-1,sizeof(son));
}
void addedge(int u,int v)
{
	edge[tot].to=v; edge[tot].next=head[u]; head[u]=tot++;
}
void dfs1(int u,int pre,int d)//第一遍dfs求出fa，deep，num，son
{
	deep[u]=d;
	fa[u]=pre;
	num[u]=1;
	for(int i=head[u];i!=-1;i=edge[i].next)
	{
		int v=edge[i].to;
		if(v!=pre)
		{
			dfs1(v,u,d+1);
			num[u]+=num[v];
			if(son[u]==-1 || num[v]>num[son[u]])
				son[u]=v;
		}
	}
}
void getpos(int u,int sp)//第二遍dfs求出top和p
{
	top[u]=sp;
	p[u]=pos++;
	fp[p[u]]=u;
	if(son[u]==-1) return ;
	getpos(son[u],sp);
	for(int i=head[u];i!=-1;i=edge[i].next)
	{
		int v=edge[i].to;
		if(v!=son[u] && v!=fa[u])
			getpos(v,v);
	}
}

void change(int u,int v,int val)//查詢u->v邊的最大值
{
	int f1=top[u],f2=top[v];
	while(f1!=f2)
	{
		if(deep[f1]<deep[f2])
		{
			swap(f1,f2);
			swap(u,v);
		}
		plu[p[f1]].push_back(val);//p[f1],p[u]
		sub[p[u]+1].push_back(val);
		u=fa[f1]; f1=top[u];
	}
	if(deep[u]>deep[v]) swap(u,v);//p[u],p[v]
	plu[p[u]].push_back(val);
	sub[p[v]+1].push_back(val);
}
struct node
{
	int l,r,mx,ID;
	int mid()
	{
		return (l+r)>>1;
	}
}tree[4*N];

void build(int id,int x,int y)
{
	tree[id].l=x;
	tree[id].r=y;
	tree[id].mx=0;
	if(x==y)
	{
		tree[id].ID=x;
		return ;
	}
	int mid=(x+y)>>1;
	build(id<<1,x,mid);
	build(id<<1|1,mid+1,y);
}
void pushup(int id)
{
	if(tree[id<<1].mx>=tree[id<<1|1].mx)
	{
		tree[id].mx=tree[id<<1].mx;
		tree[id].ID=tree[id<<1].ID;
	}
	else
	{
		tree[id].mx=tree[id<<1|1].mx;
		tree[id].ID=tree[id<<1|1].ID;
	}
}
void update(int id,int x,int val)
{
	if(tree[id].l==x && tree[id].r==x)
	{
		tree[id].mx+=val;
		return ;
	}
	int mid=(tree[id].l+tree[id].r)>>1;
	if(x<=mid)
		update(id<<1,x,val);
	else
		update(id<<1|1,x,val);
	pushup(id);
}
int ans[N];
int main()
{
	int n,m,i,j,k;
	while(scanf("%d%d",&n,&m))
	{
		if(!n && !m) break;
		init();
		for(i=1;i<n;i++)
		{
			int u,v;
			scanf("%d%d",&u,&v);
			addedge(u,v);
			addedge(v,u);
		}
		dfs1(1,0,0);
		getpos(1,1);
		for(i=1;i<=100001;i++)
		{
			plu[i].clear();
			sub[i].clear();
		}
		while(m--)
		{
			int u,v,z;
			scanf("%d%d%d",&u,&v,&z);
			change(u,v,z);
		}
		build(1,1,100000);
		for(i=1;i<=n;i++)
		{
			for(j=0;j<plu[i].size();j++)
				update(1,plu[i][j],1);
			for(j=0;j<sub[i].size();j++)
				update(1,sub[i][j],-1);
			if(tree[1].mx<=0)
				ans[fp[i]]=0;
			else
				ans[fp[i]]=tree[1].ID;
		}
		for(i=1;i<=n;i++)
			printf("%d\n",ans[i]);
	}
}