Eyjafjalla

本文介绍了如何使用主席树(Segment Tree)简化线段树中的区间查询操作,通过构建树结构并利用递归思想高效地统计指定区间内元素的大小关系。通过实例展示了如何在温度数组和离散化数组上进行操作,以及如何在LCA查询基础上进行区间内的特定条件计数。
摘要由CSDN通过智能技术生成

区间查询有关比大小的数目,

主席树感觉学线段树的时候不用x<<1,x<<1|1去建一次树那样就容易理解多了;

#include<cstdio>
#include<iostream>
#include<cstring>
#include<vector>
#include<queue>
#include<algorithm>
#include<cmath>
using namespace std;
typedef long long ll;
const int N=1e5+10;
int h[N],ne[N<<1],e[N<<1],idx=0;
void add(int a,int b){e[idx]=b,ne[idx]=h[a],h[a]=idx++;return ;}
int fa[N][32],v[N],b[N],s[N],n,siz[N],id[N],bs,cnt=0;
// fa倍增求最高点,v是温度数组,b是离散化数组,bs是b数组中不重复数字的个数,s是dfs序数组
//siz存每颗子树的节点数目,id存该节点再dfs序中的位置; 
void dfs(int u,int f)
{
	fa[u][0]=f;
	for(int i=1;i<=30;i++)fa[u][i]=fa[fa[u][i-1]][i-1];
	siz[u]=1;id[u]=++cnt;s[cnt]=v[u];
	for(int i=h[u];~i;i=ne[i])
	{
		int j=e[i];
		if(j==f)continue;
		dfs(j,u);
		siz[u]+=siz[j];
	}
}
struct po{int l,r,sum;}tr[N<<5];
int root[N<<5];/ 
void update(int &x,int y,int l,int r,int num)
{
	x=++cnt; //新建一个节点,让x指向这个节点; 
	tr[x]=tr[y];//节点赋值 
	tr[x].sum++; 
	int mid=l+r>>1;
	if(l==r)return ;
	if(mid>=num)update(tr[x].l,tr[y].l,l,mid,num); 
	else update(tr[x].r,tr[y].r,mid+1,r,num);
}

int quary(int u,int v,int l,int r,int L,int R)
{
	int x=tr[u].sum-tr[v].sum,ans=0;	
	if(R>=r&&L<=l)return x;
	int mid=l+r>>1;
	if(L<=mid)ans=quary(tr[u].l,tr[v].l,l,mid,L,R);
	if(R>mid)ans+=quary(tr[u].r,tr[v].r,mid+1,r,L,R);
	return ans;
}

int LCA(int u,int r)//LCA是因为起不动名字了。。。; 
{
	for(int i=30;i>=0;i--)if(v[fa[u][i]]<=r)u=fa[u][i];
	return u;
}

int se(int u){return lower_bound(b+1,b+bs+1,u)-b;}

int main()
{
	int q;
	memset(h,-1,sizeof h);
	scanf("%d",&n);
	for(int i=1,u,v;i<n;i++)scanf("%d%d",&u,&v),add(u,v),add(v,u);
	for(int i=1;i<=n;i++)scanf("%d",v+i),b[i]=v[i];
	scanf("%d",&q);
	v[0]=0x3f3f3f3f;//这个比较重要;;; 
	dfs(1,0);
	sort(b+1,b+n+1);
	bs=unique(b+1,b+1+n)-b-1;
	cnt=0;
	for(int i=1;i<=n;i++)
		update(root[i],root[i-1],1,bs,se(s[i]));
	while(q--)
	{
		int x,l,r;
		scanf("%d%d%d",&x,&l,&r);
		if(v[x]<l||v[x]>r){
			printf("%d\n",0);
			continue;
		}
		x=LCA(x,r);
		printf("%d\n",quary(root[id[x]+siz[x]-1],root[id[x]-1],1,bs,se(l),se(s[id[x]])));
	}
	return 0;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

李昌荣。

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值