POJ 2104 K-th Number

题目链接:传送门

题目大意:长为 n 的数组,m 次询问,每次问区间 [l,r] 内第 k 小的数。

题目思路:主席树

              应该是按权值建立线段树,先将权值离散化,然后利用前缀和的思想来解决。

     主席树还是有点难理解,,,

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <algorithm>
#include <cstring>
#include <stack>
#include <cctype>
#include <queue>
#include <string>
#include <vector>
#include <set>
#include <map>
#include <climits>
#define lson rt<<1,l,mid
#define rson rt<<1|1,mid+1,r
#define fi first
#define se second
#define ping(x,y) ((x-y)*(x-y))
#define mst(x,y) memset(x,y,sizeof(x))
#define mcp(x,y) memcpy(x,y,sizeof(y))
using namespace std;
#define gamma 0.5772156649015328606065120
#define MOD 1000000007
#define inf 0x3f3f3f3f
#define N 100005
#define maxn 30010
typedef pair<int,int> PII;
typedef long long LL;
LL read(){
    LL x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=(x<<3)+(x<<1)+ch-'0';ch=getchar();}
    return x*f;
}

int n,m,k,sz,L,R;
int a[N],b[N],c[N];
int root[N];
struct Seg{
    int l,r,v;
}seg[N*30];
void update(int &rot,int rt,int l,int r){
    seg[++sz]=seg[rot],rot=sz;
    ++seg[rot].v;
    if(l==r)return;
    int mid=l+r>>1;
    if(L<=mid)update(seg[rot].l,lson);
    else update(seg[rot].r,rson);
}
int query(int rt,int l,int r,int k){
    if(l==r)return a[l];
    int num=seg[seg[R].l].v-seg[seg[L].l].v;
    int mid=l+r>>1;
    if(num>=k){
        L=seg[L].l;R=seg[R].l;
        return query(lson,k);
    }
    else{
        L=seg[L].r;R=seg[R].r;
        return query(rson,k-num);
    }
}
int main(){
    int i,j,group,x,y;
    n=read();m=read();
    for(i=1;i<=n;++i)a[i]=read(),c[i]=b[i]=a[i];
    sort(a+1,a+n+1);
    int _n=unique(a+1,a+n+1)-a;
    for(i=1;i<=n;++i)b[i]=lower_bound(a+1,a+_n+1,b[i])-a;
    for(i=1;i<=n;++i)L=b[i],update(root[i]=root[i-1],1,1,n);
    while(m--){
        x=read(),y=read(),k=read();
        L=root[x-1];R=root[y];
        printf("%d
",query(1,1,n,k));
    }
    return 0;
}
原文地址:https://www.cnblogs.com/Kurokey/p/5918580.html