【主席树入门 && 区间内第k小的数】POJ - 2104 K-th Number

Step1 Problem:

输入n, m代表有n个元素，m个访问。接下来输入n个元素。输入m行访问，每行有三个元素l r k。问你下标l - r里面第k小的数是几？

Step2 Ideas:

学长教的入门
主席树由 n 个线段树组成，每个线段树在前一个线段树的基础上改变了 logn 个节点，第 n 棵线段树，代表着区间 [1, n] 的数据所对应的线段树。
这道题我们维护的是权值线段树，要求 [L, R] 内的第 k 小，拿第 R 棵线段树 - 第 L-1 棵
线段树，我们就得到 [L, R] 的权值线段树，这样问题就算是解决了。

Step3 Code:

#include<cstdio>
#include<algorithm>
using namespace std;
#define MID int mid = (l+r)/2
const int N = 1e5+5;
struct node
{
    int data, l, r;
};
int a[N], sca[N], ns, top, root[N];
node tree[N*40];
int creat(int data, int l, int r)
{
    tree[top].data = data;
    tree[top].l = l;
    tree[top++].r = r;
    //tree[top++] = (node){data, l, r};
    return top-1;
}
// 新建一棵线段树，只沿更新路径新建出较上个版本有修改的结点
// 调用参数
// root: 插入后新生成的线段树的根结点，直接赋值到 root 中
// pre: 上一棵线段树的根
// pos: 本次要插入的值
// l, r: 递归参数。默认填写 1, num
void updata(int &root, int pre, int l, int r, int pos, int v)
{
    root = creat(tree[pre].data+v, tree[pre].l, tree[pre].r);
    if(l == r) return ;
    MID;
    if(pos <= mid) updata(tree[root].l, tree[pre].l, l, mid, pos, v);
    else updata(tree[root].r, tree[pre].r, mid+1, r, pos, v);
}
int query(int e, int s, int l, int r, int k)
{
    if(l == r) return l;
    int tmp = tree[tree[e].l].data - tree[tree[s].l].data;
    MID;
    int ret;
    if(tmp >= k) ret = query(tree[e].l, tree[s].l, l, mid, k);
    else ret = query(tree[e].r, tree[s].r, mid+1, r, k - tmp);
    return ret;
}
int get_id(int x)
{
    return lower_bound(sca, sca+ns, x) - sca + 1;
}

int main()
{
    int n, m;
    while(~scanf("%d %d", &n, &m))
    {
        for(int i = 0; i < n; i++)
        {
            scanf("%d", &a[i]);
            sca[i] = a[i];
        }
        sort(sca, sca+n);//权值线段树，由于数据范围很大，所以需要离散化。
        ns = unique(sca, sca+n) - sca;
        top = 0;
        root[0] = creat(0, 0, 0);//初始化第一棵树
        for(int i = 0; i < n; i++)
        {
            int id = get_id(a[i]);
            updata(root[i+1], root[i], 1, ns, id, 1);//在前一棵树的基础上，新建一颗线段树，只需要改变 logn 个节点
        }
        int ul, ur, k;
        while(m--)
        {
            scanf("%d %d %d", &ul, &ur, &k);
            printf("%d\n", sca[query(root[ur], root[ul-1], 1, ns, k)-1]);//求 [ul, ur] 的第 k 小
        }
    }
    return 0;
}