Pku2104 k - th number

解题方法：线段树 + 归并排序 + 二分枚举

本人比较笨，这道题看了N天，主要花在二分枚举上，虽然AC，至今仍然还不是很清楚二分枚举的原理。

希望牛人路过，指导下。

1. #include <stdio.h>
2. #include <memory.h>
4. const long MAX = 100001 ;
6. inline int tMin( int u , int v )
7. {
8.     return ( u > v ? v : u ) ;
9. }
10. inline int tMax( int u , int v )
11. {
12.     return ( u > v ? u : v ) ;
13. }
15. struct STREE
16. {
17.     int start ;
18.     int end ;
19.     int level ;
20.     struct STREE *leftc ;
21.     struct STREE *rightc ;
23.     void BuildST( int s , int e , int lv ) ;
24.     int Query( int s , int e , int val ) ;
25. }Stree[(MAX << 1) + 100] ;
26. STREE *root = &Stree[0] ;
27. int pos = 1 ;
29. int sortarr[18][MAX] ;
30. int gNum[MAX] ;
32. void STREE::BuildST( int s, int e, int lv )
33. {
34.     start = s ;
35.     end = e ;
36.     level = lv ;
37.     if ( s == e )
38.     {
39.         leftc = rightc = NULL ;
40.         sortarr[lv][s] = gNum[s] ;
41.         return ;
42.     }
43.     
44.     int mid = ( s + e ) >> 1 ;
45.     leftc = &Stree[pos++] ;
46.     rightc = &Stree[pos++] ;
48.     leftc->BuildST( s , mid , lv + 1 ) ;
49.     rightc->BuildST( mid + 1 , e , lv + 1 ) ;
51.     int i = s , j = mid + 1 , k = s ;
52.     
53.     while ( i <= mid && j <= e )
54.     {
55.         if ( sortarr[lv + 1][i] < sortarr[lv + 1][j] )
56.             sortarr[lv][k++] = sortarr[lv + 1][i++] ;
57.         else {
58.             sortarr[lv][k++] = sortarr[lv + 1][j++] ;
59.         }
60.     }
61.     while ( i <= mid )
62.     {
63.         sortarr[lv][k++] = sortarr[lv + 1][i++] ;
64.     }
65.     while ( j <= e )
66.     {
67.         sortarr[lv][k++] = sortarr[lv + 1][j++] ;
68.     }
69. }
71. int STREE::Query( int s, int e, int val )
72. {
73.     int ret , mid , l = s , r = e ;
75.     if ( s == start && end == e )
76.     {
77.         if ( val <= sortarr[level][start] ) return 0 ;
78.         if ( val > sortarr[level][end] ) return ( end - start + 1 ) ;
79.         if ( val == sortarr[level][end] ) return ( end - start ) ;
81.         while ( l <= r )
82.         {
83.             mid = l + (( r - l ) >> 1) ;
84.             if ( val <= sortarr[level][mid] )
85.                 r = mid - 1 ;
86.             else l = mid + 1 ;
87.         }
89.         return l - start ;
90.     }
91.     else {
92.         ret = 0 ;
93.         mid = start + (( end - start ) >> 1) ;
94.         if ( mid >= s ) ret += leftc->Query( s, tMin(e,mid), val) ;
95.         if ( mid + 1 <= e ) ret += rightc->Query( tMax(s,mid+1), e, val) ;
97.         return ret ;
98.     }
99. }
101. int main()
102. {
103.     int n , m , s , e , k , i , l , r , mid ;
105.     scanf("%d %d", &n, &m) ;
107.     for ( i = 1 ; i <= n ; i++ )
108.     {
109.         scanf("%ld", &gNum[i]);
110.     }
112.     root->BuildST( 1 , n , 0 ) ;
114.     for ( i = 1 ; i <= m ; i++ )
115.     {
116.         scanf("%d %d %d", &s, &e, &k) ;
118.         l = 1 , r = n ;
120.         while ( l <= r )
121.         {
122.             mid = l + ((r - l) >> 1) ;
124.             int p = root->Query( s, e, sortarr[0][mid] ) ;
126.             if ( k > p )
127.                 l = mid + 1 ;
128.             else r = mid - 1 ;
129.         }
131.         printf("%d/n", sortarr[0][r] ) ;
132.     }
133.     return 0 ;
134. }

参考某牛人的代码：http://www.wiskey86.cn/wordpress/?p=24

在此感谢！