【分块模板】P2068统计和

发布时间:2024年01月12日

??题目传送门:统计和 - 洛谷

题目大意

给定一个长度为?n,初始值都为?0?的序列?,有w次操作,每次操作:

x a b 表示将a的值加上b,y a b 表示查询a到b的数字和。

思路

这题其实有很多做法,包括线段树、树状数组等。

但大家既然是来看板子的,那就用分块的写法。先画个图理解一下:

e0e3e6ef9cdf4747919a9bb1e6ab128f.png

这里,我们把n分成了? eq?%5Csqrt%20n?块,对于其中的每一块,我们都可以直接用for循环求出其区间和。

那查询怎么做呢?分三种情况:

44fb0de600ab42d79d21f1213b2ff8c1.png53b3396b4ff1454caf1c1410490c8365.png
430f6ce82a6a4240afc97e1862469150.png

三种情况中,红色部分可以直接用for循环求,而蓝色部分则是通过访问单个块内的区间和得到的

代码

#include<iostream>
#include<cmath>
using namespace std;
#define MaxN 100005
#define MaxB 320
#define For(i, j, k) for(int i = j; i <= k; i++)
#define int long long

int n, bs, mb;
//bs:每个块的大小;mb:块的数量
int block[MaxN], a[MaxN];
//block[i]:第i个点所属的块
int l[MaxB], r[MaxB];
//l[i]/r[i]:第i个块左端点和右端点
long long Sum[MaxB];

void init(){
	bs = sqrt(n);
	For(i, 1, n){
		block[i] = (i-1) / bs + 1;
		Sum[block[i]] += a[i];
	}
	mb = (n-1) / bs + 1;
	For(j, 1, mb){            //计算每个块的左右端点
		l[j] = (j-1) * bs + 1;
		r[j] = j * bs;
	}
	r[mb] = n;
}

int query(int x, int y){        //查询[x, y]区间的总和
	int ans = 0;
	if(block[x] == block[y]){    
		For(i, x, y) ans += a[i];    //两边直接相加
	} else{
		For(i, block[x]+1, block[y]-1) ans += Sum[i];
		For(i, x, r[block[x]]) ans += a[i];
		For(i, l[block[y]], y) ans += a[i];
	}
	return ans;
}

void modify(int x, int y){        //将x改为y
	a[x] += y; Sum[block[x]] = 0;    
	For(i, l[block[x]], r[block[x]])
		Sum[block[i]] += a[i];
}

signed main()
{
	int q, x, y;
    char op;
	cin >> n;
	For(i, 1, n) a[i] = 0;
	init();
	cin >> q;
	while(q--){
		cin >> op >> x >> y;
		if(op == 'y') cout << query(x, y) << endl;
		else modify(x, y);
	}
	return 0;
}

总结

各个数据结构的模板题,

即区间查询、单点修改的区间求和问题,可以说是入门数据结构的基础了,十分重要。

最后,如果你觉得这篇文章还不错,可以点个关注点个赞,这是免费的,你可以随时取消。你的支持永远是作者最大的动力!感谢大家的观看,我们下次见!

文章来源:https://blog.csdn.net/weixin_52398991/article/details/135439489
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。