poj 分桶法和平方分割

1. #include<stdio.h>  
2. #include<algorithm>  
3. #include<vector>  
4. #include<cmath>  
5. using namespace std;  
6. const int B = 1000 ;  
7. const int MAX_N = 100000;  
8. int nums[200000];  
9. int L[10000];  
10. int R[10000];  
11. int K[10000];  
12. int A[200000];  
13.   
14. vector<int> bucket[MAX_N/B];  
15.   
16. int main()  
17. {  
18.     int n,m;  
19.     scanf("%d%d",&n,&m);  
20.     for(int i=1;i<=n;i++)  
21.     {  
22.         scanf("%d",&nums[i]);  
23.         A[i]=nums[i];  
24.     }  
25.   
26.     sort(A+1,A+n+1);  
27.     for(int i=1;i<=m;i++)  
28.         scanf("%d%d%d",&L[i],&R[i],&K[i]);  
29.   
30.     int l,r;  
31.     for(int i=1;i<=n;i++)  
32.         bucket[i/B].push_back(nums[i]);  
33.     for(int i=0;i<n/B;i++)  
34.         sort(bucket[i].begin(),bucket[i].end());  
35.   
36.     for(int i=1;i<=m;i++)  
37.     {  
38.         int l=L[i],r=R[i],k=K[i];  
39.         int lb=0,ub=n;  
40.         while(ub-lb>1)  
41.         {  
42.             int c=0;  
43.             int tl=l,tr=r+1;  
44.             int mid=(ub+lb)/2;  
45.             int x=A[mid];  
46.             while(tl<tr&&tl%B!=0)  
47.             {  
48.                 if(nums[tl++]<=x)  
49.                 {  
50.                     c++;  
51.                 }  
52.             }  
53.             while(tl<tr&&tr%B!=0) if(nums[--tr]<=x) c++;  
54.             for(int i=tl/B;i*B+B<=tr;i++)  
55.             {  
56.                 c+=upper_bound(bucket[i].begin(),bucket[i].end(),x)-bucket[i].begin();  
57.             }  
58.             if(c>=k)//c 满足条件，但是可以更大一点  
59.                 ub=mid;  
60.             else  
61.                 lb=mid;  
62.         }  
63.         printf("%d\n",A[ub]);  
64.     }  
65.     return 0;  
66. }