【算法 高级数据结构】树状数组:一种高效的数据结构(二)

🚀个人主页:为梦而生~ 关注我一起学习吧!
💡专栏:算法题、 基础算法、数据结构~赶紧来学算法吧
💡往期推荐
【算法基础 & 数学】快速幂求逆元(逆元、扩展欧几里得定理、小费马定理)
【算法基础】深搜
数据结构各内部排序算法总结对比及动图演示(插入排序、冒泡和快速排序、选择排序、堆排序、归并排序和基数排序等)
【算法 & 高级数据结构】树状数组:一种高效的数据结构(一)

上一篇文章我们介绍了树状数组这个数据结构,并且进行了其原理的数学推导,这篇文章基于上一篇文章,来讲一下这个数据结构在算法题中的应用

还不知道树状数组是什么的来看这篇文章:【算法 & 高级数据结构】树状数组:一种高效的数据结构(一)

题目来源:AcWing


文章目录

  • 0 通用模板
  • 1 单点修改,区间求和
  • 2 区间增减,单点查询
  • 3 区间增减,区间求和


0 通用模板

首先,通用模板先摆上

//lowbit
int lowbit(int x){return x & -x;
}//修改操作
void add(int x, int c){for(int i = x; i <= n; i += lowbit(i)) tr[i] += c;
}//查询操作
int sum(int x){int res = 0;for(int i = x; i; i -= lowbit(i)) res += tr[i];return res;
}

1 单点修改,区间求和

  • 例题
    在这里插入图片描述

输入格式
第一行一个数 n n n

第二行是 n n n
个数,分别代表 y 1 , y 2 , … , y n y_1,y_2,…,y_n y1y2,,yn

输出格式
两个数,中间用空格隔开,依次为 V 的个数和 的个数。

数据范围
对于所有数据, n ≤ 200000 n≤200000 n200000,且输出答案不会超过 int64。
y 1 ∼ y n y_1∼y_n y1yn 1 1 1 n n n 的一个排列。

输入样例

5
1 5 3 2 4

输出样例

3 4
  • 主要思路

因为两个符号的特点对应的每个点的纵坐标有这样的特点: V 意味着两边点的纵坐标高于当前点的纵坐标, 意味着两边的纵坐标小于当前的纵坐标。所以我们只需要遍历两次数组,记录每个点左右高于或低于当前点的纵坐标的点的数量,然后利用排列组合,计算出总数。

以下题解来源:https://www.acwing.com/solution/content/13818/

  1. 从左向右依次遍历每个数a[i],使用树状数组统计在i位置之前所有比a[i]大的数的个数、以及比a[i]小的数的个数。
    统计完成后,将a[i]加入到树状数组。
  2. 从右向左依次遍历每个数a[i],使用树状数组统计在i位置之后所有比a[i]大的数的个数、以及比a[i]小的数的个数。
    统计完成后,将a[i]加入到树状数组。
  • 代码
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;typedef long long LL;const int N = 200010;
int n;
int a[N], tr[N];
int Greater[N], Lower[N];int lowbit(int x){return x & -x;
}//修改操作
void add(int x, int c){for(int i = x; i <= n; i += lowbit(i)) tr[i] += c;
}//查询操作
int sum(int x){int res = 0;for(int i = x; i; i -= lowbit(i)) res += tr[i];return res;
}int main(){cin >> n;for(int i = 1; i <= n; i++) scanf("%d", &a[i]);//将区间分成n份,看每一份左边有多少个大于它和小于它的for(int i = 1; i <= n; i++){int y = a[i];Greater[i] = sum(n) - sum(y);Lower[i] = sum(y - 1);add(y, 1);}//memset一下tr数组,倒着求每一份右边有多少个小于和大于它的memset(tr, 0, sizeof(tr));LL res1 = 0, res2 = 0;for(int i = n; i; i--){int y = a[i];res1 += Greater[i] * (LL)(sum(n) - sum(y));res2 += Lower[i] * (LL)sum(y - 1);add(y, 1);}cout << res1 << " " << res2 << endl;return 0;
}

2 区间增减,单点查询

主要思想:建立差分数组

  • 例题

