树状数组(HDU1754 、洛谷)

最新推荐文章于 2024-02-13 21:37:54 发布
最新推荐文章于 2024-02-13 21:37:54 发布
阅读量365 收藏 2
点赞数 1
分类专栏： 树状数组 文章标签： 算法
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/m0_50623076/article/details/109208919
版权

目录

树状数组与线段树：

lowbit(x)

单点更新+区间修改

单点更新：

区间查询

洛谷：树状数组模板题1：

单点查询+区间修改：

洛谷：树状数组模板题2：

维护区间最大值的单点更新+区间查询(比较复杂)：

单点更新：

代码：

区间查询：

代码：

HDU 1754

区间查询+区间修改：

利用树状数组求解逆序数：

图片来源：bestsort.cn

树状数组与线段树：

树状数组和线段树复杂度一样，查询更新为O(logn)，树状数组比线段树的代码简洁，巧妙利用二进制lowbit(x)(表示x的二进制最右边的1)。但是线段树能解决的树状数组未必可以解决，树状数组可以解决的线段树一定可以解决。

lowbit(x)

如何取出二进制的1，对于一个二进制…..1….00..0(1前面是0或1，1后面全部是0)，得到这个1，可以想到移位，但是又要循环到1。寻找更优化的复杂度，对每一位取反再相与全部为0，怎么样让那个1取反仍然是1，1后面的取反是什么都无所谓。那么对取反后的(….1111.1111)加1操作就变成了(…..1..00..0)也就是该1前面的全取反，该1后面的全不变，包括它自己。这样相与就得到了最低位1代表的值。取反+1又是补码，也就是x与它的相反数相与，即x&(-x)。

所以代码为：

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

}

网图:

单点更新+区间修改

单点更新：

规定一个结点的上一个结点，即父节点为x+lowbit(x)。

所以对于单点修改来说(下面以区间和为例)

void update(int x,int k){      
     for(;x <= n;x+=lowbit(x))     t[x]+=k;
}

既然这样规定，那么一个结点所代表的数的个数为最低为1二进制的权值，比如…10只能由….01相加lowbit而来，加上本身所以为2。…100可由…010和….011（2+1+1=4）。…10…0(n个0)可由…01…0(2^n-2)…001…0(2^n-3)  一直到……1(1)等比数列求和公式可得2^n-1+1（本身）。(有递归的感觉)

所代表的数组范围也就是(x-lowbit(x)+1,x)

区间查询

所以对于区间查询来说：

int getsum(int x){//-lowbit(x)原因就是它对应结点所代表的大小，在计算除去它代表
       int ans = 0;//结点的个数就行了。 
      for(;x;x-=lowbit(x))      ans+=t[x]; 
      return ans;
}

注意，这时候得到的是[1,x]的getsum(为了更快，不用因为l在某t内二拆分)。求[x,y]利用

getsum(y)-getsum(x-1)得到。

洛谷：树状数组模板题1：

#include<algorithm>
#include<iostream>
#include<cstring>
#include<string>
#include<cstdlib>
#include<map>
#include<cmath>
#include<vector>
#include<cstdio>
//单点修改、区间查询 
using namespace std;
typedef long long ll;
const int maxn = 1e6+50;
int t[maxn];
int n;
int lowbit(int x){
	return x&(-x);
} 
void update(int x,int k){
	for(;x <= n;x+=lowbit(x))	t[x]+=k;
}
int getsum(int x){
	int ans = 0;
	for(;x;x-=lowbit(x))	ans+=t[x];
	return ans;
}
int main(){
	int m,a;
	cin >> n >> m;
	for(int i = 1;i <= n;i++)	scanf("%d",&a),update(i,a);
	int z,x,y;
	while(m--){
		scanf("%d%d%d",&z,&x,&y);
		if(z==1){
			update(x,y);
		}
		else if(z==2){
			int ans = 0;
			ans = getsum(y)-getsum(x-1);
			printf("%d\n",ans);
		}
	}

	return 0;
}

单点查询+区间修改：

