主席树解决区间第k大
2017-06-03 17:07
591 查看
【填坑】可持久化线段树解决无修改的区间k大问题 - FAreStorm的博客 - 博客频道 - CSDN.NET
http://blog.csdn.net/farestorm/article/details/50337527
看上面这个学会的
我的理解是,一共有n个数,理想的方法是依次创建n颗权值线段树,第i颗线段树代表1到i的各个区间内存在的数的个数,那么,求解区间[s,t],便可通过第t颗减去第s-1颗得到,类似于前缀和,第k大的数二分区间得到,(别忘了线段树的结点定义),然而这样建树太浪费空间,应为有些区间在加入一个数后,并没有改变,所以,动态开点,保存下各个节点左右子树所对应的编号,最后为了进一步压缩空间,离散化原来的值。
例题
K-th Number POJ - 2104
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=5000000;
int ls[maxn]; //左儿子的编号
int rs[maxn]; //右儿子的编号
int num[maxn]; //有两次变换,第一次为原序列的值,第二次在lower_bound之后,为离散化后的值,也就是下标
int san[maxn]; //排序后原序列的值
int sum[maxn]; //线段树里保存的值
int T[maxn]; //第i颗线段树的编号
int n,m,tot = 0;
void build(int l,int r,int &x){
x=++tot;
sum[x]=0;
if(l==r) return;
int mid=(l+r)/2;
build(l,mid,ls[x]);
build(mid+1,r,rs[x]);
}
void update(int last,int p,int l,int r,int &x){
x=++tot;
ls[x]=ls[last];
rs[x]=rs[last];
sum[x]=sum[last]+1;
if(l==r) return;
int mid=(l+r)/2;
if(p<=mid) update(ls[last],p,l,mid,ls[x]);
else update(rs[last],p,mid+1,r,rs[x]);
}
int query(int s,int t,int l,int r,int k){
if(l==r) return l;
int mid=(l+r)/2;
int cnt=sum[ls[t]]-sum[ls[s]];
if(k<=cnt) return query(ls[s],ls[t],l,mid,k);
else return query(rs[s],rs[t],mid+1,r,k-cnt);
}
int main()
{
/* ios_base::sync_with_stdio(0);
cin.tie(0);*/
int n,m;
while(scanf("%d %d",&n,&m)!=EOF){
tot=0;
for(int i=0;i<n;i++){
scanf("%d",&san[i]);
num[i]=san[i];
}
sort(san,san+n);
int cnt=unique(san,san+n)-san;
build(1,cnt,T[0]);
for(int i=0;i<n;i++){
num[i]=lower_bound(san,san+cnt,num[i])-san+1;
update(T[i],num[i],1,cnt,T[i+1]);
}
while(m--){
int s,t,k;
scanf("%d %d %d",&s,&t,&k);
printf("%d\n",san[query(T[s-1],T[t],1,cnt,k)-1]);
}
}
return 0;
}
http://blog.csdn.net/farestorm/article/details/50337527
看上面这个学会的
我的理解是,一共有n个数,理想的方法是依次创建n颗权值线段树,第i颗线段树代表1到i的各个区间内存在的数的个数,那么,求解区间[s,t],便可通过第t颗减去第s-1颗得到,类似于前缀和,第k大的数二分区间得到,(别忘了线段树的结点定义),然而这样建树太浪费空间,应为有些区间在加入一个数后,并没有改变,所以,动态开点,保存下各个节点左右子树所对应的编号,最后为了进一步压缩空间,离散化原来的值。
例题
K-th Number POJ - 2104
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=5000000;
int ls[maxn]; //左儿子的编号
int rs[maxn]; //右儿子的编号
int num[maxn]; //有两次变换,第一次为原序列的值,第二次在lower_bound之后,为离散化后的值,也就是下标
int san[maxn]; //排序后原序列的值
int sum[maxn]; //线段树里保存的值
int T[maxn]; //第i颗线段树的编号
int n,m,tot = 0;
void build(int l,int r,int &x){
x=++tot;
sum[x]=0;
if(l==r) return;
int mid=(l+r)/2;
build(l,mid,ls[x]);
build(mid+1,r,rs[x]);
}
void update(int last,int p,int l,int r,int &x){
x=++tot;
ls[x]=ls[last];
rs[x]=rs[last];
sum[x]=sum[last]+1;
if(l==r) return;
int mid=(l+r)/2;
if(p<=mid) update(ls[last],p,l,mid,ls[x]);
else update(rs[last],p,mid+1,r,rs[x]);
}
int query(int s,int t,int l,int r,int k){
if(l==r) return l;
int mid=(l+r)/2;
int cnt=sum[ls[t]]-sum[ls[s]];
if(k<=cnt) return query(ls[s],ls[t],l,mid,k);
else return query(rs[s],rs[t],mid+1,r,k-cnt);
}
int main()
{
/* ios_base::sync_with_stdio(0);
cin.tie(0);*/
int n,m;
while(scanf("%d %d",&n,&m)!=EOF){
tot=0;
for(int i=0;i<n;i++){
scanf("%d",&san[i]);
num[i]=san[i];
}
sort(san,san+n);
int cnt=unique(san,san+n)-san;
build(1,cnt,T[0]);
for(int i=0;i<n;i++){
num[i]=lower_bound(san,san+cnt,num[i])-san+1;
update(T[i],num[i],1,cnt,T[i+1]);
}
while(m--){
int s,t,k;
scanf("%d %d %d",&s,&t,&k);
printf("%d\n",san[query(T[s-1],T[t],1,cnt,k)-1]);
}
}
return 0;
}
相关文章推荐
- HDU2665 主席树原理解决静态区间第K大值问题总结 有详细图解和代码解释
- poj 2104 区间第K小 主席树入门题
- 【主席树】51Nod 1175 区间中第K大的数
- POJ2104 POJ2761 区间第K大 主席树
- POJ2104-K-th Number-区间第k大-可持久化线段树/主席树
- 求第k大连续区间和/第k大子序列和 - 二分+树状数组+前缀和(或主席树+堆)
- zoj-2112(主席树动态求区间第k小数)
- 【poj2104-求区间第k大数(不修改)】主席树/可持续化线段树
- hdu 2665 区间第K大 主席树入门
- ZOJ 2112 & BZOJ 1901 Dynamic Rankings(主席树 单点更新 区间第K大)
- 区间第K值——主席树详解
- 【BZOJ】1901: Zju2112 Dynamic Rankings(区间第k小+树状数组套主席树)
- [poj 2104 K-th Number] 主席树 区间第K大
- poj2761&&poj2104 主席树(静态区间第K大)
- POJ - 2104 K-th Number && POJ 2761 Feed the dogs(主席树静态区间第k大)
- 动态区间第K大(树状数组+主席树)
- 主席树(静态区间第k大)
- 可持久化线段树|主席树 POJ 2104 区间第k大的数
- poj 2104 区间第k大 主席树
- bzoj 1901 Zju2112 Dynamic Rankings(动态区间第k大,主席树)