20190616 权值线段树

线段树咕咕咕

我来写一个好写的权值线段树的解析吧

权值线段树是什么

线段树每个点维护的是点的值,而权值线段树叶子点维护的是一个数出现的次数,父节点维护的是它代表的区间里的数出现的次数的和

权值线段树基本操作

其实权值线段树的基本操作与线段树没有太大的不同

建树

注意:由于一个叶子节点代表一个数出现的次数,所以,在数的范围非常大时,我们常常需要把这些数排个序,按排序重新赋值(因为权值线段树里数本身的值不重要,我们只需要关注它出现的次数),这也叫做离散化

void build(int l,int r,int ro)
{
    tr[ro].l = l,tr[ro].r = r;
    if(l == r)tr[ro].w = 0;
    else
    {
        int mid = (l + r) / 2;
        build(l,mid,2 * ro);
        build(mid + 1,r,2 * ro + 1);
        tr[ro].w = tr[2 * ro].w + tr[2 * ro + 1].w;
    }
}

添加

就是找到代表这个数的叶子节点,将它的权值++

void add(int x,int ro)
{
    if(tr[ro].l == tr[ro].r && tr[ro].l == x)tr[ro].w ++;
    else
    {
        int mid = (tr[ro].l + tr[ro].r) / 2;
        if(x <= mid)add(x,2 * ro);
        else if(x > mid)add(x,2 * ro + 1);
        tr[ro].w = tr[2 * ro].w + tr[2 * ro + 1].w;
    }
}

查询出现次数

例题:luogu p1903 逆序对

查询和线段树的查询操作一样,

若节点代表的区间与要查询的区间对应,返回它的权值

若要查询的区间处于节点左子节点的区间内,去它的左子节点查询

若要查询的区间处于节点右子节点的区间内,去它的右子节点查询

若要查询的区间横跨节点左右子节点的范围,那分别去左右子节点查询,并加和

int find(int l,int r,int ro)
{
    if(tr[ro].l == l && tr[ro].r == r)return tr[ro].w;
    else
    {
        int mid = (tr[ro].l + tr[ro].r) / 2;
        if(r <= mid)return find(l,r,2 * ro);
        else if(l > mid)return find(l,r,2 * ro + 1);
        else return find(l,mid,2 * ro) + find(mid + 1,r,2 * ro + 1);
    }
}

查找第k大的数

例题:luogu p3332 [ZJOI2013]K大数查询(这是我在洛谷里能找到的有关系的题)

查找第k大的数,也就相当于查找第kk(kk = n- k + 1)小的数

若kk<节点左子节点的范围,在左子节点里找就好了

若kk>节点左子节点的范围,说明这个数在右子节点里,到右子节点找,但是,需要注意的是,kk的值要减去左子节点的范围,代表我们要在右子树里找第kk - tr[zuo].w + 1小的数

若查找到叶子节点,那叶子节点代表的数即为所求

int numk(int l,int r,int ro,int k)
{
    if(l == r)return l;
    else
    {
        int mid = (l + r) / 2;
        if(k <= tr[2 * ro].w)return numk(l,mid,2 * ro,k);
        else return numk(mid + 1,r,2 * ro + 1,k - tr[ro].w);
    }
}

贴一个40分逆序对代码

#include<cstdio>
#include<cmath>
#include<algorithm>
using namespace std;
struct in
{
    int w;
    int k;
}a[500005];
struct node
{
    int l,r,w;
}tr[4000005];
void build(int l,int r,int ro)
{
    tr[ro].l = l,tr[ro].r = r;
    if(l == r)tr[ro].w = 0;
    else
    {
        int mid = (l + r) / 2;
        build(l,mid,2 * ro);
        build(mid + 1,r,2 * ro + 1);
        tr[ro].w = tr[2 * ro].w + tr[2 * ro + 1].w;
    }
}
int find(int l,int r,int ro)
{
    if(tr[ro].l == l && tr[ro].r == r)return tr[ro].w;
    else
    {
        int mid = (tr[ro].l + tr[ro].r) / 2;
        if(r <= mid)return find(l,r,2 * ro);
        else if(l > mid)return find(l,r,2 * ro + 1);
        else return find(l,mid,2 * ro) + find(mid + 1,r,2 * ro + 1);
    }
}
void add(int x,int ro)
{
    if(tr[ro].l == tr[ro].r && tr[ro].l == x)tr[ro].w ++;
    else
    {
        int mid = (tr[ro].l + tr[ro].r) / 2;
        if(x <= mid)add(x,2 * ro);
        else if(x > mid)add(x,2 * ro + 1);
        tr[ro].w = tr[2 * ro].w + tr[2 * ro + 1].w;
    }
}
bool cmp(in x,in y)
{
    return x.w < y.w;
}
bool cmpp(in x,in y)
{
    return x.k < y.k;
}
int main()
{
    int n;
    int ans = 0;
    scanf("%d",&n);
    for(int i = 1;i <= n;i ++)
    {
        scanf("%d",&a[i].w);
        a[i].k = i;
    }
    sort(a + 1,a + 1 + n,cmp);
    for(int i = 1;i <= n;i ++)
    {
        a[i].w = i;
    }
    sort(a + 1,a + 1 + n,cmpp);
    build(1,n,1);
    for(int i = 1;i <= n;i ++)
    {
        ans += find(a[i].w,n,1);
        add(a[i].w,1);
    }
    printf("%d",ans);
    return 0;
}

欢迎大佬指出错误啊

原文地址:https://www.cnblogs.com/djfuuxjz/p/11030217.html

时间: 2024-10-21 10:59:28

20190616 权值线段树的相关文章

[bzoj3932][CQOI2015]任务查询系统-题解[主席树][权值线段树]

