题目:http://poj.org/problem?id=2104
题意:给定一个长度为n的无序数组,然后有m组询问l r k,求区间[l,r]中的第k大数
思路:以前用划分树做过,现在学习主席树,模板题。个人对主席树的理解:就是把线段树更新过程中所有的历史状态记录下来,例如更新m次的话,那么就会有m种状态,直接建m棵线段树的话,时间和空间都不允许,于是用了一些很巧妙的思想,因为更新线段树某个节点时,只会改变从此点到根节点路径上的点的状态,其他点不会受到影响,也就是说一种状态和前一种状态只有一条路径是不同的,有大量的重复信息,于是直接利用这些重复信息,对于改变的那部分信息再新建节点。弱现在只能理解这么多了,这篇博客讲的很好,图文并茂http://www.cnblogs.com/zyf0163/p/4749042.html
总结:主席树真是个神奇的东西,不过空间开多大不太明白,此题中直接乘了20
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#define debug() puts("here")
using namespace std;
const int N = 100100;
int lson[N*20], rson[N*20], sum[N*20], root[N];
int arr[N], brr[N];
int n, m, tot;
void build(int l, int r, int &rt) //建立一个空树,所有节点的值为0
{
rt = ++tot;
sum[rt] = 0;
if(l == r) return;
int mid = (l + r) >> 1;
build(l, mid, lson[rt]);
build(mid + 1, r, rson[rt]);
}
void update(int pre, int &rt, int l, int r, int val) //sum[i]储存的是它所管辖区间内元素出现的次数,即[l,r]区间
{
rt = ++tot;
lson[rt] = lson[pre], rson[rt] = rson[pre], sum[rt] = sum[pre] + 1;
if(l == r) return;
int mid = (l + r) >> 1;
if(val <= mid) update(lson[pre], lson[rt], l, mid, val); //此元素在左区间
else update(rson[pre], rson[rt], mid + 1, r, val);//右区间
}
int query(int ss, int tt, int l, int r, int k) //返回值是第k大数在离散去重后的数组中的下标,线段树管理的区间是离散后的区间
{//要查询第k大数,先求左区间内元素的个数tmp,tmp >= k,直接查询左区间内的第k大,否则查询右区间内的第k-tmp大
if(l == r) return l; //此时无论k值,我们确定k就在[l,r]区间中,又l==r,结果求出
int mid = (l + r) >> 1;
int tmp = sum[lson[tt]] - sum[lson[ss]]; //左区间内元素个数,这个想一下就明白了
if(tmp >= k) return query(lson[ss], lson[tt], l, mid, k);
else return query(rson[ss], rson[tt], mid + 1, r, k - tmp);
}
int main()
{
int a, b, c;
while(~ scanf("%d%d", &n, &m))
{
tot = 0;
for(int i = 1; i <= n; i++) scanf("%d", &arr[i]), brr[i] = arr[i];
sort(brr + 1, brr + 1 + n);
int len = unique(brr + 1, brr + 1 + n) - brr - 1;
build(1, len, root[0]);
for(int i = 1; i <= n; i++)
{
arr[i] = lower_bound(brr + 1, brr + 1 + len, arr[i]) - brr;
update(root[i-1], root[i], 1, len, arr[i]);
}
while(m--)
{
scanf("%d%d%d", &a, &b, &c);
printf("%d\n", brr[query(root[a-1], root[b], 1, len, c)]);
}
}
return 0;
}