权值线段树是一种可以用作求出已知序列中第K大的数的一种线段树。
工作原理
1.加入操作
每次加入一个数，就把该数所在的所有区间都加一。
void update(node *rt, int u) { if(rt->l == u && rt->r == u) { rt->ti ++; return ; } rt->ti ++; int mid = (rt->l + rt->r) >> 1; if(u <= mid) update(rt->lson, u); else update(rt->rson, u); }
2.查询操作
每次查询在所有已插入的数中第K大的一个数：
查询这个数的左子树中的所有数的个数ti。
a.如果ti比K大，证明这个数在左子树上，则递归左子树。
b.如果ti比K小，证明这个数在右子树上，则K变成K-ti，再递归右子树。
int query(node *rt, int k) { if(rt->l == rt->r) return rt->l; if(rt->lson->ti >= k) return query(rt->lson, k); else return query(rt->rson, k - rt->lson->ti); } 例题 黑匣子
题目描述
Black Box是一种原始的数据库。它可以储存一个整数数组，还有一个特别的变量i。最开始的时候Black Box是空的．而i等于0。这个Black Box要处理一串命令。
命令只有两种：
ADD(x):把x元素放进BlackBox;
GET:i加1，然后输出Blackhox中第i小的数。
记住：第i小的数，就是Black Box里的数的按从小到大的顺序排序后的第i个元素。例如：
我们来演示一下一个有11个命令的命令串。（如下图所示）
现在要求找出对于给定的命令串的最好的处理方法。ADD和GET命令分别最多200000个。现在用两个整数数组来表示命令串：
1.A(1)，A(2)，…A(M)：一串将要被放进Black Box的元素。每个数都是绝对值不超过2000000000的整数，M$200000。例如上面的例子就是A=(3，1，一4，2，8，-1000，2)。
2.u(1)，u(2)，…u(N)：表示第u(j)个元素被放进了Black Box里后就出现一个GET命令。例如上面的例子中u=(l，2，6，6)。输入数据不用判错。
输入输出格式
输入格式：
第一行，两个整数，M，N。
第二行，M个整数，表示A(l)
……A(M)。
第三行，N个整数，表示u(l)
…u(N)。
输出格式：
输出Black Box根据命令串所得出的输出串，一个数字一行。
代码（指针板线段树）
#includeusing namespace std;const int maxn = 200005;struct node{ int l, r; int ti; node *lson, *rson;} tree[maxn << 1];struct number{ long long val; int pos;} a[maxn];int u[maxn], s[maxn];int n, m;int t = 0;bool comp(const number & a, const number & b) { return a.val < b.val; }void build(node *rt, long long l, long long r) { rt->l = l; rt->r = r; rt->ti = 0; if(l!=r) { rt->lson = &tree[++ t]; rt->rson = &tree[++ t]; int mid = (rt->l + rt->r) >> 1; build(rt->lson, l, mid); build(rt->rson, mid+1, r); } }void update(node *rt, int u) { if(rt->l == u && rt->r == u) { rt->ti ++; return ; } rt->ti ++; int mid = (rt->l + rt->r) >> 1; if(u <= mid) update(rt->lson, u); else update(rt->rson, u); }int query(node *rt, int k) { if(rt->l == rt->r) return rt->l; if(rt->lson->ti >= k) return query(rt->lson, k); else return query(rt->rson, k - rt->lson->ti); }int main(){ int j = 1; scanf("%d%d", &m, &n); for(int i = 1; i <= m; i ++) scanf("%lld", &a[i].val), a[i].pos = i; for(int i = 1; i <= n; i ++) scanf("%d", &u[i]); sort(a +1, a +m+1, comp); for(int i=1; i <= m; i++) s[a[i].pos] = i; build(tree, 1, m); for(int i = 1; i <=m; i ++) { update(tree, s[i]); while(i == u[j]) { printf("%lld\n", a[query(tree, j)].val); ++ j; } } return 0;}
参考文献
【专题】权值线段树 http://blog.csdn.net/cdy1206473601/article/details/76943927