救救孩子
查看原帖
救救孩子
279800
ExplodingKonjac楼主2021/8/29 11:29

rt,线段树很模板题写挂了,75pts,一直找不到错

#include <bits/stdc++.h>
using namespace std;

namespace FastIO
{
/*
这里是一堆80多行的快读快些模板
即后面使用的qin、qout
不用想了,绝对不是这里的问题
*/
}
using namespace FastIO;
int n,m,nn,b[100005],ans[100005];
struct Edge{ int to,nxt; }e[200005];
int cnt,head[200005];
inline void addEdge(int u,int v)
{
	e[++cnt]=(Edge){v,head[u]};
	head[u]=cnt;
}
struct TreeNode
{
	int id,max;
	int lc,rc;
}t[8000005];
int rt[100005];
#define LC t[i].lc
#define RC t[i].rc
inline void pushup(int i)
{
	if(t[LC].max<t[RC].max)
		t[i].max=t[RC].max,t[i].id=t[RC].id;
	else
		t[i].max=t[LC].max,t[i].id=t[LC].id;
}
void modify(int p,int x,int &i,int l=1,int r=nn)
{
	if(!i)	i=++cnt,t[i].max=0;
	if(l==r)	t[i].max+=x,t[i].id=l;
	else
	{
		int mid=(l+r)>>1;
		if(mid>=p)	modify(p,x,LC,l,mid);
		else	modify(p,x,RC,mid+1,r);
		pushup(i);
	}
}
void merge(int &dst,int &src,int l=1,int r=nn)
{
	if(!src)	return;
	if(!dst)	dst=src,src=0;
	else if(l==r)
		t[dst].max+=t[src].max,t[dst].id=l;
	else
	{
		int mid=(l+r)>>1;
		merge(t[dst].lc,t[src].lc,l,mid);
		merge(t[dst].rc,t[src].rc,mid+1,r);
		pushup(dst),src=0;
	}
}
int f[100005][20],dep[100005];
void dfs1(int u,int fa)
{
	f[u][0]=fa,dep[u]=dep[fa]+1;
	for(int i=1;(1<<i)<=dep[u];i++)
		f[u][i]=f[f[u][i-1]][i-1];
	for(int i=head[u];i;i=e[i].nxt)
		if(e[i].to!=fa)
			dfs1(e[i].to,u);
}
int lca(int x,int y)
{
	if(dep[x]<dep[y])	swap(x,y);
	for(int i=17;i>=0;i--)
		if(dep[f[x][i]]>=dep[y])
			x=f[x][i];
	if(x==y)	return x;
	for(int i=17;i>=0;i--)
		if(f[x][i]!=f[y][i])
			x=f[x][i],y=f[y][i];
	return f[x][0];
}
void dfs2(int u,int fa)
{
	for(int i=head[u];i;i=e[i].nxt)
		if(e[i].to!=fa)
			dfs2(e[i].to,u),merge(rt[u],rt[e[i].to]);
	if(t[rt[u]].max)	ans[u]=b[t[rt[u]].id];
}
struct Oper{ int u,v,c; }a[100005];
int main()
{
//	freopen("P4556_5.in","r",stdin);
//	freopen("P4556_5.ans","w",stdout);
	qin>>n>>m;
	for(int i=1;i<n;i++)
	{
		int u,v;
		qin>>u>>v;
		addEdge(u,v),addEdge(v,u);
	}
	for(int i=1;i<=m;i++)
		qin>>a[i].u>>a[i].v>>a[i].c,b[i]=a[i].c;
	sort(b+1,b+n+1),nn=unique(b+1,b+n+1)-b-1;
	dfs1(1,0),cnt=0;
	for(int i=1;i<=m;i++)
	{
		int p=lower_bound(b+1,b+nn+1,a[i].c)-b,x=a[i].u,y=a[i].v,l;
		l=lca(x,y);
		modify(p,1,rt[x]),modify(p,1,rt[y]);
		modify(p,-1,rt[l]);
		if(f[l][0])	modify(p,-1,rt[f[l][0]]);
	}
	dfs2(1,0);
	for(int i=1;i<=n;i++)	qout.writeln(ans[i]);
	return qout.flush(),0;
}
2021/8/29 11:29
加载中...