博客
关于我
树状数组(单点修改区间查询、区间修改单点查询、区间修改区间查询)
阅读量:274 次
发布时间:2019-03-01

本文共 2808 字,大约阅读时间需要 9 分钟。

数据结构三种形式的模板应用

模板一:单点修改与区间查询

在数据处理领域,单点修改与区间查询的需求经常出现。传统的解决方案可能需要逐个修改或遍历查询,这在大数据量下效率低下。通过离线处理和差分数组的技术,可以实现高效的在线操作。

差分数组的原理

差分数组是一种将区间操作转换为单点操作的巧妙方法。对于数组 a,其差分数组 d 满足关系式:

[ a_n = \sum_{i=1}^n d_i ]

同时,前缀和可以表示为:

[ \sum_{i=1}^n a_i = \sum_{i=1}^n \sum_{j=1}^i d_j = \sum_{i=1}^n (n - i + 1) d_i ]

因此,为了支持区间查询,我们需要维护两个树状数组:

  • c1 用于存储差分数组 d_i 的值。
  • c2 用于存储 d_i * i 的值。

代码实现

#include 
#include
#include
#include
#include
#include
using namespace std;long long c[200005][2];int n, q;int lowbit(int x) { return x & (-x);}void add(int pos, int x, int f) { while (pos <= n) { c[pos][f] += x; pos += lowbit(pos); }}long long query(int pos, int f) { long long res = 0; while (pos > 0) { res += c[pos][f]; pos -= lowbit(pos); } return res;}long long ask(int pos) { long long sum = (pos + 1) * query(pos, 0) - query(pos, 1); return sum;}int main() { scanf("%d", &n); long long a = 0, b; for (int i = 1; i <= n; ++i) { scanf("%lld", &b); add(i, b - a, 0); add(i, (b - a) * i, 1); a = b; } scanf("%d", &q); for (int i = 1; i <= q; ++i) { int opt; scanf("%d", &opt); if (opt == 1) { int x, y, k; scanf("%d %d %d", &x, &y, &k); add(x, k, 0); add(y + 1, -k, 0); add(x, x * k, 1); add(y + 1, -(y + 1) * k, 1); } else if (opt == 2) { int x, y; scanf("%d %d", &x, &y); printf("%lld\n", ask(y) - ask(x - 1)); } } return 0;}

模板二:区间修改与单点查询

在某些场景下,我们需要对多个位置进行相同的区间修改,并快速查询某个点的最终值。这种情况可以通过前缀和和差分数组结合树状数组来实现。

树状数组的应用

树状数组(Fenwick Tree)在支持高效区间查询和更新操作方面表现优异。通过对差分数组进行离线处理,可以将区间修改转化为前缀和的形式,从而在在线查询时即可快速响应。

代码实现

#include 
#include
#include
#include
#include
using namespace std;int c[500005];int n, m;int lowbit(int x) { return (-x) & x;}void add(int pos, int x) { while (pos <= n) { c[pos] += x; pos += lowbit(pos); }}int query(int pos) { int res = 0; while (pos > 0) { res += c[pos]; pos -= lowbit(pos); } return res;}int main() { scanf("%d %d", &n, &m); int x = 0, y; for (int i = 1; i <= n; ++i) { scanf("%d", &y); add(i, y - x); x = y; } for (int i = 1; i <= m; ++i) { int opt, x, y, k; scanf("%d", &opt); if (opt == 1) { scanf("%d %d %d", &x, &y, &k); add(x, k); add(y + 1, -k); } else if (opt == 2) { scanf("%d", &x); printf("%d\n", query(x)); } } return 0;}

模板三:区间修改与区间查询

对于需要对多个区间进行操作并查询某个区间的最终结果,差分数组和树状数组的组合提供了高效的解决方案。这种方法通过离线处理差分数组,将连续的区间操作转化为离散的点更新,从而在查询时快速计算所需结果。

优化思路

本文采用了差分数组的方法,将区间操作转化为前缀和的形式。通过维护两个树状数组,分别存储差分值和其加权和,实现了高效的区间查询和更新。这种方法在处理大量数据时表现优异,且代码结构清晰易懂。

总结

以上三种模板分别针对不同的操作需求,结合差分数组和树状数组技术,提供了高效且灵活的解决方案。选择具体的模板取决于实际的应用场景和数据处理需求。通过离线处理和差分数组技术,可以显著提升数据操作的效率,使系统在处理大规模数据时表现更为出色。

转载地址:http://jpio.baihongyu.com/

你可能感兴趣的文章
NLP 基于kashgari和BERT实现中文命名实体识别(NER)
查看>>
NLP 模型中的偏差和公平性检测
查看>>
Vue3.0 性能提升主要是通过哪几方面体现的?
查看>>
NLP 项目:维基百科文章爬虫和分类【01】 - 语料库阅读器
查看>>
NLP_什么是统计语言模型_条件概率的链式法则_n元统计语言模型_马尔科夫链_数据稀疏(出现了词库中没有的词)_统计语言模型的平滑策略---人工智能工作笔记0035
查看>>
NLP三大特征抽取器:CNN、RNN与Transformer全面解析
查看>>
NLP学习笔记:使用 Python 进行NLTK
查看>>
NLP度量指标BELU真的完美么?
查看>>
NLP的不同研究领域和最新发展的概述
查看>>
NLP的神经网络训练的新模式
查看>>
NLP采用Bert进行简单文本情感分类
查看>>
NLP问答系统:使用 Deepset SQUAD 和 SQuAD v2 度量评估
查看>>
NLP项目:维基百科文章爬虫和分类【02】 - 语料库转换管道
查看>>
NLP:使用 SciKit Learn 的文本矢量化方法
查看>>
nmap 使用方法详细介绍
查看>>
Nmap扫描教程之Nmap基础知识
查看>>
nmap指纹识别要点以及又快又准之方法
查看>>
Nmap渗透测试指南之指纹识别与探测、伺机而动
查看>>
Nmap端口扫描工具Windows安装和命令大全(非常详细)零基础入门到精通,收藏这篇就够了
查看>>
NMAP网络扫描工具的安装与使用
查看>>