K-th Number （POJ - 2104，可持久化线段树模板）

一.题目链接：

POJ-2104

二.题目大意：

求区间第 k 大值.

三.分析：

因上午迟到，被队长强迫学习主席树.....

今天心血来潮，想学习一下主席树！！！

因为这题是模板，具体的内容这里不再详谈，只写一些自己对主席树的理解.

主席树的建树过程

插入两个节点：1，2 （n == 4）

起初是一颗空树

然后插入第一个节点

然后第 2 个节点

可以看到每插入一个节点，就会多出一条树链.

并且，这条树链仅与前一条树链（如果存在的话）有关.

ps：主席树的互异叶子结点个数为len，而不是 n.

四.代码实现：

#include <set>
#include <map>
#include <ctime>
#include <queue>
#include <cmath>
#include <stack>
#include <bitset>
#include <vector>
#include <cstdio>
#include <sstream>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>
#define eps 1e-8
#define lc k * 2
#define rc k * 2 + 1
#define pi acos(-1.0)
#define ll long long int
using namespace std;

const int M = (int)1e5;
const ll mod = (ll)1e9 + 7;
const int inf = 0x3f3f3f3f;

int a[M + 5];
int b[M + 5];

int cnt;
int root[M + 5];///记录第i条树链的树根编号

struct node
{
    int sum;
    int l, r;
}tree[M * 40 + 5];

int discrete(int n)
{
    for(int i = 1; i <= n; ++i)
        b[i] = a[i];
    sort(b + 1, b + n + 1);
    return unique(b + 1, b + n + 1) - (b + 1);
}

int tofind(int x, int len)
{
    return lower_bound(b + 1, b + len + 1, x) - b;
}

void update(int l, int r, int &x, int y, int v)
{
    tree[++cnt] = tree[y];
    tree[cnt].sum++;
    x = cnt;
    if(l == r)
        return;
    int mid = (l + r) >> 1;
    if(v <= mid)
        update(l, mid, tree[x].l, tree[y].l, v);
    else
        update(mid + 1, r, tree[x].r, tree[y].r, v);
}

int query(int l, int r, int x, int y, int k)
{
    if(l == r)
        return l;
    int mid = (l + r) >> 1;
    int sum = tree[tree[y].l].sum - tree[tree[x].l].sum;
    if(k <= sum)
        return query(l, mid, tree[x].l, tree[y].l, k);
    else
        return query(mid + 1, r, tree[x].r, tree[y].r, k - sum);
}

int main()
{
    int n, m;
    scanf("%d %d", &n, &m);
    for(int i = 1; i <= n; ++i)
        scanf("%d", &a[i]);
    int len = discrete(n);
    for(int i = 1; i <= n; ++i)
        update(1, len, root[i], root[i - 1], tofind(a[i], len));
    int l, r, k;
    while((m--) > 0)
    {
        scanf("%d %d %d", &l, &r, &k);
        printf("%d\n", b[query(1, len, root[l - 1], root[r], k)]);
    }
    return 0;
}