这篇是我写的归并树的博客,也是用的poj2104 归并树
题目:poj 2104
大意:给你一个长度为n的数列,每次查询数列下标为[i,j]区间里面升序排序后第k大的数
主席树在速度上比归并树快得多,但相对的内存占用会比较大,据我的计算,主席树的数组会比归并树多用大概nlogn+n个(但毕竟内存不值钱)
对于这道题,用归并树:
用主席树:(原谅我的渣渣代码能力)
对于主席树网上的讲解很多,推荐一个:https://blog.csdn.net/woshinannan741/article/details/53012682
我再对于我自己学习过程中的一些关键点做一个补充
主席树是解决区间内第k个数的有力武器。
主席树其实就是一堆线段树,它做法就是先建一棵空的线段树,此时线段树里面存的是大小在【l,r】内的数字个数,一开始都是0
(就拿 1到4 这个区间来说)
然后根据给的数列,比如 3 1 2 4,那么先将第一个数放进线段树里面
(红色数字是区间存的值)
那么这就是第二课线段树,然后继续操作,建第三棵树
到最后:
(1~4那里写错了,应该是4的orz。。。)
然后假如 我要找第二到第四区间内第2大(升序排列)的数,我就可以找到第四颗树和第一棵树
(这并不是最终做法)
然鹅,然鹅,实际操作中,还会存在很多问题,比如:你建这么多线段树,内存会爆炸(虽然说它不值钱),还有实际区间会存在大量重复的数字,上面这么找是找不到滴,最后一个是现实中数字之间差距可能会很大(数据范围1~1e9),不好开这么大的线段树。
先来解决第一个问题,那就是共用线段树的问题,这个上面发那个博客的链接里也有
就是对于一棵线段树,它的区别相对于上一棵树来说只是多塞进去一个数,那么对于其他没有塞进去的树枝,就可以利用上一颗树的指针等于这一颗树的指针(数组下标),来重复利用,然后对于每一棵树,用一个root数组记录其树根,图就不做了,上面博客有。
那么我就来实战♂讲解一下
struct node{
int l,r,sum;
}tree[maxn*40];
void build_tree(int l,int r,int &t,int l_t,int aim){
tree[++ind]=tree[l_t];
t=ind;
tree[ind].sum++;
if(l==r) return ;
int mid=(l+r)>>1;
if(aim<=mid) build_tree(l,mid,tree[ind].l,tree[l_t].l,aim);
if(aim>mid) build_tree(mid+1,r,tree[ind].r,tree[l_t].r,aim);
}
for(int i=1;i<=n;i++) build_tree(1,n,root[i],root[i-1],get_id(a[i]));
l,r是左右区间, &t就是这次要建立的树,l_t (last_tree) 就是上一棵树,aim 就是要塞进去的那个数字,ind(index)是建树要用的(你懂得♂)
首先直接将上一颗树复制一遍给这棵树,然后因为这棵树多塞进去一个数字,所以 计数 ++, 然后再根据塞的那个数字找到左右子树哪一棵不一样,重复动作
第二个问题,实际中会有多个重复的数字(类似的我在归并树里面讲过)
比如再 1 3 3 3 4 ,里面找第k个,k=3。
你想,找4 的话,返回数就大于k,找3 的话,返回数会大于或者等于k,找2的话,返回数一定小于k
那么只需要找大于等于k 的数中最小的那个就可以了(还需看着程序细细体会。。。)
第三个问题,解决方法就是离散化,具体做法就是
将所有出现的数存在一个vector里面,排序+去重。当你要对一个数离散化的时候,在vector里面find那个数字,将其下标作为其离散化后的数来用,用完之后凭着这个下标就可以在vector里面找回来那个数勒。。。
int get_id(int num){
return lower_bound(V.begin(),V.end(),num)-V.begin()+1;//a 数列从1开始就要加
}
(最后关于主席树要开多少数组的话,我看别人是开40倍的,我自己没算233)
最后是AC代码
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<vector>
using namespace std;
const int maxn=100005;
int a[maxn],root[maxn],ind;
vector<int> V;
struct node{
int l,r,sum;
}tree[maxn*40];
int get_id(int num){
return lower_bound(V.begin(),V.end(),num)-V.begin()+1;//a 数列从1开始就要加
}
void build_tree(int l,int r,int &t,int l_t,int aim){
tree[++ind]=tree[l_t];
t=ind;
tree[ind].sum++;
if(l==r) return ;
int mid=(l+r)>>1;
if(aim<=mid) build_tree(l,mid,tree[ind].l,tree[l_t].l,aim);
if(aim>mid) build_tree(mid+1,r,tree[ind].r,tree[l_t].r,aim);
}
int query(int l,int r,int l_t,int t,int aim){
if(l==r) return l;
int mid=(l+r)>>1;
int res=tree[ tree[t].l ].sum-tree[ tree[l_t].l ].sum;
if(res>=aim) query(l,mid,tree[l_t].l,tree[t].l,aim);
else query(mid+1,r,tree[l_t].r,tree[t].r,aim-res);
}
int main(){
int n,q; cin>>n>>q;
for(int i=1;i<=n;i++){
scanf("%d",&a[i]);
V.push_back(a[i]);
}
sort(V.begin(),V.end());
V.erase(unique(V.begin(),V.end()),V.end()); //去重得编号
for(int i=1;i<=n;i++) build_tree(1,n,root[i],root[i-1],get_id(a[i]));
while(q--){
int fro,to,k; scanf("%d %d %d",&fro,&to,&k);
printf("%d\n",V[query(1,n,root[fro-1],root[to],k)-1]); //getid +1 这里减回去
}
return 0;
}