洛谷4556 BZOJ3307 [Vani有约会]雨天的尾巴 线段树合并 差分 LCA

题目链接

题意：
首先村落里的一共有$n$座房屋，并形成一个树状结构。然后救济粮分$m$次发放，每次选择两个房屋$(x,y)$，然后对于$x$到$y$的路径上（含$x$和$y$)每座房子里发放一袋z类型的救济粮。然后问当所有的救济粮发放完毕后，每座房子里存放的最多的是哪种救济粮。如果有多种数量相同的输出编号最小的。$n,m,z<=1e5$。

题解：
还是挺好想的吧。这个题思路挺简单的，但是调试的时候快把我搞傻了。

思路很不难想。首先你要对于每一个点维护$z$种权值出现的次数，并支持链加，这个东西要是你去考虑用LCT或者树剖，感觉其实可能不太能维护的样子吧，就算能应该也不是很好写的吧。那么我们就去尝试用线段树合并，我们发现线段树合并的时候开的权值线段树正好很适合维护每一个值的出现次数，也很方便维护最大值出现的位置。那么我们考虑修改的话怎么维护，其实链上的修改除了LCT和树剖之外还有一个常用的方法是树上差分。我们在两个端点处打上+1的标记，在LCA处和LCA的父节点处打上-1的标记，这样从下向上线段树合并就可以了。

但是代码写起来发现有各种奇怪的问题。首先是这个题空间很紧，开大了就MLE，开小了就RE，我反复了好几次。最后的解决方法是我不记录最大值出现的位置，而是去在线段树上查询。再就是注意一些奇怪的细节，比如在合并的时候即使是$l=r$的叶子节点在更新时也要新建节点。再或者说查询位置的时候判断一下是不是最大出现次数就是$0$，别不小心在这种情况输出一个线段树的左端点$1$。

代码：

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

int n,m,hed[100010],cnt,f[100010][19];
int root[200010],dep[100010];
const int qwq=1e5;
struct node
{
	int to,next; 
}a[200010];
struct tree
{
	int l,r,mx;
}tr[9400010];
inline int read()
{
	int x=0;
	char s=getchar();
	while(s>'9'||s<'0')
	s=getchar();
	while(s>='0'&&s<='9')
	{
		x=x*10+s-'0';
		s=getchar();
	}
	return x;
}
inline void add(int from,int to)
{
	a[++cnt].to=to;
	a[cnt].next=hed[from];
	hed[from]=cnt;
}
inline void dfs(int x)
{
	for(int i=1;i<=18;++i)
	f[x][i]=f[f[x][i-1]][i-1];
	for(int i=hed[x];i;i=a[i].next)
	{
		int y=a[i].to;
		if(y==f[x][0])
		continue;
		f[y][0]=x;
		dep[y]=dep[x]+1;
		dfs(y);
	}
}
inline int lca(int x,int y)
{
	if(dep[x]<dep[y])
	swap(x,y);
	for(int i=18;i>=0;--i)
	{
		if(dep[f[x][i]]>=dep[y])
		x=f[x][i];
	}
	if(x==y)
	return x;
	for(int i=18;i>=0;--i)
	{
		if(f[x][i]!=f[y][i]&&dep[f[x][i]])
		{
			x=f[x][i];
			y=f[y][i];
		}
	}
	return f[x][0];
}
inline void update(int &rt,int l,int r,int x,int y)
{
	if(!rt)
	{
		rt=++cnt;
		tr[rt].mx=0;
	}
	if(l==r)
	{
		tr[rt].mx+=y;
		return;
	}
	int mid=(l+r)>>1;
	if(x<=mid)
	update(tr[rt].l,l,mid,x,y);
	else
	update(tr[rt].r,mid+1,r,x,y);
	tr[rt].mx=max(tr[tr[rt].l].mx,tr[tr[rt].r].mx);
}
inline void merge(int &rt,int x,int y,int l,int r)
{
	if(!x||!y)
	{
		rt=x+y;
		return;
	}	
	rt=++cnt;
	if(l==r)
	{
		tr[rt].mx=tr[x].mx+tr[y].mx;
		return;
	}
	int mid=(l+r)>>1;	
	tr[rt].l=tr[x].l;
	tr[rt].r=tr[x].r;
	merge(tr[rt].l,tr[x].l,tr[y].l,l,mid);
	merge(tr[rt].r,tr[x].r,tr[y].r,mid+1,r);
	tr[rt].mx=max(tr[tr[rt].l].mx,tr[tr[rt].r].mx);
}
inline void dfs1(int x)
{
	for(int i=hed[x];i;i=a[i].next)
	{
		int y=a[i].to;
		if(y==f[x][0])
		continue;
		dfs1(y);
		merge(root[x],root[x],root[y],1,qwq);
	}
}
inline int query(int rt,int l,int r)
{
	if(!rt)
	return 0;
	if(l==r)
	{
		if(tr[rt].mx==0)
		return 0;
		else
		return l;
	}	
	int mid=(l+r)>>1;
	if(tr[tr[rt].l].mx>=tr[tr[rt].r].mx)
	return query(tr[rt].l,l,mid);
	else
	return query(tr[rt].r,mid+1,r);
}
int main()
{
	n=read();
	m=read();
	for(int i=1;i<=n-1;++i)
	{
		int x=read(),y=read();
		add(x,y);
		add(y,x);
	}
	dep[1]=1;
	dfs(1);
	cnt=0;
	for(int i=1;i<=m;++i)
	{
		int x=read(),y=read(),z=read();
		int ji=lca(x,y);
		update(root[x],1,qwq,z,1);
		update(root[y],1,qwq,z,1);
		update(root[ji],1,qwq,z,-1);
		if(f[ji][0])
		update(root[f[ji][0]],1,qwq,z,-1);
	}
	dfs1(1);
	for(int i=1;i<=n;++i)
	printf("%d\n",query(root[i],1,qwq));
	return 0;
}