【42.38%】【BZOJ 3196】二逼平衡树
Time Limit: 10 Sec Memory Limit: 128 MB
Submit: 1363 Solved: 579
[Submit][Status][Discuss]
Description
您需要写一种数据结构(可参考题目标题),来维护一个有序数列,其中需要提供以下操作:
1.查询k在区间内的排名
2.查询区间内排名为k的值
3.修改某一位值上的数值
4.查询k在区间内的前驱(前驱定义为小于x,且最大的数)
5.查询k在区间内的后继(后继定义为大于x,且最小的数)
Input
第一行两个数 n,m 表示长度为n的有序序列和m个操作
第二行有n个数,表示有序序列
下面有m行,opt表示操作标号
若opt=1 则为操作1,之后有三个数l,r,k 表示查询k在区间[l,r]的排名
若opt=2 则为操作2,之后有三个数l,r,k 表示查询区间[l,r]内排名为k的数
若opt=3 则为操作3,之后有两个数pos,k 表示将pos位置的数修改为k
若opt=4 则为操作4,之后有三个数l,r,k 表示查询区间[l,r]内k的前驱
若opt=5 则为操作5,之后有三个数l,r,k 表示查询区间[l,r]内k的后继
Output
对于操作1,2,4,5各输出一行,表示查询结果
Sample Input
9 6
4 2 2 1 9 4 0 1 1
2 1 4 3
3 4 10
2 1 4 3
1 2 5 9
4 3 9 5
5 2 8 5
Sample Output
2
4
3
4
9
HINT
1.n和m的数据范围:n,m<=50000
2.序列中每个数的数据范围:[0,1e8]
3.虽然原题没有,但事实上5操作的k可能为负数
【题解】
线段树套一个平衡树。
对于线段树的每个节点所代表的区间维护一个平衡树(包含了这个区间里面的所有元素);
用的是SBT(这可能是这份题解唯一的卖点了,我看网上其他人用的都是treap,200+行左右,然而我用了300+行)。
找第k小的数的时候用到了二分答案。
修改操作可以看成是在平衡树里面删掉原来的值。然后再把新的值加入到平衡树中。
这道题的rank是这样的
1 2 3 3 3 5 6 7 8 9
3的rank是3
但是如果查rank4也是3,rank5 也是3,rank3也是3
删除操作有点复杂,因为左右儿子都有的情况需要找个节点来替代,而且这个节点的重复次数也需要改变(同一个数字记录在同一个节点,用”重复”域来表示这个数字重复了几次;
还有,因为同一数字压缩在了同一个节点。所以左旋、右旋代码的size域的修改有所改变
/**************************************************************
Problem: 3196
User: chengchunyang
Language: C++
Result: Accepted
Time:7420 ms
Memory:65572 kb
****************************************************************/
#include <cstdio>
const int MAXN = 51000;
const int INF = 1e8 + 10;
//记住要乘16倍就好。
//有MAX*4个是存放树的根节点
//MAX*4*15个是这些树的儿子节点
int root[MAXN * 4], l[MAXN * 4 * 16], r[MAXN * 4 * 16], key[MAXN * 4 * 16], size[MAXN * 4 * 16], cnt = 0;
int a[MAXN], n, m, rank = 0, cf[MAXN * 4 * 16];//cf代表这个数字重复出现了几次
int qianqu, houji;
int min(int a, int b)
{
return a > b ? b : a;
}
int max(int a, int b)
{
return a > b ? a : b;
}
void r_ro(int &t)//右旋代码
{
int k = l[t];
l[t] = r[k];
r[k] = t;
size[k] = size[t];
size[t] = size[l[t]] + size[r[t]] + cf[t];//最后加上的不是1而是cf[t]了。
t = k;
}
void l_ro(int &t)//左旋
{
int k = r[t];
r[t] = l[k];
l[k] = t;
size[k] = size[t];
size[t] = size[l[t]] + size[r[t]] + cf[t];
t = k;
}
void maintain(int &t, bool flag)//sbt的修复函数
{
if (flag)
{
if (size[l[l[t]]] > size[r[t]])
r_ro(t);
else
if (size[r[l[t]]] > size[r[t]])
l_ro(l[t]), r_ro(t);
else
return;
}
else
{
if (size[r[r[t]]] > size[l[t]])
l_ro(t);
else
if (size[l[r[t]]] > size[l[t]])
r_ro(r[t]), l_ro(t);
else
return;
}
maintain(l[t], true);
maintain(r[t], false);
maintain(t, true);
maintain(t, false);
}
void insert(int &t, int x)//插入过程
{
if (!t)
{
t = ++cnt;
size[t] = cf[t] = 1;
l[t] = r[t] = 0;
key[t] = x;
}
else
{
size[t]++;
if (x == key[t])
cf[t]++;
else
if (x < key[t])
insert(l[t], x);
else
insert(r[t], x);
if (x != key[t])
maintain(t, x < key[t]);
}
}
void ask_rank(int t, int x)//获取排名,获取的是比它小的数字的个数
{
if (!t)
return;
if (x == key[t])//等于就即加上左子树的大小
rank += size[l[t]];
else
if (x < key[t])
ask_rank(l[t], x);
else
if (x > key[t])//大于
{
rank += size[l[t]] + cf[t];//整个左子树和这个节点都行
ask_rank(r[t], x);
}
}
void get_rank(int rt, int begin, int end, int l, int r, int x)
{//线段树,获取这个l,r区间所在的平衡树
if (begin == l && end == r)
{
ask_rank(root[rt], x);
return;
}
int m = (begin + end) >> 1;
if (r <= m)
get_rank(rt << 1, begin, m, l, r, x);
else
if (m < l)
get_rank(rt << 1 | 1, m + 1, end, l, r, x);
else
{
get_rank(rt << 1, begin, m, l, m, x);
get_rank(rt << 1 | 1, m + 1, end, m + 1, r, x);
}
}
void build(int rt, int begin, int end, int pos, int x)
{//建线段树,并维护平衡树
insert(root[rt], x);
if (begin == end)
return;
int m = (begin + end) >> 1;
if (pos <= m)
build(rt << 1, begin, m, pos, x);
else
build(rt << 1 | 1, m + 1, end, pos, x);
}
void de_lete(int &t, int x)
{//删掉x这个元素
if (!t)
return;
size[t]--;
if (key[t] == x)
{
if (cf[t] > 1)
cf[t]--;
else
{
cf[t] = 0;
bool flag1 = (l[t]>0), flag2 = (r[t]>0);
if (!flag1 && !flag2)
t = 0;
else
if (!flag1 && flag2)
t = r[t];
else
if (flag1 && !flag2)
t = l[t];
else
{
int temp = r[t];
while (l[temp]) temp = l[temp];//这里要用一个节点替代它
int dd = cf[temp] - 1;
temp = r[t];
size[temp] -= dd;//相应的size域都要发生变化
while (l[temp])
{
temp = l[temp];
size[temp] -= dd;
}
key[t] = key[temp];//替代
cf[t] = cf[temp];//那个节点的cf域变成1
cf[temp] = 1;
de_lete(r[t], key[temp]);//下次就会删掉了。
}
}
}
else
if (x < key[t])
de_lete(l[t], x);
else
de_lete(r[t], x);
}
void change(int rt, int begin, int end, int pos, int x)
{//更改值实际上就是在平衡树中删掉一个元素,然后再添加进去
de_lete(root[rt], a[pos]);
insert(root[rt], x);
if (begin == end)
return;
int m = (begin + end) >> 1;
if (pos <= m)
change(rt << 1, begin, m, pos, x);
else
change(rt << 1 | 1, m + 1, end, pos, x);
}
void ask_before(int t, int x)
{//询问前趋是啥
if (!t)
return;
if (key[t] < x)
{
qianqu = max(qianqu, key[t]);
ask_before(r[t], x);
}
else
if (x <= key[t])
ask_before(l[t], x);
}
void get_before(int rt, int begin, int end, int l, int r, int k)
{//获取这个区间所在的平衡树
if (begin == l && end == r)
{
ask_before(root[rt], k);
return;
}
int m = (begin + end) >> 1;
if (r <= m)
get_before(rt << 1, begin, m, l, r, k);
else
if (m < l)
get_before(rt << 1 | 1, m + 1, end, l, r, k);
else
{
get_before(rt << 1, begin, m, l, m, k);
get_before(rt << 1 | 1, m + 1, end, m + 1, r, k);
}
}
void ask_houji(int t, int x)
{//询问后继
if (!t)
return;
if (x < key[t])
{
houji = min(houji, key[t]);
ask_houji(l[t], x);
}
else
if (x >= key[t])
ask_houji(r[t], x);
}
void get_houji(int rt, int begin, int end, int l, int r, int k)
{//获取这个区间所在的平衡树
if (begin == l && end == r)
{
ask_houji(root[rt], k);
return;
}
int m = (begin + end) >> 1;
if (r <= m)
get_houji(rt << 1, begin, m, l, r, k);
else
if (m < l)
get_houji(rt << 1 | 1, m + 1, end, l, r, k);
else
{
get_houji(rt << 1, begin, m, l, m, k);
get_houji(rt << 1 | 1, m + 1, end, m + 1, r, k);
}
}
int main()
{
//freopen("F:\\rush.txt", "r", stdin);
scanf("%d%d", &n, &m);
for (int i = 1; i <= n; i++)
{
scanf("%d", &a[i]);
build(1, 1, n, i, a[i]);
}
for (int i = 1; i <= m; i++)
{
int opt;
scanf("%d", &opt);
if (opt == 1)
{
int l, r, k;
scanf("%d%d%d", &l, &r, &k);
rank = 1;
get_rank(1, 1, n, l, r, k);
printf("%d\n", rank);
}
else
if (opt == 2)
{
int l, r, k;
scanf("%d%d%d", &l, &r, &k);
int le = 0, ri = INF, num;//询问第k小数实际上是二分一个数字,然后判断它的排名
while (le <= ri)//不断调整
{
int m = (le + ri) >> 1;
rank = 1;//我们找的是小于这个数的个数.所以一开始加上1就是这个数的排名了。
get_rank(1, 1, n, l, r, m);
if (rank <= k)//小于等于k就记录这个m为答案。
{//最后记录的是满足排名小于等于k的最大的一个数字.
le = m + 1;
num = m;
}
else
if (rank > k)
ri = m - 1;
}
printf("%d\n", num);
}
else
if (opt == 3)
{
int pos, x;
scanf("%d%d", &pos, &x);
change(1, 1, n, pos, x);
a[pos] = x;
}
else
if (opt == 4)
{
int l, r, k;
scanf("%d%d%d", &l, &r, &k);
qianqu = 0;
get_before(1, 1, n, l, r, k);
printf("%d\n", qianqu);
}
else
if (opt == 5)
{
int l, r, k;
scanf("%d%d%d", &l, &r, &k);
houji = INF;
get_houji(1, 1, n, l, r, k);
printf("%d\n", houji);
}
}
return 0;
}
【42.38%】【BZOJ 3196】二逼平衡树的更多相关文章
- BZOJ 3196 二逼平衡树
Description 您需要写一种数据结构(可参考题目标题),来维护一个有序数列,其中需要提供以下操作:1.查询k在区间内的排名2.查询区间内排名为k的值3.修改某一位值上的数值4.查询k在区间内的 ...
- bzoj 3196二逼平衡树 线段树套平衡树
比较裸的树套树,对于区间K值bz上有一道裸题,详见题解http://www.cnblogs.com/BLADEVIL/p/3455336.html(其实题解也不是很详细) //By BLADEVIL ...
- BZOJ 3196 二逼平衡树 ——树套树
[题目分析] 全靠运气,卡空间. xjb试几次就过了. [代码] #include <cmath> #include <cstdio> #include <cstring ...
- bzoj 3196 Tyvj 1730 二逼平衡树(线段树套名次树)
3196: Tyvj 1730 二逼平衡树 Time Limit: 10 Sec Memory Limit: 128 MBSubmit: 1807 Solved: 772[Submit][Stat ...
- BZOJ 3196: Tyvj 1730 二逼平衡树( 树套树 )
这道题做法应该很多吧.... 我用了线段树套treap.... -------------------------------------------------------------------- ...
- bzoj 3196 && luogu 3380 JoyOI 1730 二逼平衡树 (线段树套Treap)
链接:https://www.lydsy.com/JudgeOnline/problem.php?id=3196 题面; 3196: Tyvj 1730 二逼平衡树 Time Limit: 10 Se ...
- bzoj 3196/ Tyvj 1730 二逼平衡树 (线段树套平衡树)
3196: Tyvj 1730 二逼平衡树 Time Limit: 10 Sec Memory Limit: 128 MB[Submit][Status][Discuss] Description ...
- bzoj 3196: Tyvj 1730 二逼平衡树
#include<cstdio> #include<ctime> #include<cstdlib> #include<iostream> #defin ...
- 【BZOJ 3196】二逼平衡树 线段树套splay 模板题
我写的是线段树套splay,网上很多人写的都是套treap,然而本蒟蒻并不会treap 奉上sth神犇的模板: //bzoj3196 二逼平衡树,支持修改某个点的值,查询区间第k小值,查询区间某个值排 ...
- 【BZOJ-3196】二逼平衡树 线段树 + Splay (线段树套平衡树)
3196: Tyvj 1730 二逼平衡树 Time Limit: 10 Sec Memory Limit: 128 MBSubmit: 2271 Solved: 935[Submit][Stat ...
随机推荐
- Qt 5.11的QChar、QString、QTextBoundaryFinder和双向文本算法现在完全兼容Unicode 10
本文翻译自:Qt 5.11 released 原文作者: Qt公司CTO兼Qt开源项目维护官Lars Knoll翻译校审:Richard.Hongfei.Haipeng 5月22日,我们提发布了Qt ...
- WIN8.1 上安装 debian8.7 遇到的问题及解决方法
WIN8.1 上安装 debian8.7 遇到的问题及解决方法 参照百度经验 <win7下硬盘安装debian7> ( http://jingyan.baidu.com/article/8 ...
- 35.Node.js GET/POST请求
转自:http://www.runoob.com/nodejs/nodejs-module-system.html 在很多场景中,我们的服务器都需要跟用户的浏览器打交道,如表单提交. 表单提交到服务器 ...
- 很好的资源 for android
//texttospeach http://examples.javacodegeeks.com/android/core/text-to-speech/android-text-to-speech- ...
- Large Division (大数求余)
Given two integers, a and b, you should check whether a is divisible by b or not. We know that an in ...
- Android平台中的三种翻页效果机器实现原理
本文给开发者集中展现了Android平台中的三种翻页效果机器实现原理,希望能够对开发者有实际的帮助价值! 第一种翻页效果如下: 实现原理: 当前手指触摸点为a,则 a点坐标为(ax,ay), ...
- 13.Zookeeper的java客户端API使用方法
转自:https://blog.csdn.net/jiuqiyuliang/article/details/56012027
- 81.内存模式实现cgi查询
创建全局的二级指针 char ** g_pp;//全局的二级指针 获取数据有多少行 //获取行数 int getimax() { ; FILE *pf = fopen(path, "r&qu ...
- Flask项目之手机端租房网站的实战开发(五)
说明:该篇博客是博主一字一码编写的,实属不易,请尊重原创,谢谢大家! 接着上一篇博客继续往下写 :https://blog.csdn.net/qq_41782425/article/details/8 ...
- 【前端图表】echarts实现散点图x轴时间轴
话不多说,老规矩,先上图,实现echarts实现散点图,x轴数据为时间年月日. 实现代码如下: <!DOCTYPE html> <html> <head> < ...