hdu 2665 Kth number (静态区间第k大,可持久化线段树模板题)

题意:

求区间第k大数

思路:

可持久化线段树。

其实这东西在国内更常见的名字应该是 zhu xi 树?应该是由于“中国高端数据结构领导者”的hjt比较早得引入(?推广?)而得来的。

不过之前没写什么东西都被封,写个zhu xi 树,怕是政治敏感够枪毙100回了orz

似乎也叫函数式线段树。

写个模板题练下手。学习笔记之后补。

  1/* ***********************************************
  2Author :111qqz
  3Created Time :2017年10月16日 星期一 21时51分55秒
  4File Name :2104.cpp
  5************************************************ */
  6
  7#include <cstdio>
  8#include <cstring>
  9#include <iostream>
 10#include <algorithm>
 11#include <vector>
 12#include <queue>
 13#include <set>
 14#include <map>
 15#include <string>
 16#include <cmath>
 17#include <cstdlib>
 18#include <ctime>
 19#define PB push_back
 20#define fst first
 21#define sec second
 22#define lson l,m,rt<<1
 23#define rson m+1,r,rt<<1|1
 24#define ms(a,x) memset(a,x,sizeof(a))
 25typedef long long LL;
 26#define pi pair < int ,int >
 27#define MP make_pair
 28
 29using namespace std;
 30const double eps = 1E-8;
 31const int dx4[4]={1,0,0,-1};
 32const int dy4[4]={0,-1,1,0};
 33const int inf = 0x3f3f3f3f;
 34const int N=1E5+7;
 35int a[N],H[N];
 36int cnt;
 37int Hash( int x){ return lower_bound(H,H+cnt,x)-H;}
 38int n,m;
 39int tot;//线段树的总数。
 40int root[N];//每个元素对应的根节点
 41struct PTree
 42{
 43    int sum,left,right;
 44}tree[N*30];
 45inline int add_node( int _sum,int _left,int _right)
 46{
 47    int idx = ++tot;
 48    tree[idx].sum = _sum;
 49    tree[idx].left = _left;
 50    tree[idx].right = _right;
 51    return idx;
 52}
 53void Insert( int &root,int pre_rt,int pos,int l,int r)
 54{
 55    root = add_node(tree[pre_rt].sum + 1,tree[pre_rt].left,tree[pre_rt].right);
 56    //printf("l:%d r %d root:%d\n",l,r,root);
 57    if (l==r) return;
 58    int mid = (l+r)>>1;
 59    if (pos<=mid)
 60    Insert(tree[root].left,tree[pre_rt].left,pos,l,mid);
 61    else
 62    Insert(tree[root].right,tree[pre_rt].right,pos,mid+1,r);
 63}
 64int Query( int L,int R,int l,int r,int k)//区间第k大
 65{
 66    if (l==r) return l;
 67    int mid = (l+r)>>1;
 68    int sum = tree[tree[R].left].sum - tree[tree[L].left].sum;
 69    //sum表示 左区间的元素个数
 70    if (k<=sum)
 71    return Query(tree[L].left,tree[R].left,l,mid,k);
 72    else return Query(tree[L].right,tree[R].right,mid+1,r,k-sum);
 73}
 74
 75
 76
 77int main()
 78{
 79    #ifndef  ONLINE_JUDGE 
 80    freopen("./in.txt","r",stdin);
 81  #endif
 82    int T;
 83    cin>>T;
 84    while (T--){
 85    tot = 0 ;
 86    scanf("%d%d",&n,&m);
 87    for ( int i = 1 ; i <= n ; i++) scanf("%d",a+i),H[i] = a[i];
 88    root[0] = 0;
 89    sort(H+1,H+n+1);
 90    cnt = unique(H+1,H+n+1)-(H+1);
 91    for ( int i = 1; i <= n ; i++)
 92    {
 93        int pos = Hash(a[i]);
 94        //printf("pos:%d\n",pos);
 95        Insert(root[i],root[i-1],pos,1,cnt);
 96        //cout<<"i"<<i<<endl;
 97    }
 98    while (m--)
 99    {
100        int l,r,k;
101        scanf("%d %d %d",&l,&r,&k);
102        int pos = Query(root[l-1],root[r],1,cnt,k);
103        printf("%d\n",H[pos]);
104    }
105    }
106
107
108  #ifndef ONLINE_JUDGE  
109  fclose(stdin);
110  #endif
111    return 0;
112}