树状数组

最新推荐文章于 2024-07-23 18:35:33 发布
最新推荐文章于 2024-07-23 18:35:33 发布
阅读量113 收藏
点赞数
分类专栏： 算法 文章标签： c++
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/weixin_51009975/article/details/119646093
版权

引入：用数组模拟树形结构，可以解决一些区间更新，求和的问题。比线段树更简单一些的数据结构，树状数组能够解决的问题，线段树（还没学）都可以解决。

原理啥的，我也讲不明白，看大佬博客

构建树状数组

单点更新，区间查询

int n,a[1000010],c[1000010];

int lowbit(int x){
	return x&-x;//返回二进制最后一位1
}

void updata(int i,int k){//在i的位置上加上k
	for(;i<=n;i+=lowbit(i)) c[i]+=k;
}

int getsum(int i){//求a[1]-a[i]的和
	int ans=0;
	for(;i>0;i-=lowbit(i)) ans+=c[i];
	return ans;
}

来一个模板题

HDU1166

代码：

#include<bits/stdc++.h>
using namespace std;

int n,a[1000010],c[1000010];

int lowbit(int x){
	return x&-x;
}

void updata(int i,int k){
	for(;i<=n;i+=lowbit(i)) c[i]+=k;
}

int getsum(int i){
	int ans=0;
	for(;i>0;i-=lowbit(i)) ans+=c[i];
	return ans;
}


int main(){
	int t;
	cin>>t;
	for(int tot=1;tot<=t;tot++){
		cout<<"Case "<<tot<<":"<<endl;
		memset(a,0,sizeof a);
		memset(c,0,sizeof c);
		cin>>n;
		for(int i=1;i<=n;i++) {
			cin>>a[i];
			updata(i,a[i]);
		}
		string s;
		int x,y;
		while(cin>>s&&s[0]!='E'){
			cin>>x>>y;
			if(s[0]=='Q'){
				int sum=getsum(y)-getsum(x-1);
				cout<<sum<<endl;
			}
			else if(s[0]=='S'){
				updata(x,-y);
			}
			else if(s[0]=='A'){
				updata(x,y);
			}
		}
	}
	
}

我的板子好像有问题，以后都不敢用了，TLE一下午不知道咋回事，世界未解之谜，代码还是写简单一些不加乱七八糟的东西，不然debug浪费时间

上面这种情况是单点更新、区间查询

单点更新，单点查询

普通数组就可以了。

区间更新、单点查询

给某一段区间同时加上一个数，最后加之后某个点的值，我们知道在某一个区间加上一个数，可以用差分，这里我们用差分数组建立树状数组。

int n,a[1000010],c[1000010];//a原数组，c差分数组

inline int lowbit(int x){return x&-x;}

void updata(int i,int k){
	for(;i<=n;i+=lowbit(i)) c[i]+=k;
}

int getsum(int i){
	int ans=0;
	for(;i>0;i-=lowbit(i)) ans+=c[i];
	return ans;
}

void add(int l,int r,int k){//在区间[l-r]加上一个数k
	updata(l,k);
	updata(r+1,-k);
}

洛谷P4939

模板题，给你两个操作，0，在给你区间+1，1输出某个点的值。

#include<bits/stdc++.h>
using namespace std;
#define int long long
const int N=1000010;
int n,a[10000010],c[10000010];//a原数组，c差分数组

inline int lowbit(int x){return x&-x;}

void updata(int i,int k){
    for(;i<=n;i+=lowbit(i)) c[i]+=k;
}

int getsum(int i){
    int ans=0;
    for(;i>0;i-=lowbit(i)) ans+=c[i];
    return ans;
}

void add(int l,int r,int k){//在区间[l-r]加上一个数k
    updata(l,k);
    updata(r+1,-k);
}

signed main(){
	int m;
	scanf("%lld%lld",&n,&m);
	while(m--){
		int op,l,r;
		scanf("%lld",&op);
		if(op&1){
			scanf("%lld",&l);
			int ans=getsum(l);
			printf("%lld\n",ans);
		}
		else {
			scanf("%lld%lld",&l,&r);
			add(l,r,1);
		}
	}
}