在这里插入图片描述
数据范围
1 ≤ N , M ≤ 1 0 5 , 1≤N,M≤10^5, 1N,M105,
∣ d ∣ ≤ 10000 , |d|≤10000, d10000,
∣ A [ i ] ∣ ≤ 1 0 9 |A[i]|≤10^9 A[i]109

输入样例

10 5
1 2 3 4 5 6 7 8 9 10
Q 4
Q 1
Q 2
C 1 6 3
Q 2

输出样例

4
1
2
5
  • 主要思路

树状数组每次修改对应的是某个范围的前缀和的值的修改,所以如果需要在大数据量的情况下进行区间修改操作,大概率会TLE的。

但是我们想到,区间操作我们可以利用最基础的差分,使得区间操作优化成 O ( 1 ) O(1) O(1)的复杂度。

可以看到,将题目在原始组上的操作,转换到差分数组上,和树状数组解决的问题一致。

  • 代码
#include<iostream>
#include<algorithm>
using namespace std;typedef long long LL;const int N = 100010;
int tr[N], a[N];
int n, m;int lowbit(int x){return x & -x;
}void add(int x, int c){for(int i = x; i <= n; i += lowbit(i)) tr[i] += c;
}LL quary(int x){LL res = 0;for(int i = x; i; i -= lowbit(i)) res += tr[i];return res;
}int main(){cin >> n >> m;for(int i = 1; i <= n; i++){cin >> a[i];add(i, a[i] - a[i - 1]);}while(m--){char op;cin >> op;if(op == 'C'){int l, r, d;cin >> l >> r >> d;add(l, d), add(r + 1, -d);}else{int x;cin >> x;cout << quary(x)<< endl;}}return 0;
}

3 区间增减,区间求和

主要思想:建立差分数组+公式

  • 例题

在这里插入图片描述
数据范围
1 ≤ N , M ≤ 1 0 5 , 1≤N,M≤10^5, 1N,M105,
∣ d ∣ ≤ 10000 , |d|≤10000, d10000,
∣ A [ i ] ∣ ≤ 1 0 9 |A[i]|≤10^9 A[i]109

输入样例

10 5
1 2 3 4 5 6 7 8 9 10
Q 4 4
Q 1 10
Q 2 4
C 3 6 3
Q 2 4

输出样例

4
55
9
15
  • 主要思想

上一个应用虽然利用差分数组解决了区间操作的问题,但是区间操作完之后,利用差分数组不利于进行区间查询,所以需要进行一些推导,看看有什么性质可以利用。

数学推导见如下题解:https://www.acwing.com/solution/content/44886/
因此只需维护两个树状数组即可
一个是差分数组d[i]的树状数组tr[i],还有一个是i*d[i]的树状数组tri[i]

  • 代码
#include<iostream>
#include<algorithm>
using namespace std;typedef long long LL;const int N = 100010;
int a[N];   
LL tr1[N], tr2[N];  //tr1[i] : b[i]的前缀和  tr2[i] : i * b[i]的前缀和
int n, m;int lowbit(int x){return x & -x;
}void add(LL tr[], int x, LL c){for(int i = x; i <= n; i += lowbit(i)) tr[i] += c;
}LL quary(LL tr[], int x){LL res = 0;for(int i = x; i; i -= lowbit(i)) res += tr[i];return res;
}LL prefix_sum(int x){return quary(tr1, x) * (LL)(x + 1) - quary(tr2, x);
}int main(){cin >> n >> m;for(int i = 1; i <= n; i++){cin >> a[i];int b = a[i] - a[i - 1];add(tr1, i, (LL)b);add(tr2, i, (LL)i * b);}while(m--){char op;cin >> op;if(op == 'C'){int l, r, d;cin >> l >> r >> d;add(tr1, l, (LL)d), add(tr1, r + 1, (LL)-d);add(tr2, l, (LL)l * d), add(tr2, r + 1, (LL)(r + 1) * -d);}else{int l, r;cin >> l >> r;cout << prefix_sum(r) - prefix_sum(l - 1) << endl;}}return 0;
}

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.mzph.cn/news/769630.shtml

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

