【模板】可持久化線段樹 1(主席樹) 洛谷p3834
阿新 • • 發佈:2018-11-02
題目背景
這是個非常經典的主席樹入門題——靜態區間第K小
資料已經過加強,請使用主席樹。同時請注意常數優化
題目描述
如題,給定N個正整數構成的序列,將對於指定的閉區間查詢其區間內的第K小值。
輸入輸出格式
輸入格式:
第一行包含兩個正整數N、M,分別表示序列的長度和查詢的個數。
第二行包含N個正整數,表示這個序列各項的數字。
接下來M行每行包含三個整數l, r, kl,r,k , 表示查詢區間[l, r][l,r]內的第k小值。
輸出格式:
輸出包含k行,每行1個正整數,依次表示每一次查詢的結果
輸入輸出樣例
輸入樣例#1: 複製
5 5 25957 6405 15770 26287 26465 2 2 1 3 4 1 4 5 1 1 2 2 4 4 1
輸出樣例#1: 複製
6405 15770 26287 25957 26287
說明
資料範圍:
對於20%的資料滿足:1 \leq N, M \leq 101≤N,M≤10
對於50%的資料滿足:1 \leq N, M \leq 10^31≤N,M≤103
對於80%的資料滿足:1 \leq N, M \leq 10^51≤N,M≤105
對於100%的資料滿足:1 \leq N, M \leq 2\cdot 10^51≤N,M≤2⋅105
對於數列中的所有數a_iai,均滿足-{10}^9 \leq a_i \leq {10}^9−109≤ai≤109
樣例資料說明:
N=5,數列長度為5,數列從第一項開始依次為[25957, 6405, 15770, 26287, 26465 ][25957,6405,15770,26287,26465]
第一次查詢為[2, 2][2,2]區間內的第一小值,即為6405
第二次查詢為[3, 4][3,4]區間內的第一小值,即為15770
第三次查詢為[4, 5][4,5]區間內的第一小值,即為26287
第四次查詢為[1, 2][1,2]區間內的第二小值,即為25957
第五次查詢為[4, 4][4,4]區間內的第一小值,即為26287
#include<bits/stdc++.h>
#define f(i,l,r) for(i=(l);i<=(r);i++)
using namespace std;
const int MAXN=200005;
struct Node{
int l,r,w;
}T[MAXN*30];
int root[MAXN],tot;
int n,m,a[MAXN],h[MAXN];
inline void change(int last,int &x,int l,int r,int pos,int d)
{
x=++tot;
T[x]=T[last];
T[x].w+=d;
if(l==r) return;
int mid=l+r>>1;
if(pos<=mid) change(T[last].l,T[x].l,l,mid,pos,d);
else change(T[last].r,T[x].r,mid+1,r,pos,d);
}
inline int Kth(int i,int j,int l,int r,int k)
{
if(l==r) return l;
int mid=l+r>>1;
int num=T[T[i].l].w-T[T[j].l].w;
if(k<=num) return Kth(T[i].l,T[j].l,l,mid,k);
return Kth(T[i].r,T[j].r,mid+1,r,k-num);
}
int main()
{
ios::sync_with_stdio(false);
int i,j,x,L,l,r,k;
cin>>n>>m;
f(i,1,n){
cin>>a[i];
h[i]=a[i];
}
sort(h+1,h+1+n);
L=unique(h+1,h+1+n)-(h+1);
f(i,1,n){
x=lower_bound(h+1,h+1+L,a[i])-h;
change(root[i-1],root[i],1,L,x,1);
}
f(i,1,m){
cin>>l>>r>>k;
cout<<h[Kth(root[r],root[l-1],1,L,k)]<<endl;
}
return 0;
}