【线段树】[LUOGU CF600E Lomsat gelral] 区间众数和 [线段树合并]

题目:

题目链接:[LUOGU CF600E Lomsat gelral]
题解:

这个题就是个(动态开点)权值线段树的板子题,就是在每个点上建立一个取值线段树,记录区间众数和,不断是合并权值线段树就行,在up里面就进行对于众数的判断即可。

这里再说一下权值线段树,之前说过,这就是个桶,但是桶只能查找到权值等于 x x x这个值的个数,但是权值线段树可以查找到权值在 x x x y y y之间的的个数。

代码:

#include<bits/stdc++.h>
#define int long long
using namespace std;
inline int read()
{
	int s=0,w=1;char ch=getchar();
	while(ch>'9'||ch<'0'){if(ch=='-')w=-1;ch=getchar();}
	while(ch<='9'&&ch>='0')s=s*10+ch-'0',ch=getchar();
	return s*w;
}
const int sea=1e5+7;
struct see{int ver,next;}e[sea<<1];
struct hit{int ls,rs,sum,max,ans;}tr[sea*40];//max就是现在的众数,sum就是该点出现的次数,ans就是答案 
int n,tot,seg,last[sea<<1],root[sea],col[sea],res[sea];
void add(int x,int y){e[++tot].ver=y;e[tot].next=last[x],last[x]=tot;}
void up(int k)
{
	int l=tr[k].ls,r=tr[k].rs;
	if(tr[l].sum>tr[r].sum) tr[k].max=tr[l].max,tr[k].sum=tr[l].sum,tr[k].ans=tr[l].ans;
	if(tr[l].sum<tr[r].sum)	tr[k].max=tr[r].max,tr[k].sum=tr[r].sum,tr[k].ans=tr[r].ans;
	if(tr[l].sum==tr[r].sum) tr[k].max=tr[l].max,tr[k].sum=tr[l].sum,tr[k].ans=tr[l].ans+tr[r].ans;
	//如果在序列中的两个数出席现次数相同的话,那ans就进行累加更新答案 
}
void build(int &k,int l,int r,int val,int tt)
{
	if(!k) k=++seg;
	if(l==r) {tr[k].max=l,tr[k].sum+=tt,tr[k].ans=l; return ;}
	//出现次数初值就是1,然后累加 
	int mid=(l+r)/2;
	if(val<=mid) build(tr[k].ls,l,mid,val,tt);
	else build(tr[k].rs,mid+1,r,val,tt);
	up(k);
}
int incor(int x,int y,int l,int r)
{
	if(!x) return y; if(!y) return x;
	if(l==r) {tr[x].max=l,tr[x].sum+=tr[y].sum,tr[x].ans=l; return x;}
	//在由于是权值线段树,所以l,r,就可以表示这个点的权值了,所以这里众数和答案就是l ,出现次数相加就是。 
	int mid=(l+r)/2; 
	tr[x].ls=incor(tr[x].ls,tr[y].ls,l,mid);//这样合并就是权值在一段区间之间的个数
	tr[x].rs=incor(tr[x].rs,tr[y].rs,mid+1,r);
	up(x);	return x;
}
void dfs(int x,int fa)
{
	for(int i=last[x];i;i=e[i].next)
	{
		int y=e[i].ver;	if(y==fa) continue; 
		dfs(y,x); incor(root[x],root[y],1,sea);
	}	
	build(root[x],1,sea,col[x],1);
	//一个小小的trick是在最早一次插入之前将所有点的根先编号,这样子可能即使写翻了也有概率A掉(因为root[x]=x)
	res[x]=tr[root[x]].ans;
}
signed main()
{
	n=read(); 
	for(int i=1;i<=n;i++) root[i]=i,col[i]=read();   	
	for(int i=1,x,y;i<n;i++) x=read(),y=read(),add(x,y),add(y,x); 
	//因为开了单边这个WA在了第25个点,存两条边吧,毕竟是记录的是每个点为根是的结果 
	seg=n; //这是权值线段树标号,所以要和n之前的树进行区分所以是在n之后 
	dfs(1,0);
	for(int i=1;i<=n;i++) printf("%lld ",res[i]); 
	return 0;
}

Continue……

  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值