K-th Number（poj 2104）

题意：静态第K大

#include<cstdio>
#include<iostream>
#include<cstring>
#define N 200010
#define inf 1000000000
using namespace std;
int n,m,num,tmp[N],ans[N],sum[N];
struct node{
    int x,y,k,s,tp,cur;
};node a[N],q1[N],q2[N];
void modify(int x,int v){
    while(x<=n){
        sum[x]+=v;
        x+=x&(-x);
    }
}
int query(int x){
    int tot=;
    while(x){
        tot+=sum[x];
        x-=x&(-x);
    }
    return tot;
}
void solve(int head,int tail,int l,int r){
    if(head>tail)return;
    if(l==r){
        for(int i=head;i<=tail;i++)
            if(a[i].tp==)ans[a[i].s]=l;
        return;
    }
    int mid=l+r>>;
    for(int i=head;i<=tail;i++){
        if(a[i].tp==&&a[i].y<=mid) modify(a[i].x,);
        if(a[i].tp==) tmp[i]=query(a[i].y)-query(a[i].x-);
    }
    for(int i=head;i<=tail;i++)
        if(a[i].tp==&&a[i].y<=mid) modify(a[i].x,-);
    int l1=,l2=;
    for(int i=head;i<=tail;i++){
        if(a[i].tp==){
            if(a[i].cur+tmp[i]>=a[i].k) q1[++l1]=a[i];
            else a[i].cur+=tmp[i],q2[++l2]=a[i];
        }
        else {
            if(a[i].y<=mid) q1[++l1]=a[i];
            else q2[++l2]=a[i];
        }
    }
    for(int i=;i<=l1;i++)a[head+i-]=q1[i];
    for(int i=;i<=l2;i++)a[head+l1+i-]=q2[i];
    solve(head,head+l1-,l,mid);
    solve(head+l1,tail,mid+,r);
}
void work(){
    for(int i=;i<=n;i++){
        int x;scanf("%d",&x);
        a[++num].x=i;a[num].y=x;a[num].s=;a[num].tp=;
    }
    for(int i=;i<=m;i++){
        int x,y,k;scanf("%d%d%d",&x,&y,&k);
        a[++num].x=x;a[num].y=y;a[num].k=k;a[num].s=i;a[num].tp=;
    }
    solve(,num,-inf,inf);
    for(int i=;i<=m;i++)
        printf("%d\n",ans[i]);
}
int main(){
    while(scanf("%d%d",&n,&m)!=EOF){
        memset(tmp,,sizeof(tmp));
        memset(sum,,sizeof(sum));
        memset(ans,,sizeof(ans));
        memset(a,,sizeof(a));
        num=;
        work();
    }
    return ;
}