一.题目链接:
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;
}