静态区间第k大(主席树)

POJ 2104为例(主席树入门题)

思想:

可持久化线段树,也叫作函数式线段树,也叫主席树(高大上)。

可持久化数据结构(Persistent data structure):利用函数式编程的思想使其支持询问历史版本、同时充分利用它们之间的共同数据来减少时间和空间消耗。
主席树:对原序列的每一个前缀[1..i]建立出一棵线段树维护值域上每个数的出现次数(所以要先离散化)。线段树每个节点保存的是区间中前缀对应的出现的次数

注意:

  • 这里没有使用指针,而是给每个节点编号,通过编号来将节点与左右子节点连接起来。
  • 对于前缀[1,i]和前缀[1,i+1]的线段树,如果离散化后newa[i+1]<=mid ,那么这两棵线段树的右边是完全相同的,不需要重复建立。
  • 查询过程,先查看左子树中元素的出现次数是否大于k,如果是,继续查左子树,反之查询右子树。
  • 同一区间出现次数可以直接相减得到。

代码:

#include<cstdio>
#include<algorithm>
using namespace std;//[]
const int maxn = 100010, maxm = 20 * maxn;
int tot, c;
int a[maxn], newa[maxn];
int lson[maxm], rson[maxm], t[maxm], tree[maxm];
//lson,rson记录左右节点标号,t记录每一个前缀构成的线段树的根节点标号,tree记录标号对应区间中数字出现次数
int compress(int x)//离散化
{
    return lower_bound(newa+1, newa+1+c, x) - newa;
}
int build(int l, int r)
{
    int root = tot++; tree[root] = 0;
    int mid = (l+r)/2;
    if(l == r ) return root;
    lson[root] = build(l, mid);
    rson[root] = build(mid + 1, r);
    return root;
}
void update(int root, int newroot, int l, int r, int num)
{
    tree[newroot] = tree[root] + 1;
    if(l == r) return;
    int mid = (l + r)/2;
    if(num <= mid){
        lson[newroot]  = tot++;//有变动,重新建立
        rson[newroot] = rson[root];//右边不变
        update( lson[root], lson[newroot], l, mid, num);
    }else{
        rson[newroot] = tot++;//有变动,重新建立
        lson[newroot] = lson[root];//左边不变
        update(rson[root], rson[newroot], mid + 1, r, num);
    }
}
int query(int leftroot, int rightroot, int l, int r, int k)
{
    if(l == r ) return l;
    int mid = (l + r)/2;
    if(tree[lson[rightroot]] - tree[lson[leftroot]] >= k){
        query(lson[leftroot],  lson[rightroot], l, mid, k);
    }else{
        int temp = tree[lson[rightroot]] - tree[lson[leftroot]];
        query(rson[leftroot],  rson[rightroot], mid + 1, r, k - temp);
    }
}
int main (void)
{
    int n,m;scanf("%d%d",&n,&m);
    tot = 0;
    for(int i = 1; i <= n; i++) {
            scanf("%d",&a[i]);
            newa[i] = a[i];
    }
    sort(newa+1, newa+1+n);
    c = unique(newa+1, newa+1+n) - newa-1;//去重
    t[0] = build(1, c);//初始化
    for(int i = 1; i <= n; i++){
        t[i] = tot++;
        update(t[i-1], t[i], 1 , c, compress(a[i]));//不断更新,建树
    }
    int l, r, k;
    while(m--){
        scanf("%d%d%d",&l,&r,&k);
        printf("%d\n", newa[query(t[l-1], t[r], 1 ,c, k)]);
    }
    return 0;
}//1800ms

还是划分树快些。。。
真的是理解花了好久,连写再调试又花了好久。。。。。。然而只学了点毛皮。
动态区间第k大貌似要用到树状数组,过几天再来研究一下!

posted @ 2016-02-16 12:52  zhuyujiang  阅读(257)  评论(1编辑  收藏  举报