poj2761_划分树

题意：给一串数列，然后求其中的各个范围内的第k大数

分析：划分树，具体分析见划分树算法。跟之前做过2104相比，这道题好象没有重复的元素。

代码：

View Code

  1 #include <iostream>

  2 #include <stdio.h>

  3 #include <cstring>

  4 #include <algorithm>

  5 using namespace std;

  6 //14256K 1813MS

  7 //划分树

  8 const int maxnum=100001;

  9 struct tree

 10 {

 11     int array[maxnum];

 12     int sum[maxnum];

 13 }tree[20];

 14 int sorted[maxnum];

 15 

 16 void build_tree(int cur,int l,int r)

 17 {

 18     if(l==r)

 19         return ;

 20     int i;

 21     int m=(l+r)>>1;

 22     int lsame=m-l+1;//?

 23     for(i=l;i<=r;i++)

 24         if(tree[cur].array[i]<sorted[m])

 25             lsame--;

 26 

 27     int lnum=l,rnum=m+1;

 28     for(i=l;i<=r;i++)

 29     {

 30         if(i==l)

 31             tree[cur].sum[i]=0;

 32         else

 33             tree[cur].sum[i]=tree[cur].sum[i-1];

 34         if(tree[cur].array[i]<sorted[m])

 35         {

 36             tree[cur].sum[i]++;

 37             tree[cur+1].array[lnum++]=tree[cur].array[i];

 38         }

 39         else if(tree[cur].array[i]>sorted[m])

 40             tree[cur+1].array[rnum++]=tree[cur].array[i];

 41         else

 42         {

 43             if(lsame>0)

 44             {

 45                 lsame--;

 46                 tree[cur].sum[i]++; //?

 47                 tree[cur+1].array[lnum++]=tree[cur].array[i];

 48             }

 49             else

 50                 tree[cur+1].array[rnum++]=tree[cur].array[i];

 51         }

 52     }

 53     build_tree(cur+1,l,m);

 54     build_tree(cur+1,m+1,r);

 55 }

 56 

 57 int find(int cur,int st,int ed,int l,int r,int k)

 58 {

 59     if(l==r)

 60         return tree[cur].array[l];

 61     int lnum,rnum;

 62     int m=(st+ed)>>1;

 63     if(l-1<st)

 64         lnum=0;

 65     else

 66         lnum=tree[cur].sum[l-1];

 67     rnum=tree[cur].sum[r];

 68     if(rnum-lnum>=k)

 69         return find(cur+1,st,m,st+lnum,st+rnum-1,k);

 70     else

 71         return find(cur+1,m+1,ed,m+1+l-st-lnum,m+1+r-st-rnum,k-(rnum-lnum));

 72 }

 73 

 74 int main()

 75 {

 76     int n,m,i;

 77     scanf("%d%d",&n,&m);

 78     for(i=1;i<=n;i++)

 79     {

 80         scanf("%d",&tree[0].array[i]);

 81         sorted[i]=tree[0].array[i];

 82     }

 83     sort(sorted+1,sorted+1+n);

 84     build_tree(0,1,n);

 85 

 86     int p,r,k;

 87     while(m--)

 88     {

 89         scanf("%d%d%d",&p,&r,&k);

 90         printf("%d\n",find(0,1,n,p,r,k));

 91     }

 92     return 0;

 93 }

 94 

 95 /*

 96 7 2

 97 1 5 2 6 3 7 4

 98 1 5 3

 99 2 7 1

100 */