对于区间修改，如何修改。在线段树中我们用一个add来记录，在这里我们用一个算法——差分。何为差分,就是利用一个数组记录该下标与该下标的上一个元素在原数组中的差值，这样只需要修改端点处即可。这样理解，比如对[x,y]做+k操作，我们就让div[x]+=k,div[y+1]-=k；在x之前的不受影响，是原来的，在[x,y]之间的会有①，对于大于y的有②。

a[n]=a[n]+div[1….n]=a[n]+div[x]=a[n]+k;①

a[n]=a[n]+div[1….n]=a[n]+div[x]+div[y+1]=a[n]+k-k=a[n];②

这样，对于a[i]只要是求的差分数组的前i项和与a[i]相加即可。转化为更新差分数组和求差分数组前i项和。所以将t作为差分数组即可；

洛谷：树状数组模板题2：

#include<algorithm>
#include<iostream>
#include<cstring>
#include<string>
#include<cstdlib>
#include<map>
#include<cmath>
#include<vector>
#include<cstdio>
//区间修改、单点查询--差分法 
using namespace std;
typedef long long ll;
const int maxn = 1e6+50;
int tree[maxn<<2];
int arr[maxn];
int n,m;
int lowbit(int x){
	return x&(-x);
} 
void update(int x,int k){
	for(;x<=n;x+=lowbit(x))	tree[x]+=k;
	return ;
}
int getsum(int x){
	int ans = 0;
	for(;x;x-=lowbit(x))	ans+=tree[x];
	return ans;
}
int main(){
	cin >> n >> m;
	for(int i = 1;i <= n;i++)	scanf("%d",&arr[i]);
	while(m--){
		int idx;
		cin >> idx;
		if(idx == 1){
			int x,y,z;
			cin >> x >> y >> z;
			update(x,z);
			update(y+1,-z);
		}
		else if(idx == 2){
			int x;
			cin >> x;
			cout << arr[x]+getsum(x) <<endl;
		}
	} 
	return 0;
}

维护区间最大值的单点更新+区间查询(比较复杂)：

单点更新：

对于一个数组a[x]更新，那么要重算x之后的包括x的所有数组值。

代码：

void  update(int x){
	for(;x<=n;x+=lowbit(x)){
		t[x] = a[x];
		int lx = lowbit(x);
		for(int i = 1;i < lx;i<<=1)		t[x] = max(t[x],t[x-i]);
	}
	return ;
}

外层for循环毫无疑问，就是更新x之后的所有t数组值。如何更新呢，之前所讲：

…10…0(n个0)可由…01…0(2^n-2)(数2^n-2)…001…0(2^n-3)(数x-2^n-3)  一直到……1(1)(数x-1)；

也就有了内层for循环，每个for复杂度O(logn)，所以复杂度为O((logn))^2，不算太高。

区间查询：

区间查询不能像sum一样相减了，只能精确定位。

之前说过，对于一个t[x]，所代表的范围为[x-lowbit(x)+1,x]；

如果l <= r-lowbit(r)+1 ans=ans(ans,t[r]){因为t[r]里面的代表的元素都在[l,r]区间内}

如果l>r-lowbit(r)+1 ans=ans(ans,a[r]),r--;{因为t[t]里面代表的元素有不是[l,r]的，不可以直接运用，所以需要拆分}

代码：

int getmax(int l,int r){
	int ans = 0;
	for(;r >= l;){
		ans = max(ans,a[r]);r--;
		for(;r-lowbit(r) > l;r-=lowbit(r))	ans = max(ans,t[r]);
	}
	return ans;
}

HDU 1754 

