POJ P2104 K-th Number

You are working for Macrohard company in data structures department. After failing your previous task about key insertion you were asked to write a new data structure that would be able to return quickly k-th order statistics in the array segment. 
That is, given an array a[1...n] of different integer numbers, your program must answer a series of questions Q(i, j, k) in the form: "What would be the k-th number in a[i...j] segment, if this segment was sorted?" 
For example, consider the array a = (1, 5, 2, 6, 3, 7, 4). Let the question be Q(2, 5, 3). The segment a[2...5] is (5, 2, 6, 3). If we sort this segment, we get (2, 3, 5, 6), the third number is 5, and therefore the answer to the question is 5.

（主席树求区间第k小）

——by POJ；

http://poj.org/problem?id=2104

---

---

先%hjt吧；

主席树静态求区间第K小；

先对数据按大小离散化；

然后明确两个概念：

• 排名区间：排名区间[l,r]表示要维护排名在lr间的数据的相关信息；
• 原序列区间：原序列区间[x,y]表示要维护在原序列（至把数据按读入的顺序排列的序列）中xy间的数据的相关信息；

我们对原序列中的每一个点，维护她的前缀信息——

信息是：一棵线段树——以排名区间为区间，维护有多少数据在该区间里（如原序列：3,3,2,1,3,则第3位的前缀是{3,3,2}，于是维护{3,3,2}中有几个在排名区间[1,3]中，有几个在排名区间[1,2]，有几个在排名区间[3,3]...）；

这样你得到了如下信息：

• 原序列区间[1,j]中数据在排名区间中的分布（因为你记录了前缀影响下的线段树嘛）；
• 利用上一条信息得出：原序列区间[i,j]中数据在排名区间中的分布（对前缀进行差分）;

在我们有原序列区间[i,j]中数据在排名区间中的分布信息后，

　　若排名区间[l,mid]中分布k+个数据，则区间[l,r]的第k小必然在该区间里；

　　否则在排名区间[mid+1,r]中；

　　然后在线段树上递归分治即可；

以上是中心思路——

然后，是一些要点：

　　如，如何N棵线段树的内存问题？

　　　　其实，表面上看我们要开N棵树，但是相邻两树只有受一个数据影响而产生的差距——这只会造成一条树链的不同；

　　这样只存一棵树和N-1条链即可；

　　习惯上我们一般以完全二叉树的方式存储，在这里是不行的；

　　用最朴素的二叉树存储方法，即对一个点，维护她所代表的区间的信息和左右子节点的下标信息即可；

　　然后可以考虑内存池模拟系统内存分配的方法（我一开始还以为是内存尺的说）；

（终于把这篇拖了许久的东西发上来了，大概是我写的最长的随笔了吧；）

代码如下：

 #include<cstdio>
 #include<algorithm>
 using namespace std;
 struct ss
 {
     int size,num;
 }a[];
 int b[],fa[],L,R,K,now;
 struct tre
 {
     int l,r,num;
 }tr_data[];
 int tot;

 bool cmp1(ss a,ss b)
 {    return a.size<b.size;    }

 bool cmp2(ss a,ss b)
 {    return a.num<b.num;        }
 void build(int ,int ,int ,int );
 void work(int ,int ,int ,int );

 int main()
 {
     int i,j=,k;
     int n,m;
     scanf("%d%d",&n,&m);
     for(i=;i<=n;i++)
         scanf("%d",&a[i].size),a[i].num=i;

     sort(a+,a+n+,cmp1);
     for(i=;i<=n;i++)
     {
         if(a[i].size!=a[i-].size)++j;
         b[j]=a[i].size;
         a[i].size=j;

     }
     sort(a+,a+n+,cmp2);

     for(i=;i<=n;i++)
     {
         fa[i]=++tot;
         now=a[i].size;
         build(,j,tot,fa[i-]);//////
     }
     for(i=;i<=m;i++)
     {
         scanf("%d%d%d",&L,&R,&K);
         work(,j,fa[R],fa[L-]);//////
     }
     return ;
 }

 void build(int l,int r,int nu,int pre)
 {
     if(l==r)
     {
         tr_data[nu].num=tr_data[pre].num+;
         return;
     }
     int mid=(l+r)>>;
     if(now<=mid)
     {
         tr_data[nu].l=++tot;
         build(l,mid,tr_data[nu].l,tr_data[pre].l);
         tr_data[nu].r=tr_data[pre].r;
     }
     else
     {
         tr_data[nu].r=++tot;
         build(mid+,r,tr_data[nu].r,tr_data[pre].r);
         tr_data[nu].l=tr_data[pre].l;
     }
     tr_data[nu].num=tr_data[tr_data[nu].l].num+tr_data[tr_data[nu].r].num;
 }

 void work(int l,int r,int now,int pre)
 {
     if(l==r)
     {
         printf("%d\n",b[l]);
         return ;
     }
     int mid=(l+r)>>;
     if(tr_data[tr_data[now].l].num-tr_data[tr_data[pre].l].num>=K)
         work(l,mid,tr_data[now].l,tr_data[pre].l);
     else
         {    K=K-(tr_data[tr_data[now].l].num-tr_data[tr_data[pre].l].num);
             work(mid+,r,tr_data[now].r,tr_data[pre].r);    }
 }

祝AC哟；