zoj 2112 Dynamic Rankings 带修改区间第k大的几种解法

1.分块算法 (n*sqrt(n)*log^2n) 3200+ms

给出n个数，有两种操作，一种是修改某个数的值，另一种查询指定区间第k大。

比较快的做法是树套树，而分块算法复杂度比较高写起来方便。分块算法可以很简单的处理单独修改某个值的情况。

将n个数分成num块，每块大小siz=n/num。每一个块内部进行排序，查询[l,r]第k大时，先二分答案，对于完全包含在区间的内块直接二分搜索，而对于区间两端只有部分包含的则直接遍历查找。复杂度是logn*(2*siz+num*log(siz)),这里取siz=sqrt(n),复杂度主要就是logn*sqrt(n)*logn。

对于修改，因为一个数只在一个块里，直接在那个块里进行修改，原本所在块是已排序的，修改后只有一个数不满足顺序，做一次冒泡就好了，复杂为siz=sqrt(n)。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <cctype>
#include <string>
#include <vector>
#include <map>
#include <set>
#include <vector>
#include <queue>
#include <stack>
#include <algorithm>
using namespace std;
const int maxn=5e4+10;
int n,m;
int num,siz;
vector<int> sorted[1000];
int arr[maxn];

int main()
{
    int X;
    cin>>X;
    while(X--){
        cin>>n>>m;
        for(int i=0; i<n; i++)
            scanf("%d", arr+i);

        siz=sqrt(n);//每一块的大小
        num=(n+siz-1)/siz;//共num块
        for(int i=0; i<num; i++) sorted[i].clear();
        for(int i=0; i<n; i++){
            sorted[i/siz].push_back(arr[i]);
        }
        for(int i=0; i<num; i++)
            sort(sorted[i].begin(), sorted[i].end());

        while(m--){
            char s[5]; int a,b,k;
            scanf("%s%d%d", s, &a, &b);
            if(s[0]=='Q'){
                scanf("%d", &k);
                a--;b--;
                int l=0,r=1e9+1;
                int ans;
                while(l<r){//二分答案统计<=m的有多少个数
                    int m=(l+r)>>1;
                    int cnt=0;
                    for(int i=a/siz+1; i<b/siz; i++)//处理一定完全包含的块
                        cnt+=upper_bound(sorted[i].begin(), sorted[i].end(), m)-sorted[i].begin();
                        
                    if(a%siz==0)//处理左端可能部分包含的块
                        cnt+=upper_bound(sorted[a/siz].begin(), sorted[a/siz].end(), m)-sorted[a/siz].begin();
                    else{
                        for(int i=a; i%siz!=0; i++)
                            cnt+=arr[i]<=m;
                    }
                    
                    if(b%siz==siz-1)//处理右端可能部分包含的块
                        cnt+=upper_bound(sorted[b/siz].begin(), sorted[b/siz].end(), m)-sorted[b/siz].begin();
                    else{
                        for(int i=b; i%siz!=siz-1; i--)
                            cnt+=arr[i]&l