洛谷P5057

跟上一题类似，判断最后的奇偶就行了。

#include<bits/stdc++.h>
using namespace std;
#define int long long
const int N=1000010;
int n,a[10000010],c[10000010];//a原数组，c差分数组

inline int lowbit(int x){return x&-x;}

void updata(int i,int k){
    for(;i<=n;i+=lowbit(i)) c[i]+=k;
}

int query(int i){
    int ans=0;
    for(;i>0;i-=lowbit(i)) ans+=c[i];
    return ans;
}

void add(int l,int r,int k){//在区间[l-r]加上一个数k
    updata(l,k);
    updata(r+1,-k);
}

signed main(){
	int m;
	scanf("%lld%lld",&n,&m);
	while(m--){
		int op,l,r;
		scanf("%lld",&op);
		if(op==2){
			scanf("%lld",&l);
			int ans=getsum(l)&1;
			printf("%lld\n",ans);
		}
		else {
			scanf("%lld%lld",&l,&r);
			add(l,r,1);
		}
	}
}

区间更新，区间查询

const int N=1000010;
int n,a[1000010],c[1000010];//a原数组，c差分数组
int b1[N],b2[N];
inline int lowbit(int x){return x&-x;}

void updata(int i,int k){
    int tmp=i*k;
    for(;i<=n;i+=lowbit(i)) b1[i]+=k,b2[i]+=tmp;
}

int query(int b[],int i){
    int ans=0;
    for(;i>0;i-=lowbit(i)) ans+=b[i];
    return ans;
}

void add(int l,int r,int k){//在区间[l-r]加上一个数k
    updata(l,k);
    updata(r+1,-k);
}

洛谷P2357

做了这一题发现，区间更新，区间查询也可以用于做单点更新单点查询，区间更新，单点查询。
这一题5个操作，1给区间都加一个数，2给 a 1 a_1 a1​加上一个数，3给 a 1 a_{1} a1​减一个数，4求一个区间的和，5求 a 1 a_{1} a1​的值。

代码：

#include<bits/stdc++.h>
using namespace std;
#define int long long
const int N=1000010;
int n,a[1000010],c[1000010];//a原数组，c差分数组
int b1[N],b2[N];
inline int lowbit(int x){return x&-x;}

void updata(int i,int k){
    int tmp=i*k;
    for(;i<=n;i+=lowbit(i)) b1[i]+=k,b2[i]+=tmp;
}

int query(int b[],int i){
    int ans=0;
    for(;i>0;i-=lowbit(i)) ans+=b[i];
    return ans;
}

void add(int l,int r,int k){//在区间[l-r]加上一个数k
    updata(l,k);
    updata(r+1,-k);
}

int query(int l,int r){
    return (r+1)*query(b1,r)-query(b2,r)-(l*query(b1,l-1)-query(b2,l-1));
}

signed main(){
	int f;
	cin>>n>>f;
	for(int i=1;i<=n;i++){
		cin>>a[i];
		b1[i]+=a[i]-a[i-1];
		b2[i]+=i*(a[i]-a[i-1]);
		int j=i+lowbit(i);
		if(j<=n) b1[j]+=b1[i],b2[j]+=b2[i];
	}
	while(f--){
		int op,l,r,k;
		cin>>op;
		if(op==1){
			cin>>l>>r>>k;
			add(l,r,k);
		}
		else if(op==2){
			cin>>k;
			add(1,1,k);
		}
		else if(op==3){
			cin>>k;
			add(1,1,-k);
		}
		else if(op==4){
			cin>>l>>r;
			int ans=query(l,r);
			cout<<ans<<endl;
		}
		else{
			int ans=query(1,1);
			cout<<ans<<endl;
		}
	}
	return 0;
}

洛谷P1908

求逆序对。

树状数组+离散化

