权值线段树是一种可以用作求出已知序列中第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根据命令串所得出的输出串,一个数字一行。

代码(指针板线段树)

#include<bits/stdc++.h>
using 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

优质内容筛选与推荐>>
1、Java程序练习
2、读《数学之美系列十二——余弦定理和新闻的分类》有感 + 代码规范
3、[转]C#事件简单示例
4、今天猛地看到这么一句话,当时就慌了。。。“辛苦十几年,你也不过是红警里500块一个的工程师,一条狗咬死一片的那种。。。”
5、[Go] Http / Net 相关资料


长按二维码向我转账

受苹果公司新规定影响,微信 iOS 版的赞赏功能被关闭,可通过二维码转账支持公众号。

    阅读
    好看
    已推荐到看一看
    你的朋友可以在“发现”-“看一看”看到你认为好看的文章。
    已取消,“好看”想法已同步删除
    已推荐到看一看 和朋友分享想法
    最多200字,当前共 发送

    已发送

    朋友将在看一看看到

    确定
    分享你的想法...
    取消

    分享想法到看一看

    确定
    最多200字,当前共

    发送中

    网络异常,请稍后重试

    微信扫一扫
    关注该公众号





    联系我们

    欢迎来到TinyMind。

    关于TinyMind的内容或商务合作、网站建议,举报不良信息等均可联系我们。

    TinyMind客服邮箱:support@tinymind.net.cn