1. 程式人生 > >luoguP3834 【模板】可持久化線段樹 1(主席樹)

luoguP3834 【模板】可持久化線段樹 1(主席樹)


luoguP3834 【模板】可持久化線段樹 1(主席樹)

  Time Limit: 1 Sec
  Memory Limit: 256 MB

Description

   這是個非常經典的主席樹入門題——靜態區間第K小
   資料已經過加強,請使用主席樹。同時請注意常數優化
  如題,給定N個正整數構成的序列,將對於指定的閉區間查詢其區間內的第K小值。
 

Input

   第一行包含兩個正整數N、M,分別表示序列的長度和查詢的個數。
   第二行包含N個正整數,表示這個序列各項的數字。
   接下來M行每行包含三個整數 l, r, k, 表示查詢區間[l, r]內的第k小值。
 

Output

   輸出包含k行,每行1個正整數,依次表示每一次查詢的結果

Sample Input

   5 5
   25957 6405 15770 26287 26465
   2 2 1
   3 4 1
   4 5 1
   1 2 2
   4 4 1
  

Sample Output

  6405
  15770
  26287
  25957
  26287

HINT

  資料範圍:
  對於20%的資料滿足:\(1 \leq N, M \leq 10\)
  對於50%的資料滿足:\(1 \leq N, M \leq 10^3\)
  對於80%的資料滿足:\(1 \leq N, M \leq 10^5\)


  對於100%的資料滿足:\(1 \leq N, M \leq 2\cdot 10^5\)
  對於數列中的所有數\(a_i\) ,均滿足 \(-{10}^9 \leq a_i \leq {10}^9\)
  樣例資料說明:
  N=5,數列長度為5,數列從第一項開始依次為\([25957, 6405, 15770, 26287, 26465 ]\)
  第一次查詢為\([2, 2]\)區間內的第一小值,即為6405
  第二次查詢為\([3, 4]\)區間內的第一小值,即為15770
  第三次查詢為\([4, 5]\)區間內的第一小值,即為26287
  第四次查詢為\([1, 2]\)區間內的第二小值,即為25957
  第五次查詢為\([4, 4]\)
區間內的第一小值,即為26287

題目地址:luoguP3834 【模板】可持久化線段樹 1(主席樹)

題目大意: 題目很簡潔了:)

題解:

  複習模板題 ing


AC程式碼

#include <stdio.h>
#include <algorithm>
using namespace std;
const int N=2e5+5,NlogN=36e5;
int n,Q;
int a[N],A[N];
inline int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
int Knum=0,sum[NlogN],rt[N],ch[NlogN][2];
void change(int u,int &v,int l,int r,int val){
    v=++Knum;
    sum[v]=sum[u]+1;
    ch[v][0]=ch[u][0];
    ch[v][1]=ch[u][1];
    if(l==r)return;
    int mid=(l+r)>>1;
    if(val<=mid)change(ch[u][0],ch[v][0],l,mid,val);
    else change(ch[u][1],ch[v][1],mid+1,r,val);
}
int query(int L,int R,int K,int l,int r){
    if(l==r)return l;
    int mid=(l+r)>>1;
    int x=sum[ch[R][0]]-sum[ch[L][0]];
    if(K<=x)return query(ch[L][0],ch[R][0],K,l,mid);
    else return query(ch[L][1],ch[R][1],K-x,mid+1,r);
}
int main(){
    n=read(),Q=read();
    for(int i=1;i<=n;i++)
        a[i]=A[i]=read();
    sort(A+1,A+n+1);
    int All=unique(A+1,A+n+1)-A-1;
    for(int i=1;i<=n;i++)
        a[i]=lower_bound(A+1,A+All+1,a[i])-A;
    for(int i=1;i<=n;i++)
        change(rt[i-1],rt[i],1,All,a[i]);
    while(Q--){
        int L=read(),R=read(),K=read();
        printf("%d\n",A[query(rt[L-1],rt[R],K,1,All)]);
    }
    return 0;
}


  作者:skl_win
  出處:https://www.cnblogs.com/shaokele/
  本文版權歸作者和部落格園共有,歡迎轉載,但未經作者同意必須保留此段宣告,且在文章頁面明顯位置給出原文連線,否則保留追究法律責任的權利。