P3369 【模板】普通平衡树(树状数组)

一、树状数组

树状数组(Binary Indexed Tree(B.I.T), Fenwick Tree)是一个查询和修改复杂度都为log(n)的数据结构。

这张图总是让很多初学者望而生畏(好吧只是我)

所以在学习它之前,我们来看看线段树。

(现在我默认大家都会线段树)

我们知道如果\(a + b = c\),则\(b = c - a\)。

所以,所有节点的右儿子都是不需要的。

我们把线段树上不必要的节点去掉。

它长得会像这样。

这种数据结构我们称它为树状数组。可以发现所有线段的右断点都互不相同,所以我们把它按右端点重新编号。

可以发现一些性质:

  1. 节点\(p\)的父亲即为\(p + lowbit(p)\)。(\(lowbit(x) = x and -x\))
  2. 节点\(p\)的线段长度为\(lowbit(p)\)

故我们可以写出给一个数加\(x\)的代码。即顺着边依次更行它的祖先。

void update(int x, int y) {
for (int i = x; i <= n; i += lowbit(i)) c[i] += y;
}

我们还需要查询\([l, r]\)的和,即为\([1, r]\)的和 \(-\) \([1, l - 1]\)的和。

下面有一个求\([1, x]\)的和的代码。

void query(int x) {
int ret = 0;
for (int i = x; i; i -= lowbit(i)) ret += c[i];
return ret;
}

我们再根据树状数组的图可以发现其实就是对x进行二进制拆分。

求出每一段的和。

二、这道题的解释

我们可以考虑类似计数的方法。即如果\(x\)比较小,我们可以用\(num[x]\)表示\(x\)出现的次数。所以查找排名即为查询比\(x\)小的数的\(num\)和。

三、Kth()

我们考虑在树状数组上进行类似倍增的操作。

int _kth(int k) {
int ret = 0, sum = 0;
for (int i = 20; i >= 0; --i)
if (ret + (1 << i) <= lcnt && sum + c[ret + (1 << i)] < k) {
sum += c[ret + (1 << i)];
ret += 1 << i;
}
for (int i = 0; i <= 20; ++i)
if (sum + c[ret + (1 << i)] >= k) {
ret += 1 << i;
break;
}
return ret;
}

简单地说,就是先跳大的,再跳最小一步使刚好大于等于k。

四、代码

#include <stdio.h>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <iostream>
#include <algorithm>
using namespace std;
const int MAXN = 100005;
int n, opt[MAXN], num[MAXN], lcnt, lsh[MAXN];
class BinaryIndexTree {
private:
int c[MAXN];
int _kth(int k) {
int ret = 0, sum = 0;
for (int i = 20; i >= 0; --i)
if (ret + (1 << i) <= lcnt && sum + c[ret + (1 << i)] < k) {
sum += c[ret + (1 << i)];
ret += 1 << i;
}
for (int i = 0; i <= 20; ++i)
if (sum + c[ret + (1 << i)] >= k) {
ret += 1 << i;
break;
}
return ret;
}
void _insert(int x) {
for (int i = x; i <= n; i += i & -i) ++c[i];
}
void _erase(int x) {
for (int i = x; i <= n; i += i & -i) --c[i];
}
int num(int x) {
int ret = 0;
for (int i = x; i; i -= i & -i) ret += c[i];
return ret;
}
public:
int kth(int k) { return _kth(k); }
void insert(int x) { _insert(x); }
void erase(int x) { _erase(x); }
int rank(int x) { return num(x - 1) + 1; }
int pre(int x) { return _kth(num(x - 1)); }
int suc(int x) { return _kth(num(x) + 1); }
} bitree;
int main() {
scanf("%d", &n);
for (int i = 1; i <= n; ++i) {
scanf("%d%d", &opt[i], &num[i]);
if (opt[i] != 4) lsh[++lcnt] = num[i];
}
sort(lsh + 1, lsh + lcnt + 1);
lcnt = unique(lsh + 1, lsh + lcnt + 1) - lsh - 1;
for (int i = 1; i <= n; ++i) {
if (opt[i] == 4) {
printf("%d\n", lsh[bitree.kth(num[i])]);
} else {
int x = lower_bound(lsh + 1, lsh + lcnt + 1, num[i]) - lsh;
if (opt[i] == 1) bitree.insert(x);
else if (opt[i] == 2) bitree.erase(x);
else if (opt[i] == 3) printf("%d\n", bitree.rank(x));
else if (opt[i] == 5) printf("%d\n", lsh[bitree.pre(x)]);
else if (opt[i] == 6) printf("%d\n", lsh[bitree.suc(x)]);
}
}
return 0;
}

