记录编号 606585 评测结果 AAAAAAAAAAAAAAAAAAAA
题目名称 3380.雨天的尾巴 最终得分 100
用户昵称 Gravatar李奇文 是否通过 通过
代码语言 C++ 运行时间 4.703 s
提交时间 2025-09-28 20:58:01 内存使用 41.34 MiB
显示代码纯文本
#include<bits/stdc++.h>
#define mid ((l+r)>>1)
using namespace std;
const int N=1e5+5;
int n,m;
vector<int> e[N];
int f[N][33],dep[N];
int root[N],tot,ans[N];
int ls[N*50],rs[N*50],sum[N*50],typ[N*50];
bool vis[N];
void dfs(int u,int fa){
	f[u][0]=fa;
	dep[u]=dep[fa]+1;
	for(int i=1;i<=18;i++){
		f[u][i]=f[f[u][i-1]][i-1];
	}
	for(int v:e[u]){
		if(v!=fa) dfs(v,u);
 	}
}
int lca(int u,int v){
	if(dep[u]<dep[v]) swap(u,v);
	for(int i=18;~i;i--){
		if(dep[f[u][i]]>=dep[v]) u=f[u][i];
	}
	if(v==u) return v;
	for(int i=18;~i;i--){
		if(f[u][i]!=f[v][i]){
			u=f[u][i];
			v=f[v][i];
		}
	}
	return f[u][0];
}
void pushup(int u){
	if(sum[ls[u]]>=sum[rs[u]]){
		sum[u]=sum[ls[u]];
		typ[u]=typ[ls[u]];
	}else{
		sum[u]=sum[rs[u]];
		typ[u]=typ[rs[u]];
	}
}
int merge(int x,int y,int l,int r){
	if(!x||!y) return x+y;
	if(l==r){
		sum[x]+=sum[y];
		return x;
	}
	ls[x]=merge(ls[x],ls[y],l,mid);
	rs[x]=merge(rs[x],rs[y],mid+1,r);
	pushup(x);
	return x;
}
void change(int &u,int l,int r,int p,int k){
	if(!u) u=++tot;
	if(l==r){ sum[u]+=k,typ[u]=p;return;}
	if(p<=mid) change(ls[u],l,mid,p,k);
	else change(rs[u],mid+1,r,p,k);
	pushup(u);
}
void dfs2(int x,int f){
	for(int y:e[x]){
		if(y==f) continue;
		dfs2(y,x);
		root[x]=merge(root[x],root[y],1,N);
	}
	ans[x]=(sum[root[x]]?typ[root[x]]:0);
}
int main(){
	freopen("Rainbows.in","r",stdin);
	freopen("Rainbows.out","w",stdout);
	ios::sync_with_stdio(0);
	cin.tie(0);cout.tie(0);
	cin>>n>>m;
	for(int i=1;i<n;i++){
		int a,b;
		cin>>a>>b;
		e[a].push_back(b);
		e[b].push_back(a);
	}
	dfs(1,0);
	for(int i=1;i<=m;i++){
		int x,y,z;
		cin>>x>>y>>z;
		change(root[x],1,N,z,1);
		change(root[y],1,N,z,1);
		int _lca=lca(x,y);
		change(root[_lca],1,N,z,-1);
		change(root[f[_lca][0]],1,N,z,-1);
	}
	dfs2(1,0);
	for(int i=1;i<=n;i++){
		cout<<ans[i]<<"\n";
	}
	return 0;
}