主席树求区间[L,R]内小于等于x的个数【模板】

文章目录

题目链接:

hdu 4417 Super Mario
牛客小白月赛9 E

hdu4417

/*主席树求[L,R]内小于等于x的个数*/
#include"bits/stdc++.h"
using namespace std;
typedef long long LL;
const int maxn=2e6+5;
int Ls[maxn],Rs[maxn],tree[maxn];//保存左右儿子
int a[maxn],b[maxn];//a是离散化后的,b是原数组 
vector<int>root;
int N,Q,n,tot;
int Build(int L,int R)
{
	int id=++tot;
	tree[id]=0;
	if(L==R)return id;
	int mid=L+R>>1;
	Ls[id]=Build(L,mid);
	Rs[id]=Build(mid+1,R);
	return id;
}
void Add(int id1,int x)
{
	int id2=++tot;//新建一个root
	root.push_back(id2);
	tree[id2]=tree[id1]+1;
	int L=1,R=n;
	while(L<R)
	{

		int mid=L+R>>1;
		if(x<=mid)
		{
			R=mid;
			Ls[id2]=++tot;
			Rs[id2]=Rs[id1];
			id2=tot;
			id1=Ls[id1];
		}
		else
		{
			L=mid+1;
			Rs[id2]=++tot;
			Ls[id2]=Ls[id1];
			id2=tot;
			id1=Rs[id1];
		}
		tree[id2]=tree[id1]+1;
	}
}
int query(int id,int L,int R,int x)
{
	if(x>=a[R])return tree[id];
	else if(x<a[L])return 0;
	int res=0;
	int mid=L+R>>1;
	if(x<=a[mid])res+=query(Ls[id],L,mid,x);//不要写成x<=mid了 
	else
	{
		res+=tree[Ls[id]];//左边的直接加上去 
		res+=query(Rs[id],mid+1,R,x);
	}
	return res;
}
int main()
{
	int T;
	cin>>T; 
	for(int Case=1;Case<=T;Case++)
	{
		cin>>N>>Q;
		tot=0;
		memset(tree,0,sizeof tree);
		root.clear();
		for(int i=1; i<=N; i++)
		{
			scanf("%d",b+i);
			a[i]=b[i];
		}
		cout<<"Case "<<Case<<":"<<endl;
		sort(a+1,a+1+N);
		n=unique(a+1,a+1+N)-(a+1);
		root.push_back(Build(1,n));
		for(int i=1; i<=N; i++)
		{
			int pos=lower_bound(a+1,a+1+n,b[i])-a;//找这个数是在离散化后的哪个位置 
			Add(root[i-1],pos);
		}
		while(Q--)
		{
			int L,R,x;
			scanf("%d%d%d",&L,&R,&x);
			L++,R++;
			cout<<query(root[R],1,n,x)-query(root[L-1],1,n,x)<<endl;
		}
	}
}

全部评论

相关推荐

政委qqq:这道题在算法竞赛里唯一考的就是高精度,但是只能难住C++这类语言,Python直接a+b秒天秒地
点赞 评论 收藏
分享
个人背景:学院二本计科专业&nbsp;大二开始实习个人经历:安克创新&nbsp;、理想汽车、字节跳动碎碎念:我做事只有三分钟热度。看到进了大厂的同学,我会羡慕,也会跟着努力上进;但遇到好看的小说,我又会放下手头的事沉迷其中,之前的坚持也就中断了。我有些自卑,总觉得自己学历和外貌都不够好。之前偶然在网上受到关注,我就喜欢上了上网,因为这里有很多人认可我。但我也很在意别人的评价,偶尔看到嘲讽的言论,会触发我的自卑情绪,让我感到愤怒。有时候我会强硬地回怼,有时候又会懦弱地选择无视。我也有虚荣心。不管是拿到安克、理想还是字节的机会,我在分享的时候都会带着这份心思。我会特意强调自己学历不好,是为了衬托出过程的艰难,以此显得自己更厉害。我知道,人往往会炫耀自己缺少的东西,来掩盖内心的空洞。我总想着走捷径,不太喜欢踏踏实实地做事。找实习的时候,我花了更多时间在研究面试技巧上,而不是提升专业能力。我会反复听面试录音分析技巧,看面试教程学习怎么和不同的面试官沟通,还会每天自言自语练习语言表达,同学都觉得我有点奇怪。我的实习生涯里,侥幸和运气占了很大一部分。我总在想,如果有一天我失去了这份幸运,这些特质可能会让我一蹶不振。ps:&nbsp;很多人会问我学习路线和经验&nbsp;但是就像我上面说的&nbsp;我的实习过程靠的很多是关键节点的运气&nbsp;技术上面我可能不如很多人&nbsp;&nbsp;所以请大家理性求助和理性参考我的回答&nbsp;附上我的投递记录
我的offer在哪里...:从去年看到现在,飞升哥就是榜样
我的求职进度条
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务