警钟撅烂(如果你样例全输出0)
查看原帖
警钟撅烂(如果你样例全输出0)
300166
Zikl楼主2023/10/1 23:50

因为自己淋过雨,所以总想替别人撑把伞 qwq

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<queue>
#include<cmath>
const int maxn=1e6+10;
using namespace std;
int n,m,M,cnt;
int head[maxn<<1],ver[maxn<<1],Next[maxn<<1],ans[maxn],tot;
int rt[maxn];
struct node1{
	int ls,rs,id,val;
}t[maxn*64];
struct node{
	int x,y,z;
} a[maxn];
void add(int x,int y){
	ver[++tot]=y,Next[tot]=head[x],head[x]=tot;
}
int dep[maxn],siz[maxn],top[maxn],son[maxn],fa[maxn];
void dfs1(int u,int f){
	dep[u]=dep[f]+1;
	siz[u]=1;
	fa[u]=f;
	for(int i=head[u];i;i=Next[i]){
		int y=ver[i];
		if(y==f) continue;
		dfs1(y,u);
		siz[u]+=siz[y];
		if(siz[y]>siz[son[u]])
		son[u]=y;
	}
}
void dfs2(int u,int topf){
	top[u]=topf;
	if(!son[u]) return;
	dfs2(son[u],topf);
	for(int i=head[u];i;i=Next[i]){
		int y=ver[i];
		if(y==fa[u]||y==son[u]) continue;
		dfs2(y,y);
	}
}
int LCA(int u,int v){
	while(top[u]!=top[v]){
		if(dep[top[u]]<dep[top[v]])
		swap(u,v);	
		u=fa[top[u]];
	}
	return dep[u]<dep[v]?u:v;
}
void pushup(int k){
	if(t[t[k].ls].val>=t[t[k].rs].val){
		t[k].val=t[t[k].ls].val;
		t[k].id=t[t[k].ls].id;
	}
	else{
		t[k].val=t[t[k].rs].val;
		t[k].id=t[t[k].rs].id;
	}
}//如果有多种救济粮都是存放最多的,输出种类编号最小的一种
int update(int k,int l,int r,int pos,int v){
	if(!k)k=++cnt;
	if(l==r){
		t[k].val+=v;
		t[k].id=l;
		return k;
	}
	int mid=(l+r)>>1;
	if(pos<=mid)
	t[k].ls=update(t[k].ls,l,mid,pos,v);
	else
	t[k].rs=update(t[k].rs,mid+1,r,pos,v);
	pushup(k);
	return k;
}
int merge(int a,int b,int l,int r){
	if(!a||!b) 	return a+b;
	if(l==r){
		t[a].val+=t[b].val;
		t[a].id=l;
		return a;
	}
	int mid=(l+r)>>1;
	t[a].ls=merge(t[a].ls,t[b].ls,l,mid);
	t[a].rs=merge(t[a].rs,t[b].rs,mid+1,r);
	pushup(a);
	return a;
}
void dfs(int u){
	for(int i=head[u];i;i=Next[i]){
		int y=ver[i];
		if(y==fa[u]) continue;
		dfs(y);
		rt[u]=merge(rt[u],rt[y],1,M);
	}
	if(t[rt[u]].val)
	ans[u]=t[rt[u]].id;
}
signed main(){
	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;
		add(a,b);
		add(b,a);
	}
	dfs1(1,0);
	dfs2(1,1);
	/*
	每个点开一颗权值线段树表示这个点上有什么数字,
	每个数字出现了几次,
	然后求点i的权值线段树就是将它所有孩子的线段树合并到一起,
	然后再对合并出来的线段树进行一些在这个节点的插入和删除操作。 
	*/
	for(int i=1;i<=m;i++){
		cin>>a[i].x>>a[i].y>>a[i].z;
		M=max(M,a[i].z);
	}
	for(int i=1;i<=m;i++){
		int u=a[i].x,v=a[i].y,w=a[i].z;
		int lca=LCA(u,v);
		rt[u]=update(rt[u],1,M,w,1);
		rt[v]=update(rt[v],1,M,w,1);
		rt[lca]=update(rt[lca],1,M,w,-1);
		if(fa[lca])
		rt[fa[lca]]=update(rt[fa[lca]],1,M,w,-1);
		/*而不是 
		update(rt[u],1,M,w,1);
		update(rt[v],1,M,w,1);
		update(rt[lca],1,M,w,-1);
		if(fa[lca])
		update(rt[fa[lca]],1,M,w,-1);
		为什么没报错我就不清楚了。 
		*/
	}
	dfs(1);
	for(int i=1;i<=n;i++)
	cout<<ans[i]<<endl;
	return 0;
} 
2023/10/1 23:50
加载中...