RTthread如何引入webclient和cjson来编写自己的模块代码||SecureCRT的安装与激活||安装VScode

目录 1.RTthread如何引入webclient和cjson来编写自己的模块代码 2.SecureCRT的安装与激活 3.static与const的区别 4.安装VScode 1.RTthread如何引入webclient和cjson来编写自己的模块代码 以我自己的工程为例&#xff1a; 首先将新引入的模块在applicatons下新建cpeinfo文件…

【MySQL】一条 SQL 查询语句在数据库中的执行流程 | SQL语句中各个关键字的执行顺序

创作不易&#xff0c;本篇文章如果帮助到了你&#xff0c;还请点赞 关注支持一下♡>&#x16966;<)!! 主页专栏有更多知识&#xff0c;如有疑问欢迎大家指正讨论&#xff0c;共同进步&#xff01; 给大家跳段街舞感谢支持&#xff01;ጿ ኈ ቼ ዽ ጿ ኈ ቼ ዽ ጿ ኈ ቼ …

【HTML】打造你的网络名片:制作一个简单的HTML个人网页

目录 前言 流程 1. 规划你的网页内容 2. 选择合适的工具 3. 设计网页布局 HTML结构示例 4. 编写HTML代码 头部&#xff08;Header&#xff09; 关于我&#xff08;About Me&#xff09; 作品集&#xff08;Portfolio&#xff09; 博客/文章&#xff08;Blog/Article…

知识图表示学习中的负抽样研究综述

摘要 知识图表示学习(KGRL)或知识图嵌入(KGE)在知识构建和信息探索的人工智能应用中起着至关重要的作用。这些模型旨在将知识图中的实体和关系编码到低维向量空间中。在KGE模型的训练过程中&#xff0c;使用正样本和负样本是区分的必要条件。然而&#xff0c;直接从现有的知识…

Linux: network:interrupt: python tool