#include<algorithm>
#include<iostream>
#include<cstring>
#include<string>
#include<cstdlib>
#include<map>
#include<cmath>
#include<vector>
#include<cstdio>
/*
...100...000由...011...111   ...011...110 ...011...100 ......  ...010...100  ...010...000组成 
*/
using namespace std;
typedef long long ll;
const int maxn = 3e5+50;
int t[maxn<<4];
int a[maxn];
int n,m; 
int lowbit(int x){
	return x&(-x);
}
void  update(int x){
	for(;x<=n;x+=lowbit(x)){
		t[x] = a[x];
		int lx = lowbit(x);
		for(int i = 1;i < lx;i<<=1)		t[x] = max(t[x],t[x-i]);
	}
	return ;
}
int getmax(int l,int r){
	int ans = 0;
	for(;r >= l;){
		ans = max(ans,a[r]);r--;
		for(;r-lowbit(r) > l;r-=lowbit(r))	ans = max(ans,t[r]);
	}
	return ans;
}
inline int read(){
	int f = 1,x = 0;
	char c = getchar();
	while(c>'9'||c<'0'){
		if(c=='-')	f=-1;
		c=getchar();
	} 
	while(c<='9'&&c>='0'){
		x=x*10+c-'0';
		c=getchar();
	}
	return x*f;
}
int main(){
	while(scanf("%d%d",&n,&m)!=EOF){
		memset(t,0,sizeof(t));
		for(int i = 1;i <= n;i++){
			scanf("%d",&a[i]);	update(i);
		}
		while(m--){
			char c;
			scanf("%c",&c);
			scanf("%c",&c);
			if(c == 'Q'){
				int l,r;
				scanf("%d%d",&l,&r);
				printf("%d\n",getmax(l,r));
			}
			else if(c == 'U'){
				int idx,dx;
				scanf("%d%d",&idx,&dx);
				a[idx] = dx;
				update(idx);
			}
			
		}
	}
	return 0;
}

区间查询+区间修改：

利用差分进一步修改，上一个差分数组初始化0，这次初始化赋值div[i]=a[i]-a[i-1];

其实就是求差分数组的前缀和的前缀和，比如[1,x]就是求的a[x]到a[1]的和,

a[i]=j＝1idiv[j]

构建两个树状数组t1[],t2[]; t1[x]=d[x],t2[x]=d[x]*i；

void add(int x,int k){
	y=x;
	for(;x<=n;x+=lowbit(x))	t1[x]+=k,t2[x]+=y*k;
	return ;
}
void range_add(int l,int r,int x){
	add(l,x);add(r+1,-x);
}
int getsum(int x){
	int ans = 0,y=x;
	for(;x;x-=lowbit(x))	ans +=((y+1)*t1[x]-t2[x]);
	return ans;
}
int range_getsum(int l,int r){
	return getsum(r)-getsum(l-1);
}

利用树状数组求解逆序数：

遍历一遍，查询x之前有多少个数，那么i-number就是在i后面的数(因为getsum包含自己，所以不需要加一)。

#include<algorithm>
#include<iostream>
#include<cstring>
#include<string>
#include<cstdlib>
#include<map>
#include<cmath>
#include<vector>
#include<cstdio>
/*
在一个排列中，如果一对数的前后位置与大小顺序相反，即前面的数大于后面的数，那么它们就称为一个逆序。
一个排列中逆序的总数就称为这个排列的逆序数。比如一个序列为4 5 1 3 2， 那么这个序列的逆序数为7，
逆序对分别为(4, 1), (4, 3), (4, 2), (5, 1), (5, 3), (5, 2),(3, 2)。 
输入描述:
第一行有一个整数n(1 <= n <= 100000),  然后第二行跟着n个整数，对于第i个数a[i]，(0 <= a[i] <= 100000)。
*/
using namespace std;
typedef long long ll;
const int maxn = 1e6+50;
int t[maxn<<4];
int n,a; 
int lowbit(int x){
	return x&(-x);
}
void update(int x,int k){
	for(;x<=n;x+=lowbit(x))	t[x]+=k;
}
int getsum(int x){
	int ans =  0;
	for(;x;x-=lowbit(x))	ans+=t[x];
	return ans;
}
inline int read(){
	int f = 1,number = 0;
	char c = getchar();
	while(c>'9'||c<'0'){
		if(c=='-')	f=-1;
		c=getchar();
	}
	while(c<='9'&&c>='0'){
		number = number*10+c-'0';
		c=getchar();
	}
	return number*f;
}
int main(){
	ll ans = 0;
	while(cin>>n){
		memset(t,0,sizeof(t));
		for(int i = 1;i <= n;i++){
			a=read();
			update(a,1);
			ans += (ll)(i-getsum(a));
		}
		cout << ans << endl;
	}
	return 0;
}

求逆序数方法还有归并排序，在另一篇文章里面细讲：

#include<algorithm>
#include<iostream>
#include<cstring>
#include<string>
#include<cstdlib>
#include<map>
#include<cmath>
#include<vector>

