权值线段树(+动态开点)及其简单应用

它是个啥?

我们都知道,线段树是一种功能较为强大的数据结构。普通线段树中我们在每个节点存储的是某段区间的一些信息,而权值线段树,我们存的是值为该下标的数的个数,并通过线段树统计某段区间的信息。

它能干嘛?

最最简单的用途就是动态统计在某个取值范围内的数的个数嘛,而这就足够我们解决许多问题了,例如求前驱、后继等等。

它有什么限制?

通常出题人都不会特地去限制权值的范围,而如果我们非常草率地开出一棵线段树所有的节点, 99.99 99.99 99.99%都是会 M L E MLE MLE的。
那么我们就需要用一个叫动态开点线段树的东西来防止这种悲剧发生了。
因为一段区间在线段树上最多被划分为 O ( l o g 2 n ) O(log_2n) O(log2n)个区间,而询问+修改的数量 m m m也一定在时间复杂度范围内,所以动态开点线段树的时空复杂度都是 O ( m l o g 2 n ) O(mlog_2^n) O(mlog2n)的,在绝大多数情况下都是可以接收的。(除了某些丧心病狂的出题人出一些臭名昭著的卡log的题)

它怎么实现?

首先,在座的各位大佬肯定都会线段树的嘛(^ _ ^)
动态开点的话,我们只需要将每个节点的左右儿子也记录到该节点的信息中即可。
每次修改时访问到未开出的节点时就开出这个点,询问因为该区间本身就未被更新过,直接返回原始状态即可,这可以在一定程度上减少时间复杂度。(然而并不影响上界复杂度)

它有什么代替价值吗?

权值线段树可以解决一些平衡树的问题。当然,如果你想用 C + + C++ C++ S T L STL STL s e t set set大法当然也是可以的啦(然而本蒟蒻并不会用) (P党勿喷,并且各种建议noip赛后转c)

它的代码实现长咋样?

本人一直使用的都是感觉较为易理解的普通 (DK) 线段树,听说 (事实上) 常数有点大,但出题人如此善良 (吗?), 应该不会故意卡线段树的吧 (怎么不信呢?)
这是本人普通平衡树的代码,略丑

#include<cstdio>
using namespace std;
const int L = -1e7 + 5;//尽量多开一点,主要懒得写区间不存在的特判
const int R = 1e7 + 5;
int rt , cnt;
struct inm{int l_son , r_son , cnt;};
int read()
{
	char ch = getchar();
	bool flag = 1;
	while(ch < '0' || ch > '9') flag &= ch != '-' , ch = getchar();
	int res = 0;
	while(ch >= '0' && ch <= '9') res = (res << 3) + (res << 1) + (ch ^ 48) , ch = getchar();
	return flag ? res : -res;
}
struct ST
{
	private:
		inm tree[3000005];
	public:
		void push_up(int p)
		{
			tree[p].cnt = tree[tree[p].l_son].cnt + tree[tree[p].r_son].cnt;
		}
		void add(int &p , int l , int r , int x , int k)
		{
			if(!p) p = ++cnt;
			if(l == r)
			{
				tree[p].cnt += k;
				return;
			}
			int mid = (l + r) >> 1;
			if(x <= mid) add(tree[p].l_son , l , mid , x , k);
			else add(tree[p].r_son , mid + 1 , r , x , k);
			push_up(p);
		}
		int query_cnt(int p , int l , int r , int x , int y)//求l~r区间中数的个数
		{
			if(!p) return 0;
			if(l == x && r == y) return tree[p].cnt;
			int mid = (l + r) >> 1;
			if(y <= mid) return query_cnt(tree[p].l_son , l , mid , x , y);
			else if(x > mid) return query_cnt(tree[p].r_son , mid + 1 , r , x , y);
			else return query_cnt(tree[p].l_son , l , mid , x , mid) + query_cnt(tree[p].r_son , mid + 1 , r , mid + 1 , y);
		}
		int query_number(int p , int l , int r , int k)//求排名为k的数
		{
			if(l == r) return l;
			int mid = (l + r) >> 1;
			if(tree[tree[p].l_son].cnt >= k) return query_number(tree[p].l_son , l , mid , k);
			else return query_number(tree[p].r_son , mid + 1 , r , k - tree[tree[p].l_son].cnt);
		}
}st;
int query_rank(int k)//求k的排名,在线段树上二分
{
	return st.query_cnt(rt , L , R , L , k - 1) + 1;
}
int query_prev(int k)//求k的前驱
{
	int rank = st.query_cnt(rt , L , R , L , k - 1);
	return st.query_number(rt , L , R , rank);
}
int query_next(int k)//求k的后继
{
	int rank = st.query_cnt(rt , L , R , L , k) + 1;
	return st.query_number(rt , L , R , rank);
}
int main()
{
	int q = read();
	while(q--)
	{
		int op = read() , k = read();
		if(op == 1) st.add(rt , L , R , k , 1);
		if(op == 2) st.add(rt , L , R , k , -1);
		if(op == 3) printf("%d\n",query_rank(k));
		if(op == 4) printf("%d\n",st.query_number(rt , L , R , k));
		if(op == 5) printf("%d\n",query_prev(k));
		if(op == 6) printf("%d\n",query_next(k));
	}
	return 0;
}
  • 1
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值