POJ 2104 K-th Number 平方分割

题目：

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

题意：

求区间[l,r]内第k小的元素

思路：

平方分割，效率很搓，但还是记录一下

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
using namespace std;

const int N = 100010, M = 500;
vector<int> vec[N/M+1];
int arr[N], brr[N];
int n, m;
int main()
{
    while(~ scanf("%d%d", &n, &m))
    {
        for(int i = 1; i <= n; i++)
        {
            scanf("%d", &arr[i]);
            brr[i] = arr[i];
            vec[i/M].push_back(arr[i]);
        }
        //此处与挑战稍有不同，最后一部分也要排序
        for(int i = 0; i <= n/M; i++) sort(vec[i].begin(), vec[i].end());
        sort(brr+1, brr+1+n);
        while(m--)
        {
            int a, b, k;
            scanf("%d%d%d", &a, &b, &k);
            int l = 1, r = n, res;
            while(l <= r)
            {
                int mid = (l + r) >> 1;
                int x = brr[mid];
                int tl = a, tr = b, c = 0;
                while(tl <= tr && tl % M != 0) if(arr[tl++] <= x) c++;
                //注意是(tr+1) % M，使tr移动到某块的最后一个元素，假设最后一部分恰好是一个完整的桶，且tr是这个桶的最后一个，这时不对最后一部分排序就会出错，故上面要排序
                while(tl <= tr && (tr+1) % M != 0) if(arr[tr--] <= x) c++;
                while(tl < tr)
                {
                    int d = tl / M;
                    c += upper_bound(vec[d].begin(), vec[d].end(), x) - vec[d].begin();
                    tl += M;
                }
                if(c >= k) res = mid, r = mid - 1;
                else l = mid + 1;
            }
            printf("%d\n", brr[res]);
        }
        for(int i = 0; i < N/M; i++) vec[i].clear();
    }
    return 0;
}