您的位置:首页 > 其它

主席树解决区间第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;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: