K小数查询：分块

题目描述

小明正在研究数据结构，一天他突发奇想，出了这样一道题：
有一个含有n个数的数组和Q个操作，操作分为以下两种：
1 L R x :将第L到第R个数全部加上x。
2 L R k :询问第L到第R个数中第k小的是哪个，输出这个数。
现在请你写一个程序做出这道题。

输入格式

第一行一个正整数n，表示数组长度。第二行有n个用空格隔开的整数，表示数组初始内容。第三行一个整数Q，表示询问个数。接下来每行4个整数代表一个询问，含义见题目描述。

输出格式

对于每个询问输出一行，包含一个满足要求的整数。

样例

kth.in

4
3
2 2 4 2
1 1 3 3
2 1 4 3

kth.out

1
4

数据描述

第一个操作询问的区间包含数(1,1,3)，第2小是1。
第二个操作后数组变为{5,4,4,3}。
第三个操作询问的区间包含数(3,4,4,5)，第3小是4。

数据规模

对于20%的数据，N,Q≤1000
另外20%的数据，没有修改
1≤N,Q≤80000
1≤L≤R≤N
|x|≤5000000，数组中的数的绝对值在任何时候都不超过5000000
1≤k≤R-L+1

由于有区间修改，不能使用整体二分，可以使用分块，对于每个块内的数进行排序并保留原数，询问时二分答案。复杂度O(N logN sqrt(N logN))

/*
User:Small
Language:C++
Problem No.:kth
*/
#include<bits/stdc++.h>
#define ll long long
#define inf 999999999
using namespace std;
const int M=8e4+5;
int n,a[M],lazy[M],t,q,pos[M];
vector<int> b[M];
void update(int x){
    int ed=min(x*t,n);
    b[x].clear();
    for(int i=(x-1)*t+1;i<=ed;i++) b[x].push_back(a[i]);
    sort(b[x].begin(),b[x].end());
}
void add(int l,int r,int k){
    if(pos[l]==pos[r]){
        for(int i=l;i<=r;i++) a[i]+=k;
        update(pos[l]);
        return;
    }
    for(int i=pos[l]+1;i<=pos[r]-1;i++) lazy[i]+=k;
    for(int i=l;i<=pos[l]*t;i++) a[i]+=k;
    for(int i=r;i>(pos[r]-1)*t;i--) a[i]+=k;
    update(pos[l]);
    update(pos[r]);
}
int query(int x,int y,int k){
    int l=-5000001,r=5000001,mid;
    if(pos[x]==pos[y]){
        while(l+1<r){
            int mid=(l+r)>>1,res=0;
            for(int i=x;i<=y;i++) if(a[i]+lazy[pos[i]]<=mid) res++;
            if(res>=k) r=mid;
            else l=mid;
        }
        return r;
    }
    while(l+1<r){
        int mid=(l+r)>>1,res=0;
        for(int i=x;i<=pos[x]*t;i++)
            if(a[i]+lazy[pos[i]]<=mid) res++;
        for(int i=y;i>(pos[y]-1)*t;i--)
            if(a[i]+lazy[pos[i]]<=mid) res++;
        for(int i=pos[x]+1;i<pos[y];i++) res+=upper_bound(b[i].begin(),b[i].end(),mid-lazy[i])-b[i].begin();
        if(res>=k) r=mid;
        else l=mid;
    }
    return r;
}
int main(){
    freopen("kth.in","r",stdin);
    freopen("kth.out","w",stdout);
    ios::sync_with_stdio(false);
    cin>>n;
    t=sqrt(n);
    for(int i=1;i<=n;i++){
        cin>>a[i];
        pos[i]=(i-1)/t+1;
        b[pos[i]].push_back(a[i]);
    }
    for(int i=1;i<=pos[n];i++) sort(b[i].begin(),b[i].end());
    cin>>q;
    while(q--){
        int op,l,r,k;
        cin>>op>>l>>r>>k;
        if(op==1) add(l,r,k);
        else cout<<query(l,r,k)<<endl;
    }
    return 0;
}