using namespace std;
typedef long long ll;
const int maxn = 1e6+50;
ll a[maxn];
ll b[maxn];
ll ans;
void _merge(int l,int mid,int r){
	int p1 = l,p2 = mid+1;
	for(int i = l;i <= r;i++){
		if(((a[p1] <= a[p2])||(p2 > r))&&p1<=mid){
			b[i] = a[p1++];
		}/*等号无所谓，这里为了求逆对数，相等的情况让前面的先进去，主要靠后面进行求逆对数 */
		else {
			b[i] = a[p2++];
			ans += (ll)(mid-p1+1);
		}
	}
	for(int i = l;i <= r;i++)	a[i] = b[i];
	return ;
}
void erfen(int l,int r){
	int mid = (l+r)>>1;
	if(l < r){
		erfen(l,mid);
		erfen(mid+1,r);
	}
	_merge(l,mid,r);
	return ;
} 
int main(){
	int n;
	cin >> n;
	for(int i = 0;i < n;i++)	cin >> a[i];
	ans = 0;
	erfen(0,n-1); 
	cout << ans << endl;
	return 0;
}

如果有题说大致每次置换某一区间，求每一次对应的逆序对的奇偶性，这时候不要”傻乎乎”的每次计算，利用性质：每一对调换奇偶性改变。

