POJ2104 K-th Number 静态区间第k最值 平方分割

 1 const int bktSize=1024;
 2 const int bktMaxIdx=bktSize-1;
 3 const int bktCount=128;
 4 const int bktDigit=10;
 5 const int maxV=1e9;
 6 
 7 int bucket[bktCount][bktSize];
 8 int unOrdered[bktSize*bktCount];
 9 int ordered[bktSize*bktCount];
10 int N,K;
11 
12 #include <cstdio>
13 #include <cstring>
14 #include <algorithm>
15 
16 void init()
17 {
18     scanf("%d%d",&N,&K);
19     memset(bucket[N>>bktDigit],0x7f,sizeof(bucket[N>>bktDigit]));
20     for(int i=0;i<N;i++) 
21     {
22         scanf("%d",unOrdered+i);
23         ordered[i]=unOrdered[i];
24         bucket[i>>bktDigit][i&bktMaxIdx]=unOrdered[i];
25     }
26     
27     using std::sort;
28     int bktUsed=N>>bktDigit;
29     sort(ordered,ordered+N);
30     for(int i=0;i<=bktUsed;i++) sort(bucket[i],bucket[i]+bktSize);
31 }
32 
33 inline void enumerate(int _rangeL,int _rangeR,int _val,int& _notMore)
34 {
35     for(int i=_rangeL;i<=_rangeR;i++)
36         if(unOrdered[i]<=_val) ++_notMore;
37 }
38 
39 inline void countBucket(int _bktIdx,int _val,int& _notMore)
40 {
41     using std::upper_bound;
42     
43     int* ub=upper_bound(bucket[_bktIdx],bucket[_bktIdx]+bktSize,_val);
44     _notMore+=(ub-bucket[_bktIdx]);
45 }
46 
47 int ask(int _rangeL,int _rangeR,int _k) //k-th smallest
48 {
49     int digitL=_rangeL>>bktDigit;
50     int digitR=_rangeR>>bktDigit;
51     int vL=0;
52     int vR=N-1;
53     
54     while(vL<vR)
55     {
56         int midV=(vL+vR)>>1;
57         int notMore=0;
58         if(digitL==digitR) 
59             enumerate(_rangeL,_rangeR,ordered[midV],notMore);
60         else
61         {
62             for(int i=digitL+1;i<digitR;i++)
63                 countBucket(i,ordered[midV],notMore);
64             enumerate(_rangeL,((digitL+1)<<bktDigit)-1,ordered[midV],notMore);
65             enumerate(digitR<<bktDigit,_rangeR,ordered[midV],notMore);
66         }
67         
68         if(notMore<_k) vL=midV+1;
69         else vR=midV;
70     }
71     return ordered[vL];
72 }
73 
74 int main()
75 {
76     init();
77     for(int i=0;i<K;i++)
78     {
79         int l,r,k;
80         scanf("%d%d%d",&l,&r,&k);
81         printf("%d\n",ask(l-1,r-1,k));    
82     }
83     return 0;
84 }