码迷,mamicode.com
首页 > 其他好文 > 详细

poj2104 主席树 区间K大 在线 无修改

时间:2014-08-09 09:02:57      阅读:350      评论:0      收藏:0      [点我收藏+]

标签:style   blog   http   color   使用   os   io   数据   

关于主席树:

主席树(Chairman Tree)是一种离线数据结构,使用函数式线段树维护每一时刻离散之后的数字出现的次数,由于各历史版本的线段树结构一致,可以相减得出区间信息,即该区间内出现的数字和对应的数量,由于在线段树内,左子树代表的数字都小与右子树,便可像平衡树一样进行K大询问。新建一颗树是\(O(logn)\),查询一次也为\(O(logn)\)。

比划分树好想&写多了,但是在POJ上比划分树慢一些。

CODE:

bubuko.com,布布扣
 1 #include <cstdio>
 2 #include <algorithm>
 3 using namespace std;
 4 const int maxn = 100000 + 100;
 5 
 6 int id[maxn], root[maxn], a[maxn], b[maxn], n, m, size;
 7 struct chairman_tree_node {
 8     int ls, rs, w;
 9 } T[maxn * 23];
10 void insert(int l, int r, int &pos, int val) {
11     T[++size] = T[pos], pos = size;
12     ++T[pos].w;
13     if(l == r) return ;
14     int mid = (l + r) >> 1;
15     if(val <= mid) insert(l, mid, T[pos].ls, val);
16     else insert(mid + 1, r, T[pos].rs, val);
17 }
18 int query(int p, int q, int l, int r, int k) {
19     if(l == r) return l;
20     int t = T[T[q].ls].w - T[T[p].ls].w;
21     int mid = (l + r) >> 1;
22     if(k <= t) return query(T[p].ls, T[q].ls, l, mid, k);
23     else return query(T[p].rs, T[q].rs, mid + 1, r, k - t);
24 }
25 bool cmp(int x, int y) {
26     return a[x] < a[y];
27 }
28 int main() {
29     scanf("%d%d", &n, &m);
30     for(int i = 1; i <= n; ++i) {
31         scanf("%d", &a[i]);
32         id[i] = i;
33     }
34     sort(id + 1, id + n + 1, cmp);
35     for(int i = 1; i <= n; ++i) {
36         b[id[i]] = i;
37     }
38     for(int i = 1; i <= n; ++i) {
39         root[i] = root[i - 1];
40         insert(1, n, root[i], b[i]);
41     }
42     for(int i = 1; i <= m; ++i) {
43         int l, r, k;
44         scanf("%d%d%d", &l, &r, &k);
45         printf("%d\n", a[id[query(root[l - 1], root[r], 1, n, k)]]);
46     }
47     return 0;
48 }
View Code

 

poj2104 主席树 区间K大 在线 无修改,布布扣,bubuko.com

poj2104 主席树 区间K大 在线 无修改

标签:style   blog   http   color   使用   os   io   数据   

原文地址:http://www.cnblogs.com/hzf-sbit/p/3900495.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!