_____111
关注
  • 1
    点赞
  • 踩
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
专栏目录
树状数组_洛谷_树状数组_
10-02
洛谷上关于树状数组的一些简单例题，可参考学习，适合初学者
洛谷 P3374 【模板】树状数组
qq_17807067的博客
09-28 288
树状数组简单理解
树状数组（例洛谷p1673）
最新发布
free_stmo的博客
02-13 105
应为我们要找一个三个数，而且得是连续徐上升的三个子序列，所以我们只需要记录比第i个数小的但是出现在i左边，和比第i个数大，但是出现在第i个数右边，然后ans+=(le[i]*ri[i])，就是最终的答案。树状数组算是一个非常奇葩的高级数据结构，(ge人理解）,相对于线段树理解起来更加麻烦，但是代码实现却是比较简单。这题还有一个线段树的做法，如果有需求·评论区·留言。这里以洛谷p1637作为例题。
洛谷P1972(树状数组+离线)
塞满箱子的过程
08-19 214
题目描述 HH 有一串由各种漂亮的贝壳组成的项链。HH 相信不同的贝壳会带来好运，所以每次散步完后，他都会随意取出一段贝壳，思考它们所表达的含义。HH 不断地收集新的贝壳，因此，他的项链变得越来越长。有一天，他突然提出了一个问题：某一段贝壳中，包含了多少种不同的贝壳？这个问题很难回答……因为项链实在是太长了。于是，他只好求助睿智的你，来解决这个问题。 输入格式 第一行：一个整数N，表示项链的长度。...
洛谷 - P3368 树状数组2
cowgoodsheep的博客
03-08 215
洛谷 - P3368 树状数组2
【洛谷3368】树状数组 2 树状数组+差分
Mininda
11-10 1295
树状数组 2 题目描述 如题，已知一个数列，你需要进行下面两种操作： 1.将某区间每一个数数加上x 2.求出某一个数的和 输入输出格式 输入格式：   第一行包含两个整数N、M，分别表示该数列数字的个数和操作的总个数。 第二行包含N个用空格分隔的整数，其中第i个数字表示数列第i项的初始值。 接下来M行每行包含2或4个整数，表示一个操作，具体如下： 操作1：格式：1 x y k ...
【洛谷】P3374 【模板】树状数组 1
数学、算法爱好者的博客
01-07 480
题目地址： https://www.luogu.com.cn/problem/P3374 树状数组模板题。树状数组也叫Fenwick Tree或者Binary Indexed Tree，建树时间O(nlog⁡n)O(n\log n)O(nlogn)，支持O(log⁡n)O(\log n)O(logn)查询数组前缀和（于是就支持O(log⁡n)O(\log n)O(logn)的范围求和），也支持O(...
猴子选大王（约瑟夫问题）洛谷P1996
yyf525的博客
10-08 351
题目： 题目描述 nn个人围成一圈，从第一个人开始报数,数到mm的人出列，再由下一个人重新从11开始报数，数到mm的人再出圈，依次类推，直到所有的人都出圈，请输出依次出圈人的编号。 注意：本题和《深入浅出-基础篇》上例题的表述稍有不同。书上表述是给出淘汰 n-1 名小朋友，而该题是全部出圈。 输入格式 输入两个整数n,mn,m。 输出格式 输出一行nn个整数，按顺序输出每个出圈人的编号。 输入输出样例 输入 #1复制 10 3 输出 #1复制 3 6 9 2 ...
[树状数组] 洛谷 P1972 HH的项链
weixin_30919429的博客
01-22 62
题目描述 HH 有一串由各种漂亮的贝壳组成的项链。HH 相信不同的贝壳会带来好运，所以每次散步完后，他都会随意取出一段贝壳，思考它们所表达的含义。HH 不断地收集新的贝壳，因此，他的项链变得越来越长。有一天，他突然提出了一个问题：某一段贝壳中，包含了多少种不同的贝壳？这个问题很难回答……因为项链实在是太长了。于是，他只好求助睿智的你，来解决这个问题。 输入输出格式 输入格式： 第一行...
【洛谷-P3368】（树状数组）
__似水流年__的博客
05-24 292
题目链接：https://www.luogu.org/problemnew/show/P3368#sub 题目描述 如题，已知一个数列，你需要进行下面两种操作： 1.将某区间每一个数数加上x 2.求出某一个数的值 输入输出格式 输入格式： 第一行包含两个整数N、M，分别表示该数列数字的个数和操作的总个数。 第二行包含N个用空格分隔的整数，其中第i个数字表示数列第i项的初始值。...
线段树入门学习(兼解HDU1754)
04-14
NULL 博文链接：https://128kj.iteye.com/blog/1738772
hdu.rar_hdu
09-23
我写的hdu上的一些题AC的题的代码 也许你会有用
hdu1020.rar_hdu1020
09-20
Hdu 1020解题报告，http://acm.hdu.edu.cn/showproblem.php?pid=1020
hdu.zip_ACM_hdu
09-20
写一个数据结构算法，杭电上一道题目，有关数据结构的题目。
hdu刷题入门(1)
12-21
标题前言hdu2023求平均成绩hdu2027统计元音hud2044一只小蜜蜂…hud2029Palindromes _easy versionhud2043密码hud2040亲和数hud2021发工资咯hud2032杨辉三角 前言 今天bigsai陪伴我刷题，我很高兴在他的帮助下过了8道...
2022/2/10
weixin_52621204的博客
02-10 497
p6225 位运算+树状数组，p6278 树状数组+逆序对，p6492 线段树左右区间前缀问题，p1725 dp+单调队列优化，p6510 单调栈，p6812 线段树，p7865 二维差分，二位前缀和
单点修改&区间求和
AlanJobs的博客
02-10 644
题目描述 如题，已知一个数列，你需要进行下面两种操作： 将某一个数加上xx 求出某区间每一个数的和 输入格式 第一行包含两个正整数n,mn,m，分别表示该数列数字的个数和操作的总个数。 第二行包含nn个用空格分隔的整数，其中第ii个数字表示数列第ii项的初始值。 接下来mm行每行包含33个整数，表示一个操作，具体如下： 1 x k含义：将...
树状数组入门——以洛谷3374为例
SuaSUA_He的博客
10-11 291
树状数组入门 含义：顾名思义，用树状表示的数组 功能：是一个查询和修改复杂度都为log(n)的数据结构。主要用于查询任意两位之间的所有元素之和，但是每次只能修改一个元素的值；经过简单修改可以在log(n)的复杂度下进行范围修改，但是这时只能查询其中一个元素的值(如果加入多个辅助数组则可以实现区间修改与区间查询)。 如图所示： 数组a[i]表示叶子节点 c[i]表示叶子节点的权值之和 根据上图可以...
HDU - 5513
05-25
这是一道IT类问题，HDU-5513是一道算法题目，题目描述为：给定一个长度为n的数组，求出其中任意两个元素之间的最大差值。 解题思路是，遍历整个数组，记录下来当前的最小值和最大差值，如果当前元素与最小值的差值...

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

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

余额充值