Description 最近实验室正在为其管理的超级计算机编制一套任务管理系统,而你被安排完成其中的查询部分.超级计算机中的 任务用三元组(Si,Ei,Pi)描述,(Si,Ei,Pi)表示任务从第Si秒开始,在第Ei秒后结束(第Si秒和Ei秒任务也在运行 ),其优先级为Pi.同一时间可能有多个任务同时执行,它们的优先级可能相同,也可能不同.调度系统会经常向 查询系统询问,第Xi秒正在运行的任务中,优先级最小的Ki个任务(即将任务按照优先级从小到大排序后取前Ki个 )的优先级之和是多少.特别的,如

P2234 [HNOI2002]营业额统计 (权值线段树)

P2234 [HNOI2002]营业额统计 题目描述 Tiger最近被公司升任为营业部经理,他上任后接受公司交给的第一项任务便是统计并分析公司成立以来的营业情况. Tiger拿出了公司的账本,账本上记录了公司成立以来每天的营业额.分析营业情况是一项相当复杂的工作.由于节假日,大减价或者是其他情况的时候,营业额会出现一定的波动,当然一定的波动是能够接受的,但是在某些时候营业额突变得很高或是很低,这就证明公司此时的经营状况出现了问题.经济管理学上定义了一种最小波动值来衡量这种情况: 当最小波动值越大

【BZOJ4605】崂山白花蛇草水 权值线段树+kd-tree

[BZOJ4605]崂山白花蛇草水 Description 神犇Aleph在SDOI Round2前立了一个flag:如果进了省队,就现场直播喝崂山白花蛇草水.凭借着神犇Aleph的实力,他轻松地进了山东省省队,现在便是他履行诺言的时候了.蒟蒻Bob特地为他准备了999,999,999,999,999,999瓶崂山白花蛇草水,想要灌神犇Aleph.神犇Aleph求(跪着的)蒟蒻Bob不要灌他,由于神犇Aleph是神犇,蒟蒻Bob最终答应了他的请求,但蒟蒻Bob决定将计就计,也让神犇Aleph回答

【bzoj2161】布娃娃 权值线段树

题目描述 小时候的雨荨非常听话,是父母眼中的好孩子.在学校是老师的左右手,同学的好榜样.后来她成为艾利斯顿第二代考神,这和小时候培养的良好素质是分不开的.雨荨的妈妈也为有这么一个懂事的女儿感到高兴.一次期末考试,雨荨不知道第多少次,再次考了全年级第一名.雨荨的妈妈看到女儿100分的成绩单时,脸上又泛起了幸福的笑容,作为奖励,她给雨荨买了n个布娃娃.细心的雨荨发现,第i个布娃娃有一个耐心值P[i]以及一个魅力值C[i],并且还有能够忍受的耐心值的上限R[i]以及下限L[i].当一个布娃娃j满足L[

模板——权值线段树(逆序对)

Ultra-QuickSort Time Limit: 7000MS   Memory Limit: 65536K Total Submissions: 62455   Accepted: 23259 Description In this problem, you have to analyze a particular sorting algorithm. The algorithm processes a sequence of n distinct integers by swappin

【bzoj4605】崂山白花蛇草水 权值线段树套KD-tree

题目描述 神犇Aleph在SDOI Round2前立了一个flag:如果进了省队,就现场直播喝崂山白花蛇草水.凭借着神犇Aleph的实力,他轻松地进了山东省省队,现在便是他履行诺言的时候了.蒟蒻Bob特地为他准备了999,999,999,999,999,999瓶崂山白花蛇草水,想要灌神犇Aleph.神犇Aleph求(跪着的)蒟蒻Bob不要灌他,由于神犇Aleph是神犇,蒟蒻Bob最终答应了他的请求,但蒟蒻Bob决定将计就计,也让神犇Aleph回答一些问题.具体说来,蒟蒻Bob会在一个宽敞的广场上

【bzoj3065】带插入区间K小值 替罪羊树套权值线段树

题目描述 从前有n只跳蚤排成一行做早操,每只跳蚤都有自己的一个弹跳力a[i].跳蚤国王看着这些跳蚤国欣欣向荣的情景,感到非常高兴.这时跳蚤国王决定理性愉悦一下,查询区间k小值.他每次向它的随从伏特提出这样的问题: 从左往右第x个到第y个跳蚤中,a[i]第k小的值是多少.这可难不倒伏特,他在脑袋里使用函数式线段树前缀和的方法水掉了跳蚤国王的询问.这时伏特发现有些跳蚤跳久了弹跳力会有变化,有的会增大,有的会减少.这可难不倒伏特,他在脑袋里使用树状数组套线段树的方法水掉了跳蚤国王的询问.(orz 主席

【bzoj2770】YY的Treap 权值线段树

题目描述 志向远大的YY小朋友在学完快速排序之后决定学习平衡树,左思右想再加上SY的教唆,YY决定学习Treap.友爱教教父SY如砍瓜切菜般教会了YY小朋友Treap(一种平衡树,通过对每个节点随机分配一个priority,同时保证这棵平衡树关于priority是一个小根堆以保证效率).这时候不怎么友爱的510跑了出来,他问了YY小朋友一个极不和谐的问题:怎么求Treap中两个点之间的路径长度.YY秒了之后决定把这个问题交给你来做,但只要求出树中两点的LCA. 输入 第一行两个整数n,m 第二行

权值线段树求逆序对问题

我们都知道,求逆序对数量可以用归并排序解决.但是用归并排序只能解决静态的序列问题,没有扩展的区间.因此就有了用权值线段树求逆序对的方法. 1 #include<iostream> 2 #include<iomanip> 3 #include<ctime> 4 #include<climits> 5 #include<algorithm> 6 #include<queue> 7 #include<vector> 8 #inc