#include<bits/stdc++.h>
using namespace std;
#define int long long
#define sc(n) scanf("%lld",&n)
#define SC(a,b) scanf("%lld%lld",&a,&b)
#define pr(a) printf("%lld",a)
const int N=1000010;
int n,a[1000010],c[1000010],b[N];

int lowbit(int x){
    return x&-x;//返回二进制最后一位1
}

void updata(int i,int k){//在i的位置上加上k
    for(;i<=n;i+=lowbit(i)) c[i]+=k;
}

int getsum(int i){//求a[1]-a[i]的和
    int ans=0;
    for(;i>0;i-=lowbit(i)) ans+=c[i];
    return ans;
}


signed main(){
	sc(n);
	for(int i=1;i<=n;i++) sc(a[i]),b[i]=a[i];//离散化操作
	sort(b+1,b+1+n);
	int cnt=unique(b+1,b+1+n)-(b+1);
	for(int i=1;i<=n;i++){
		a[i]=lower_bound(b+1,b+1+n,a[i])-b;
	}
	int ans=0;
	for(int i=1;i<=n;i++){
		ans+=i-1-getsum(a[i]);
		updata(a[i],1);
	}
	pr(ans);
	return 0;
}

洛谷P1637 三元上升子序列

题意：Erwin 最近对一种叫 thair 的东西巨感兴趣。。。在含有 n 个整数的序列$ a_1,a_2,…,a_n ​ ​ 中 ， 三 个 数 被 称 作 ‘ t h a i r ‘ 当 且 仅 当 ​​中，三个数被称作`thair`当且仅当 ​​中，三个数被称作‘thair‘当且仅当i<j<k$ 且 a i < a j < a k a_i<a_j<a_k ai​<aj​<ak​;求一个序列中 thair 的个数。

输入：

4
2 1 3 4
5
1 2 2 3 4

输出：

2
7

DP+动态规划+离散化

  • 假设 f [ i ] [ j ] f[i][j] f[i][j]表示以 a [ j ] a[j] a[j]​​结尾的长度为 i 的上升子序列的个数。

  • 动态转移方程 f [ i ] [ j ] = ∑ k j f [ i − 1 ] [ k ] f[i][j]=\sum_k^j f[i-1][k] f[i][j]=∑kj​f[i−1][k] , a [ k ] < a [ j ] a[k]<a[j] a[k]<a[j];

  • 要满足两个性质：

    • k < j k<j k<j;
    • a [ k ] < a [ j ] ; a[k]<a[j]; a[k]<a[j];

求和我们可以用树状数组+离散化来维护，

代码：

#include<bits/stdc++.h>
using namespace std;
#define int long long
#define sc(n) scanf("%lld",&n)
#define SC(a,b) scanf("%lld%lld",&a,&b)
#define pr(a) printf("%lld",a)
const int N=1000010;
int n,a[1000010],c[1000010],b[N],m,ans,f[4][300010];
//f[i][j]表示以a[j]结尾的长度为i,上升子序列的个数
int lowbit(int x){
    return x&-x;//返回二进制最后一位1
}

void updata(int i,int k){//在i的位置上加上k
    for(;i<=n;i+=lowbit(i)) c[i]+=k;
}

int getsum(int i){//求a[1]-a[i]的和
    int ans=0;
    for(;i>0;i-=lowbit(i)) ans+=c[i];
    return ans;
}


signed main(){
	sc(n);
	for(int i=1;i<=n;i++) sc(a[i]),b[i]=a[i];//离散化操作
	sort(b+1,b+1+n);
	int cnt=unique(b+1,b+1+n)-(b+1);
	for(int i=1;i<=n;i++){
		f[1][i]=1;//长度为1的个数初始化为1
		a[i]=lower_bound(b+1,b+1+n,a[i])-b;//离散化
	}
	for(int i=2;i<=3;i++){
		memset(c,0,sizeof(c));
		for(int j=1;j<=n;j++){
			f[i][j]=getsum(a[j]-1);//动态转移方程
			updata(a[j],f[i-1][j]);
		}
	}
	for(int i=1;i<=n;i++){
		ans+=f[3][i];
	}
	pr(ans);
	return 0;
}