【Luogu】P3369 【模板】普通平衡树(树状数组)的更多相关文章

  1. bzoj3196 二逼平衡树 树状数组套线段树

    题目传送门 思路:树状数组套线段树模板题. 什么是树状数组套线段树,普通的树状数组每个点都是一个权值,而这里的树状数组每个点都是一颗权值线段树,我们用前缀差分的方法求得每个区间的各种信息, 其实关键就 ...

  2. 【bzoj3224】【Tyvj 1728】 普通平衡树 树状数组

    您需要写一种数据结构(可参考题目标题),来维护一些数,其中需要提供以下操作:1. 插入$x$数2. 删除$x$数(若有多个相同的数,因只删除一个)3. 查询$x$数的排名(若有多个相同的数,因输出最小 ...

  3. [luogu]P2657低头一族[树状数组]

    [luogu]P2657 低头一族 题目描述 一群青年人排成一队,用手机互相聊天. 每个人的手机有一个信号接收指标,第i个人的接收指标设为v[i]. 如果位置在x[i]的人要和位置在xj的人聊天,那么 ...

  4. BZOJ 3196 Tyvj 1730 二逼平衡树 ——树状数组套主席树

    [题目分析] 听说是树套树.(雾) 怒写树状数组套主席树,然后就Rank1了.23333 单点修改,区间查询+k大数查询=树状数组套主席树. [代码] #include <cstdio> ...

  5. luogu3380/bzoj3196 二逼平衡树 (树状数组套权值线段树)

    带修改区间K大值 这题有很多做法,我的做法是树状数组套权值线段树,修改查询的时候都是按着树状数组的规则找出那log(n)个线段树根,然后一起往下做 时空都是$O(nlog^2n)$的(如果离散化了的话 ...

  6. Luogu P4901 排队 fib数列+树状数组+倍增

    这题让我升华..还好只重构了一遍 首先我们发现:$n$较小时,整个队伍的形态 跟 $n$ 比较大时的局部是一样的 所以我们预处理出这个队伍的形态,和每一行每个位置的质因子个数的前缀和,$O(nlogn ...

  7. 【Luogu】P2617Dynamic Ranking(树状数组套主席树)

    题目链接 树状数组套主席树有点难懂qwq 不好理解 树状数组套主席树的直观理解应该是:树状数组的每一个节点是一棵主席树. 普通区间修改我们是创建1个线段树,树状数组套主席树的时候我们就创建log个线段 ...

  8. luogu P4769 [NOI2018]冒泡排序 结论 树状数组 卡特兰数

    LINK:冒泡排序 神题. 可以想到爆搜 期望得分5~10分. 打成这个样子心态不得爆炸? 仔细分析 一个不合法序列还有什么标志. 容易想到某个数字离自己位置相反的方向多走了一步. 考虑单独对每个数字 ...

  9. luogu P2345 奶牛集会 |排序+树状数组

    题目描述 约翰的N 头奶牛每年都会参加"哞哞大会".哞哞大会是奶牛界的盛事.集会上的活动很多,比如堆干草,跨栅栏,摸牛仔的屁股等等.它们参加活动时会聚在一起,第i 头奶牛的坐标为X ...

  10. BZOJ 1107: [POI2007]驾驶考试egz / Luogu P3463 [POI2007]EGZ-Driving Exam (树状数组 LIS)

    能从iii走到所有跑道 相当于 能从iii走到111和nnn. 边反向后就相当于 能从111和nnn走到iii. 为了方便叙述,把111~nnn叫做x坐标,111~(m+1)(m+1)(m+1)叫做y ...

随机推荐

  1. linux 字符驱动框架(用户态的read,write,poll是怎么操作驱动的)

    前言 这篇文章是通过对一个简单字符设备驱动的操作来解释,用户态的读写操作是怎么映射到具体设备的. 因为针对不同版本的linux内核,驱动的接口函数一直有变化,这贴出我测试的系统信息: root@ubu ...

  2. Spring中同一个service中方法相互调用事务不生效问题解决方案

    问题描述: 我们在用Spring框架开发Web项目过程中,经常需要用同一个service中的一个方法调用另一个方法,如果此时调用方没有添加事务注解@Transactional,而在被调用方添加事务注解 ...

  3. day55:django:cookie&session

    目录 1.Cookie 1.Cookie前戏 2.Cookie的引入 3.django中操作cookie 2.Session 1.cookie的局限性 2.session技术 3.django操作se ...

  4. 【转】Locust-工具核心原理分析

    Locust工具在市场上不如Loadrunner / JMeter流行,使用的范围也没有那么广,但不可否认其是一款很不错的工具.我个人觉得Locust使用不是那么广泛,主要是因为一下方式: Locus ...

  5. springboot:This application has no explicit mapping for /erro

    springboot启动没有报错,但是访问的时候返回如上图的错误.看报错内容感觉是没有这个mapping对应的接口.但是确实写了. 最终发现是因为springboot的启动类放的位置不对.启动类所在的 ...

  6. OSI和TCP/IP参考模型

    分层思想: 分层模型是一种开发网络协议的设计方法. 把节点之间的通讯这个复杂的问题,分成了若干个简单的小问题逐一解决. 把网络相邻节点之间通过接口进行通信,下层为上层提供服务.当网络发生故障,很容易确 ...

  7. MyEclipse中的项目导入到Eclipse中运行的错误解决

    之前用的myEclipse,后来把项目导入eclipse发现报错,将MyEclipse中的项目导入到Eclipse中运行,不注意一些细节,会造成无法运行的后果.下面就说说具体操作:导入后出现如下错误: ...

  8. 坚果云+svn实现异地非局域网个人代码版本管理

    原理大概是A地的设备作为服务端创建仓库,将仓库传上坚果云,同步到B地,再拉取仓库的代码

  9. 正交矩阵(Orthogonal Matrix)

  10. 为cmd中的命令添加别名,以解决java:错误: 编码 GBK 的不可映射字符 (0xAF)

    使用sublineText3编写了java代码,通过cmd javac编译 提示 错误:编码GBK的不可映射字符 解决方法 使用javac -encoding UTF-8 Person.java 结果 ...