[主席树]K-th Number

正确解法：

分块（超时）

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <set>
#include <map>
#include <vector>
#include <cctype>
#include <sstream>
using namespace std;
typedef long long ll;
const int inf=0x7fffffff;
const int N=100000+100;
const int M=5000+10;
const double PI=acos(-1.0);
int B=1000;
int n,m,a[N],num[N];
int I,J,K;
vector<int> buck[N/1000];
int main()
{
    scanf("%d %d",&n,&m);
    for(int i=0;i<n;i++)
    {
        scanf("%d",&a[i]);
        num[i]=a[i];
        buck[i/B].push_back(a[i]);
    }
    sort(num,num+n);
    for(int i=0;i<n/B;i++)
        sort(buck[i].begin(),buck[i].end());
    for(int i=0;i<m;i++)
    {
        scanf("%d %d %d",&I,&J,&K);
        int lb=0,ub=n-1;
        while(lb<=ub)
        {
            int md= lb+ub >>1;
            int x=num[md];
            int tl=I-1,tr=J,c=0;
            while(tl<tr &&tl%B!=0)  if(a[tl++]<=x)  c++;
            while(tl<tr &&tr%B!=0)  if(a[--tr]<=x)  c++;
            while(tl<tr)
            {
                int b=tl/B;
                c+=upper_bound(buck[b].begin(),buck[b].end(),x)-buck[b].begin();
                tl+=B;
            }
            if(c>=K)    ub=md-1;
            else lb=md+1;
        }
        printf("%d
",num[lb]);
    }


    return 0;
}

主席树：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <set>
#include <map>
#include <vector>
#include <cctype>
#include <sstream>
using namespace std;
typedef long long ll;
const int inf=0x7fffffff;
const int N=100000+100;
const int M=5000+10;
const double PI=acos(-1.0);
int n,m,a[N],num[N];
int I,J,K;
vector<int>buck[(1<<18)-1];
void build(int rt,int l,int r)
{
    if(l==r)
    {
        buck[rt].push_back(a[l]);
        return ;
    }
    int m= l+r >>1;
    build(rt<<1,l,m);
    build(rt<<1|1,m+1,r);
    buck[rt].resize(r-l+1);
    merge(buck[rt<<1].begin(),buck[rt<<1].end(),buck[rt<<1|1].begin(),buck[rt<<1|1].end(),buck[rt].begin());
}
int query(int L,int R,int x,int rt,int l,int r)
{
    if(R<l||r<L)
        return 0;
    if(L<=l&&r<=R)
        return upper_bound(buck[rt].begin(),buck[rt].end(),x)-buck[rt].begin();
    int ans=0,m=l+r >>1;
    ans+=query(L,R,x,rt<<1,l,m);
    ans+=query(L,R,x,rt<<1|1,m+1,r);
    return ans;
}
int main()
{
    scanf("%d %d",&n,&m);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
        num[i]=a[i];
    }
    sort(num+1,num+1+n);
    build(1,1,n);
    while(m--)
    {
        scanf("%d %d %d",&I,&J,&K);
        //J++;
        int lb=1,ub=n;
        while(lb<=ub)
        {
            int md=lb+ub >>1;
            int c=query(I,J,num[md],1,1,n);
            if(c>=K)    ub=md-1;
            else lb=md+1;
        }
        printf("%d
",num[lb]);
    }

    return 0;
}