力扣1904

题意：给一个序列a,求出以每一个 a i a_i ai​结尾的最长上升子序列的长度。

树状数组+离散化

class Solution {
public:
    vector<int> tree;
    int n;
    int lowbit(int i) {
        return i & -i;
    }
    void update(int id, int x) { // 将id更新为x
        while (id <= n) {
            tree[id] = max(tree[id], x);
            id += lowbit(id);
        }
    }
    int query(int id) {
        int ans = 0;
        while (id) {
            ans = max(ans, tree[id]);
            id -= lowbit(id);
        }
        return ans;
    }
    vector<int> longestObstacleCourseAtEachPosition(vector<int>& obstacles) {
        vector<int> diff = obstacles; // 离散化数组
        sort(diff.begin(), diff.end());
        n = unique(diff.begin(), diff.end()) - diff.begin();
        tree.resize(n + 1, 0);
        vector<int> ans;
        for (auto& i : obstacles) {
            int idx = lower_bound(diff.begin(), diff.begin() + n, i) - diff.begin() + 1;
            int res = query(idx);
            ans.push_back(res + 1);
            update(idx, res + 1);
        }
        return ans;
    }
};
万伏小太阳
关注
  • 0
    点赞
  • 踩
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
专栏目录
树状数组(详细分析+应用)，看不懂打死我!
TheWayForDream
07-03 4万+
树状数组介绍 在学习一个算法之前一定要清楚它能干嘛，能解决什么样的问题，对你解题是否有帮助，然后才去学习它! 那么接下来看如下几个问题 什么是树状数组 顾名思义就是一个结构为树形结构的数组，于二叉树的结构类似但又不同，它是在二叉树的结构上删除了一些中间节点，来看两幅图就明白了. 1.这是二叉树的结构 2.这是树状数组的结构 不难发现，树状数组相比于二叉树删除了一些节点，但是为什么要删除呢？这就和树状数组的一些性质(lowbit)有关了，不懂没关系，继续往下看. 树状数组可以解决什么问题呢？ 可以解决大部
树状数组 java_树状数组详解
weixin_42220669的博客
02-16 1302
先来看几个问题吧。1.什么是树状数组？顾名思义，就是用数组来模拟树形结构呗。那么衍生出一个问题，为什么不直接建树？答案是没必要，因为树状数组能处理的问题就没必要建树。和Trie树的构造方式有类似之处。2.树状数组可以解决什么问题可以解决大部分基于区间上的更新以及求和问题。3.树状数组和线段树的区别在哪里树状数组可以解决的问题都可以用线段树解决，这两者的区别在哪里呢？树状数组的系数要少很多，就比如字...
树状数组数据结构
qq_37464878的博客
02-22 2523
1.树状数组概念 树状数组简单来说就是仅通过一个数组来描述一棵树或一个森林。数组中每个元素只存储着树中对应节点存储的信息，不存储描述树的层次结构的信息。 2.普通树的存储 对于一棵普通的树，在《数据结构》严蔚敏版中提到可以使用如下方法进行存储： 2.1 父节点表示法 用一个数组中存储节点信息，节点信息包括节点存储数据和父节点编号。 struct Node { int data; // 节点存储的数据 int parent; // 节点对应的父节点的编号 } struct Tree {
树状数组_洛谷_树状数组_
10-02
树状数组，也被称为线段树（Segment Tree），是一种数据结构，主要用于高效地处理动态区间查询和修改问题。在编程竞赛和算法设计中，树状数组是解决许多问题的利器，尤其是在面对需要频繁更新和查询区间信息的问题时...
树状数组树状数组树状数组树状数组
06-07
树状数组树状数组树状数组树状数组树状数组树状数组树状数组树状数组
js用闭包遍历树状数组的方法
12-12
做公司项目时，要求写一个方法，方法的参数为一个菜单数组集合和一个菜单id，菜单数组的格式为树状json,如下面所示： 代码如下:[{“id”:28,”text”:”公司信息”,”children”:[  {“id”:1,”text”:”公司文化...
树状数组 树状数组.ppt
09-17
树状数组 树状数组.ppt
C++二叉搜索树
Jk_Mr的博客
07-20 1096
树结构是一种重要的数据结构。树是一种非线性的数据结构，它是由n（n>=0）个有限结点组成一个具有层次关系的集合。把它叫做树是因为它看起来像一棵倒挂的树，也就是说它是根朝上，而叶朝下的。
C++ 继承篇
2301_79201049的博客
07-18 700
必须对继承而来的成员进行初始化直接通过初始化列表或者赋值的方式进行初始化调用父类构造函数进行初始化
【C++】操作符“new”和“delete”（创建和销毁对象）
qq_65596720的博客
07-19 396
由于C++引入了类的概念所以创建空间和销毁空间变得复杂，使用malloc和free会更加复杂，所以C++引入了new和delete来简化代码。
商品信息管理系统(C语言)
zeyeqianli的博客
07-22 297
该案例使用了C语言中最具特色的结构体，将每个商品的所有信息存在结构体中，并且定义一个结构体类型的数组保存所有商品的信息，并且按照模块化的编程思想，将要实现的每个功能编写成独立的函数，这样即方便阅读同时也方便差错修改。该系统的主要功能如下：(1) 登录系统。(2) 创建商品信息。(3) 打印商品信息。(4) 查询商品信息。(5) 修改商品信息。(6) 删除商品信息。(7) 商品价格排名显示。(8) 退出系统。
C++笔记3：基类指针delete子类对象的内存泄漏问题
subtitle_的博客
07-21 232
C++笔记3：基类指针delete子类对象的内存泄漏问题
2024年华为OD机试真题-虚拟游戏理财-C++-OD统一考试（C卷D卷）
面试高手的博客
07-21 9731
现有一家Bank，它提供有若干理财产品m，风险及投资回报不同，你有N（元）进行投资，能接受的总风险值为X。第一行：产品数(取值范围[1, 20])，总投资额(整数，取值范围[1, 10000])，可接受的总风险(整数，取值范围[1, 200])投资第二项30个单位，第四项40个单位，总的投资风险为两项相加为4+6=10。第四行：最大投资额度序列，输入为整数，取值范围[1,10000]第二行：产品投资回报率序列，输入为整数，取值范围[1,60]第三行：产品风险值序列，输入为整数，取值范围[1,100]
C语言 打印菱形的程序（Program to print the Diamond Shape）
csdn_aspnet的专栏
07-22 1470
O(n*n)，因为我们正在遍历网格的行和列来打印空格 ' '和星号 '*'。给定一个数字n ，编写一个程序来打印一个有2n行的菱形。O(1)，不使用额外空间。
【C++修炼之路 第三章：内存管理】new & delete、operator new & operator delete
唯有忍受别人不能忍受的，才能享受别人不能享受的
07-18 724
【含常见面试题】new & delete 的概念与用法 + 定位new表达式 + malloc/free 和 new/delete 的区别
学懂C语言（二十一）：深入理解C语言中的函数指针和回调函数
最新发布
martian665的专栏
07-23 177
学懂C语言：C语言中的函数指针和回调函数详解
C++的内存管理
2301_78694309的博客
07-19 963
C++的内存管理
树状数组详解与应用
"树状数组是一种数据结构，...它通过巧妙的位运算和树形结构设计，使得在大规模数据集上的操作变得高效。理解和掌握树状数组的原理及应用，对于解决实际编程问题，尤其是在算法竞赛和数据分析领域，都是至关重要的。
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

万伏小太阳

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付：¥1
获取中
扫码支付

您的余额不足，请更换扫码支付或充值

打赏作者

实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值