一个小工具可以用来统计interrupt dump出来的数据。然后再导入到excel里倒是可以看出 中断的趋势。 import sys import glob import re file_name=sys.argv[1] f_trace = open(file_name, r) f_message = open(file_name+.out,

鸿蒙NXET实战:高德地图定位SDK【获取Key+获取定位数据】(二)

如何申请key 1、创建新应用 进入[控制台]&#xff0c;创建一个新应用。如果您之前已经创建过应用&#xff0c;可直接跳过这个步骤。 2、添加新Key 在创建的应用上点击"添加新Key"按钮&#xff0c;在弹出的对话框中&#xff0c;依次&#xff1a;输入应用名名称&…

Muduo类详解之EventLoop

最核⼼的部分就是 EventLoop 、 Channel 以及 Poller 三个类&#xff0c;其中 EventLoop 可以看作是对业务线程的封装&#xff0c;⽽ Channel 可以看作是对每个已经建⽴连接的封装&#xff08;即 accept(3) 返回的⽂件描述符&#xff09; EventLoop class EventLoop { p…

解决SLF4J: Class path contains multiple SLF4J bindings.

JDK版本&#xff1a;jdk17 IDEA版本&#xff1a;IntelliJ IDEA 2022.1.3 SpringBoot 版本&#xff1a;v2.5.7 maven版本&#xff1a;3.6.3 文章目录 问题描述&#xff1a;原因分析&#xff1a;解决方案&#xff1a;参考资料&#xff1a; 问题描述&#xff1a; 当SpringBoot项目…

并发VS并行

参考文章 面试必考的&#xff1a;并发和并行有什么区别&#xff1f; 并发&#xff1a;一个人同时做多件事&#xff08;射击游戏队友抢装备&#xff09; 并行&#xff1a;多人同时处理同一件事&#xff08;射击游戏敌人同时射击对方&#xff09;

系统学习Python——装饰器:“私有“和“公有“属性案例-[Python不是关于控制]

分类目录&#xff1a;《系统学习Python》总目录 既然我已经用如此大的篇幅实现了Python代码的Private和Public属性声明&#xff0c;必须再次提醒&#xff0c;像这样为类添加访问控制不完全是Python的风格。实际上&#xff0c;除了充当装饰器使用的一个展示&#xff0c;大多数程…

解决ros2缺少cv_bridge包的问题——cv_bridge源码编译安装过程

安装流程 注意&#xff1a;cv_bridge 依赖boost 和 opencv 首先&#xff0c;和编译ros2功能包一样&#xff0c;先创建一个工作空间 mkdir -p cv_bridge_ws/src cd cv_bridge_ws/src然后&#xff0c;在src目录下&#xff0c;下载vision_opencv的源码,注意将-b galactic替换成…

制作 configure 文件极简教程

The magic behind configure, make, make install - How it works in Unix (thoughtbot.com) 如果要发布自己的源码&#xff0c;而自己的源码想通过configure&#xff0c;make&#xff0c;make install 编译安装&#xff0c;那么要为自己的源码建立 configure 文件&#xff0…

uniapp 安卓跳转到应用设置等页面

// #ifdef APP-PLUS const platform uni.getSystemInfoSync().platform if (platform android) {var main plus.android.runtimeMainActivity(); //获取activityvar pkName main.getPackageName();var Intent plus.android.importClass(android.content.Intent); var Set…

学习数据结构:算法的时间复杂度和空间复杂度

一、算法的复杂度 衡量一个算法的好坏&#xff0c;一般是从时间和空间两个维度来衡量的&#xff0c;即时间复杂度和空间复杂度。 时间复杂度主要衡量一个算法的运行快慢&#xff0c;而空间复杂度主要衡量一个算法运行所需要的额外空间。 算法的时间复杂度 算法中的基本操作的…

SAP BAS中Fiori开发的高阶功能(storyboard, navigation, guided development, variant)

1. 前言 在之前的几篇文章中&#xff0c;我介绍了SAP BAS的一些基本功能&#xff0c;包括账户申请&#xff0c;创建工作区&#xff0c;git的使用以及如何step-by-step去创建出你的第一个Fiori项目等等。在本篇中&#xff0c;我将进一步介绍一些在开发Fiori应用程序时会用到的高…

JAVA学习笔记19(面向对象编程)

1.面向对象编程 1.1 类与对象 1.类与对象的概念 ​ *对象[属性]/[行为] ​ *语法 class cat {String name;int age; }main() {//cat1就是一个对象//创建一只猫Cat cat1 new Cat();//给猫的属性赋值cat1.name "123";cat1.age 10; }​ *类是抽象的&#xff0c;…

前端使用正则表达式进行校验

一、定义 设计思想是用一种描述性的语言定义一个规则&#xff0c;凡是符合规则的字符串&#xff0c;我们就认为它“匹配”了&#xff0c;否则&#xff0c;该字符串就是不合法的。 在 JavaScript中&#xff0c;正则表达式也是对象&#xff0c;构建正则表达式有两种方式&#x…

【可用Claude Opus模型】Claude3国内镜像站,亲测完全超越GPT-4(可用Claude Opus,官网价值20刀)

#今天在知乎看到一个问题&#xff1a;“平民不参与内测的话没有账号还有机会使用Claude 3吗&#xff1f;” 从去年GPT大火到现在&#xff0c;关于GPT的消息铺天盖地&#xff0c;真要有心想要去用&#xff0c;途径很多&#xff0c;别的不说&#xff0c;国内GPT的镜像站到处都是…

如何实现一个自己的脚手架工具

如何实现一个自己的脚手架工具 创建自定义全局命令获取命令行的交互参数终端交互下载远程项目代码 创建自定义全局命令 在项目文件夹中创建一个bin文件夹&#xff0c;再去npm init 初始化项目&#xff0c;生成的package.json 里面就有一个bin的选项参数指定了命令和执行文件路…

有效的括号字符串

只想用一个栈解决&#xff0c;可惜一直解决不了&#xff0c;因为 * 的问题无法解决 看到标答用了两个栈&#xff0c;一个栈存储 ( 另一个栈储存 * 思路&#xff1a;当s[k] 为 ( 或 * 进栈 &#xff0c; 当s[k] 为 ) 时 &#xff0c; 优先左括号出栈&#xff08;因为*有…