目录
线段树是什么??线段树怎么写??
如果你在考提高组前一天还在问这个问题,那么你会与一等奖失之交臂;如果你还在冲击普及组一等奖,那么这篇博客会浪费你人生中宝贵的5~20分钟。
上面两句话显而易见,线段树这个数据结构是一个从萌新到正式OI选手的过渡,是一个非常重要的算法,也是一个对于萌新来说较难的算法。不得不说,我学习了这个算法5遍左右才有勇气写的这篇博客。
但是,对于OI正式选手来说,线段树不是算法,应该是一种工具。她能把一些对于区间(或者线段)的修改、维护,从O(N)的时间复杂度变成O(logN)。
第一部:线段树概念引入
第二部:简单(无pushdown)的线段树
第三部:区间+/-修改与查询
第四部:区间乘除修改与查询
第一部 概念引入
线段树是一种二叉树,也就是对于一个线段,我们会用一个二叉树来表示。比如说一个长度为4的线段,我们可以表示成这样:

这是什么意思呢? 如果你要表示线段的和,那么最上面的根节点的权值表示的是这个线段1-4的和。根的两个儿子分别表示这个线段中1-2的和,与2-3的和。以此类推。
然后我们还可以的到一个性质:节点i的权值=她的左儿子权值+她的右儿子权值。因为1-4的和就是等于1-2的和2-3的和。
根据这个思路,我们就可以建树了,我们设一个结构体tree,tree[i].l和tree[i].r分别表示这个点代表的线段的左右下标,tree[i].sum表示这个节点表示的线段和。
我们知道,一颗二叉树,她的左儿子和右儿子编号分别是她*2和她*2+1
再根据刚才的性质,得到式子: t r e e [ i ] . s u m = t r e e [ i ∗ 2 ] . s u m + t r e e [ i ∗ 2 + 1 ] . s u m ; tree[i].sum=tree[i*2].sum+tree[i*2+1].sum; tree[i].sum=tree[i∗2].sum+tree[i∗2+1].sum;就可以建一颗线段树了!代码如下:
inline void build(int i,int l,int r){
//递归建树 tree[i].l=l;tree[i].r=r; if(l==r){
//如果这个节点是叶子节点 tree[i].sum=input[l]; return ; } int mid=(l+r)>>1; build(i*2,l,mid);//分别构造左子树和右子树 build(i*2+1,mid+1,r); tree[i].sum=tree[i*2].sum+tree[i*2+1].sum;//刚才我们发现的性质return ; }
嗯,这就是线段树的构建,你可能会问为什么要开好几倍的内存去储存一条线段。这是因为我们还没有让这个过大的数组干一些实事,那么什么是实事呢?让我们进入下一部(在你看懂这一部的情况下)
第二部 简单(无pushdown)的线段树
1、单点修改,区间查询
其实这一章开始才是真正的线段树,我们要用线段树干什么?答案是维护一个线段(或者区间),比如你想求出一个1100区间中,467这些元素的和,你会怎么做?朴素的做法是for(i=4;i<=67;i++) sum+=a[i],这样固然好,但是算得太慢了。
我们想一种新的方法,先想一个比较好画图的数据,比如一个长度为4的区间,分别是1、2、3、4,我们想求出第1~3项的和。按照上一部说的,我们要建出一颗线段树,其中点权(也就是红色)表示和:

然后我们要求1-3的和,我们先从根节点开始查询,发现她的左儿子1-2这个区间和答案区间1~3有交集,那么我们跑到左儿子这个区间。
然后,我们发现这个区间1-2被完全包括在答案区间1~3这个区间里面,那就把她的值3返回。
我们回到了1-4区间,发现她的右儿子3-4区间和答案区间1~3有交集,那么我们走到3-4区间
到了3-4区间,我们发现她并没有完全包含在答案区间1-3里面,但发现她的左儿子3-3区间和1~3区间又交集,那么久走到3-3区间
到了3~3区间,发现其被答案区间完全包含,就返回她的值3一直到最开始
3-3区间的3+1-2区间的3=6,我们知道了1~3区间和为6.
有人可能会说你这样是不是疯了,我那脚都能算出1+2+3=6,为什么这么麻烦?!
因为这才几个数,如果一百万个数,这样时间会大大增快。
我们总结一下,线段树的查询方法:
- 如果这个区间被完全包括在目标区间里面,直接返回这个区间的值
- 如果这个区间的左儿子和目标区间有交集,那么搜索左儿子
- 如果这个区间的右儿子和目标区间有交集,那么搜索右儿子
写成代码,就会变成这样:
inline int search(int i,int l,int r){
if(tree[i].l>=l && tree[i].r<=r)//如果这个区间被完全包括在目标区间里面,直接返回这个区间的值 return tree[i].sum; if(tree[i].r<l || tree[i].l>r) return 0;//如果这个区间和目标区间毫不相干,返回0 int s=0; if(tree[i*2].r>=l) s+=search(i*2,l,r);//如果这个区间的左儿子和目标区间又交集,那么搜索左儿子 if(tree[i*2+1].l<=r) s+=search(i*2+1,l,r);//如果这个区间的右儿子和目标区间又交集,那么搜索右儿子 return s; }
关于那几个if的条件一定要看清楚,最好背下来,以防考场上脑抽推错。
然后,我们怎么修改这个区间的单点,其实这个相对简单很多,你要把区间的第dis位加上k。
那么你从根节点开始,看这个dis是在左儿子还是在右儿子,在哪往哪跑,
然后返回的时候,还是按照tree[i].sum=tree[i*2].sum+tree[i*2+1].sum的原则,更新所有路过的点
如果不理解,我还是画个图吧,其中深蓝色是去的路径,浅蓝色是返回的路径,回来时候红色的+标记就是把这个点加上这个值。

把这个过程变成代码,就是这个样子:
inline void add(int i,int dis,int k){
if(tree[i].l==tree[i].r){
//如果是叶子节点,那么说明找到了 tree[i].sum+=k; return ; } if(dis<=tree[i*2].r) add(i*2,dis,k);//在哪往哪跑 else add(i*2+1,dis,k); tree[i].sum=tree[i*2].sum+tree[i*2+1].sum;//返回更新 return ; }
2、区间修改,单点查询
区间修改和单点查询,我们的思路就变为:如果把这个区间加上k,相当于把这个区间涂上一个k的标记,然后单点查询的时候,就从上跑道下,把沿路的标记加起来就好。
这里面给区间贴标记的方式与上面的区间查找类似,原则还是那三条,只不过第一条:如果这个区间被完全包括在目标区间里面,直接返回这个区间的值变为了如果这个区间如果这个区间被完全包括在目标区间里面,讲这个区间标记k
具体做法很像,这里贴上代码:
void modify(int p, int l, int r, int k) {
if(tr[p].l >= l && tr[p].r <= r) {
tr[p].num += k; return ; } int mid = tr[p].l + tr[p].r >> 1; if(l <= mid) modify(p << 1, l, r, k); if(r > mid) modify(p << 1 | 1, l, r, k); } /* inline void add(int i,int l,int r,int k){ if(tree[i].l>=l && tree[i].r<=r){//如果这个区间被完全包括在目标区间里面,讲这个区间标记k tree[i].sum+=k; return ; } if(tree[i*2].r>=l) add(i*2,l,r,k); if(tree[i*2+1].l<=r) add(i*2+1,l,r,k); } */
然后就是单点查询了,这个更好理解了,就是dis在哪往哪跑,把路径上所有的标价加上就好了:
void query(int p, int x) {
ans += tr[p].num;//一路加起来 if(tr[p].l == tr[p].r) return; int mid = tr[p].l + tr[p].r >> 1; if(x <= mid) query(p << 1, x); else query(p << 1 | 1, x); } /* void search(int i,int dis){ ans+=tree[i].sum;//一路加起来 if(tree[i].l==tree[i].r) return ; if(dis<=tree[i*2].r) search(i*2,dis); if(dis>=tree[i*2+1].l) search(i*2+1,dis); } */
区间修改,单点查询完整测试代码:
#include
using namespace std; const int maxn = 5e5 + 7; int n, m, s, t; int ans; int a[maxn]; struct segment_tree {
struct node {
int l, r; int num; }tr[maxn * 4]; void build(int p, int l, int r) {
tr[p] = {
l, r, 0}; if(l == r) {
tr[p].num = a[l]; return ; } int mid = l + r >> 1; build(p << 1, l, mid); build(p << 1 | 1, mid + 1, r); } void modify(int p, int l, int r, int k) {
if(tr[p].l >= l && tr[p].r <= r) {
tr[p].num += k; return ; } int mid = tr[p].l + tr[p].r >> 1; if(l <= mid) modify(p << 1, l, r, k); if(r > mid) modify(p << 1 | 1, l, r, k); } void query(int p, int x) {
ans += tr[p].num; if(tr[p].l == tr[p].r) return; int mid = tr[p].l + tr[p].r >> 1; if(x <= mid) query(p << 1, x); else query(p << 1 | 1, x); } }ST; int main() {
cin >> n >> m; for (int i = 1; i <= n; ++ i) {
scanf("%d", &a[i]); } ST.build(1, 1, n); for (int i = 1; i <= m; ++ i) {
int c; scanf("%d", &c); if(c == 1) {
int x, y, z; scanf("%d%d%d", &x, &y, &z); ST.modify(1, x, y, z); } else {
ans = 0; int x; scanf("%d", &x); ST.query(1, x); printf("%d\n", ans); } } return 0; } /* int main() { n = 100; for (int i = 1; i <= n; ++ i) { a[i] = i; } ST.build(1, 1, n); m = 10; for (int i = 1; i <= m; ++ i) { int l = 1, r = 100; ST.modify(1, l, r, 10000); ans = 0; // query(p, x), p = 1, x 为想要查询的点的下标 ST.query(1, 50); // 单点查询 下标为 50 的点的值,ans = 50 + 10000 * i cout << i << " " << ans << '\n'; ans = 0; ST.query(1, 100); // 单点查询 ans = 100 + 10000 * i cout << i << " " << ans << '\n'; } return 0; } */
因为我们在进行区间修改的时候,若当前区间已经被完全包含在目标区间 [ l , r ] [l,r] [l,r] 里,直接将该区间 tree[i].num += k,不需要再继续往下走了,类似 lazy 标记,所以单点查询的时候要再加上路径上的值(即原本的权值再加上经过的若干次修改的值才是这个单点的值)。
不知不觉,这第二章已经结束。这样的简单(原谅我用这个词)线段树,还可除了求和,还可以求区间最小最大值,还可以区间染色。
但是!这样的线段树展现不出来她的魅力,因为区间求和,树状数组比她少了一个很大的常数。二区间最值,ST表的那神乎其技 O ( 1 ) O(1) O(1) 查询也能完爆她。这是为什么?因为线段树的魅力还没有展现出来,她最美丽的地方:pushdown还未展现于世,如果你已经对这一章充足的了解,并且能不看博客把洛谷上树状数组模板1、2都能写出来,那么请你进入下一部。
第三部 进阶线段树
区间修改、区间查询,你可能会认为,把上一章里面的这两个模块加在一起就好了,然后你就会发现你大错特错。
因为如果对于1~4这个区间,你把1~3区间+1,相当于把节点1~2和3标记,但是如果你查询2~4时,你会发现你加的时没有标记的2节点和没有标记的3~4节点加上去,结果当然是错的。
那么我们应该怎么办?这时候pushdown的作用就显现出来了。
区间修改的时候,我们按照如下原则:
1、如果当前区间被完全覆盖在目标区间里,讲这个区间的sum+k*(tree[i].r-tree[i].l+1) 2、如果没有完全覆盖,则先下传懒标记 3、如果这个区间的左儿子和目标区间有交集,那么搜索左儿子 4、如果这个区间的右儿子和目标区间有交集,那么搜索右儿子
然后查询的时候,将这个懒标记下传就好了,下面图解一下:
如图,区间1-4分别是1、2、3、4,我们要把1-3区间+1。因为1~2区间被完全覆盖,所以将其+2,并将紫色的lazytage+1,3区间同理

注意我们处理完这些以后,还是要按照tree[i].sum=tree[i*2].sum+tree[i*2+1].sum的原则返回,代码如下:
void add(int i,int l,int r,int k) {
if(tree[i].r<=r && tree[i].l>=l)//如果当前区间被完全覆盖在目标区间里,讲这个区间的sum+k*(tree[i].r-tree[i].l+1) {
tree[i].sum+=k*(tree[i].r-tree[i].l+1); tree[i].lz+=k;//记录lazytage return ; } push_down(i);//向下传递 if(tree[i*2].r>=l) add(i*2,l,r,k); if(tree[i*2+1].l<=r) add(i*2+1,l,r,k); tree[i].sum=tree[i*2].sum+tree[i*2+1].sum; return ; }
其中的pushdown,就是把自己的lazytage归零,并给自己的儿子加上,并让自己的儿子加上k*(r-l+1)
void push_down(int i) {
if(tree[i].lz!=0) {
tree[i*2].lz+=tree[i].lz;//左右儿子分别加上父亲的lz tree[i*2+1].lz+=tree[i].lz; int mid=(tree[i].l+tree[i].r)/2; tree[i*2].sum+=tree[i].lz*(mid-tree[i*2].l+1);//左右分别求和加起来 tree[i*2+1].sum+=tree[i].lz*(tree[i*2+1].r-mid); tree[i].lz=0;//父亲lz归零 } return ; }
查询的时候,和上一章的几乎一样,就是也要像修改一样加入pushdown,这里用图模拟一下。我们要查询2~4区间的和,这是查询前的情况,所有紫色的代表lazytage

然后,我们查到区间1~2时,发现这个区间并没有被完全包括在目标区间里,于是我们就pushdown,lazytage下传,并让每个区间sum加上(r-l)*lazytage。

然后查到2-2区间,发现被完全包含,所以就返3,再搜索到3~4区间,发现被完全包含,那么直接返回8,最后3+8=11就是答案
这里是代码实现:
inline int search(int i,int l,int r){
if(tree[i].l>=l && tree[i].r<=r) return tree[i].sum; if(tree[i].r<l || tree[i].l>r) return 0; push_down(i); int s=0; if(tree[i*2].r>=l) s+=search(i*2,l,r); if(tree[i*2+1].l<=r) s+=search(i*2+1,l,r); return s; }
好了,到了这里,我们就学会了用线段树进行区间加减操作,大家可以完成洛谷的线段树模板1.
第四部 乘法(根号)线段树
1、乘法线段树
如果这个线段树只有乘法,那么直接加入lazytage变成乘,然后tree[i].sum*=k就好了。但是,如果我们是又加又乘,那就不一样了。
当lazytage下标传递的时候,我们需要考虑,是先加再乘还是先乘再加。我们只需要对lazytage做这样一个处理。
lazytage分为两种,分别是加法的plz和乘法的mlz。
mlz很简单处理,pushdown时直接*父亲的就可以了,那么加法呢?
我们需要把原先的plz*父亲的mlz再加上父亲的plz.
inline void pushdown(long long i){
//注意这种级别的数据一定要开long long long long k1=tree[i].mlz,k2=tree[i].plz; tree[i<<1].sum=(tree[i<<1].sum*k1+k2*(tree[i<<1].r-tree[i<<1].l+1))%p;// tree[i<<1|1].sum=(tree[i<<1|1].sum*k1+k2*(tree[i<<1|1].r-tree[i<<1|1].l+1))%p; tree[i<<1].mlz=(tree[i<<1].mlz*k1)%p; tree[i<<1|1].mlz=(tree[i<<1|1].mlz*k1)%p; tree[i<<1].plz=(tree[i<<1].plz*k1+k2)%p; tree[i<<1|1].plz=(tree[i<<1|1].plz*k1+k2)%p; tree[i].plz=0; tree[i].mlz=1; return ; }
然后加法和减法的函数同理,维护lazytage的时候加法标记一定要记得现乘再加。
值得一提的是,计算*2时一定要改成i<<1这样能解决很多时间,还有要开long long,还有,函数前面要加inline 我在其他OJ交这道题时,就因为没加inline 就被卡了,交了就过了。
2、根号线段树
其实,根号线段树和除法线段树一样。她们乍眼一看感觉直接用lazytage标记除了多少,但是实际上,会出现精度问题。
c++的除法是向下取整,很明显,(a+b)/k!=a/k+b/k(在向下取整的情况下),而根号,很明显根号(a)+根号(b)!=根号(a+b)那么怎么办?
第一个想法就是暴力,对于每个要改动的区间l~r,把里面的每个点都单独除,但这样就会把时间复杂度卡得比大暴力都慢(因为多个常数),所以怎么优化?
我们对于每个区间,维护她的最大值和最小值,然后每次修改时,如果这个区间的最大值根号和最小值的根号一样,说明这个区间整体根号不会产生误差,就直接修改(除法同理)
其中,lazytage把除法当成减法,记录的是这个区间里每个元素减去的值。
下面是根号线段树的修改过程:
inline void Sqrt(int i,int l,int r){
if(tree[i].l>=l && tree[i].r<=r && (tree[i].minn-(long long)sqrt(tree[i].minn))==(tree[i].maxx-(long long)sqrt(tree[i].maxx))){
//如果这个区间的最大值最小值一样 long long u=tree[i].minn-(long long)sqrt(tree[i].minn);//计算区间中每个元素需要减去的 tree[i].lz+=u; tree[i].sum-=(tree[i].r-tree[i].l+1)*u; tree[i].minn-=u; tree[i].maxx-=u; //cout<<"i"<
return
;
}
if
(tree
[i
]
.r
<l
|| tree
[i
]
.l
>r
)
return
;
push_down
(i
)
;
if
(tree
[i
*
2
]
.r
>=l
)
Sqrt
(i
*
2
,l
,r
)
;
if
(tree
[i
*
2
+
1
]
.l
<=r
)
Sqrt
(i
*
2
+
1
,l
,r
)
; tree
[i
]
.sum
=tree
[i
*
2
]
.sum
+tree
[i
*
2
+
1
]
.sum
; tree
[i
]
.minn
=
min
(tree
[i
*
2
]
.minn
,tree
[i
*
2
+
1
]
.minn
)
;
//维护最大值和最小值 tree
[i
]
.maxx
=
max
(tree
[i
*
2
]
.maxx
,tree
[i
*
2
+
1
]
.maxx
)
;
//cout<<"i"<
return
;
}
然后pushdown没什么变化,就是要记得tree[i].minn、tree[i].maxx也要记得-lazytage。
附赠一份支持单点修改,区间取相反数,区间最小值,区间最大值的线段树代码
题目链接:https://www.luogu.com.cn/problem/P1505
题解:BZOJ 2157 「国家集训队」旅游(树链剖分,线段树,边权转点权)
struct Segment_Tree {
struct node {
int l, r; int sum; // int laz; int tag;//相反数的tag int maxx; int minn; }tr[maxn << 2]; void pushup(int p) {
node &l = tr[p << 1], &r = tr[p << 1 | 1], &rt = tr[p]; rt.sum = l.sum + r.sum; rt.maxx = max(l.maxx, r.maxx); rt.minn = min(l.minn, r.minn); } void pushdown(int p) {
node &l = tr[p << 1], &r = tr[p << 1 | 1], &rt = tr[p]; if(rt.tag == 0) return ; l.tag ^= 1, r.tag ^= 1; l.sum = -l.sum, r.sum = -r.sum; l.maxx = -l.maxx, r.maxx = -r.maxx; l.minn = -l.minn, r.minn = -r.minn; swap(l.maxx, l.minn), swap(r.maxx, r.minn); rt.tag = 0; } void build(int p, int l, int r) {
tr[p].l = l, tr[p].r = r; if(l == r) {
tr[p].sum = tr[p].minn = tr[p].maxx = a_after[l]; return ; } int mid = l + r >> 1; build(p << 1, l, mid); build(p << 1 | 1, mid + 1, r); pushup(p); } void modify(int p, int x, int k) {
if(tr[p].l == tr[p].r) {
tr[p].sum = tr[p].minn = tr[p].maxx = k; return ; } if(tr[p].tag) pushdown(p); int mid = tr[p].l + tr[p].r >> 1; if(x <= mid) modify(p << 1, x, k); if(x > mid) modify(p << 1 | 1, x, k); pushup(p); } void _reverse(int p, int l, int r) {
if(tr[p].l >= l && tr[p].r <= r) {
tr[p].tag ^= 1; tr[p].sum = -tr[p].sum; tr[p].minn = -tr[p].minn; tr[p].maxx = -tr[p].maxx; swap(tr[p].minn, tr[p].maxx); return ; } if(tr[p].tag) pushdown(p); int mid = tr[p].l + tr[p].r >> 1; if(l <= mid) _reverse(p << 1, l, r); if(r > mid) _reverse(p << 1 | 1, l, r); pushup(p); } int query_sum(int p, int l, int r) {
int res = 0; if(tr[p].l >= l && tr[p].r <= r) return tr[p].sum; if(tr[p].tag) pushdown(p); int mid = tr[p].l + tr[p].r >> 1; if(l <= mid) res += query_sum(p << 1, l, r); if(r > mid) res += query_sum(p << 1 | 1, l, r); pushup(p); return res; } int query_max(int p, int l, int r) {
int res = -INF; if(tr[p].l >= l && tr[p].r <= r) return tr[p].maxx; if(tr[p].tag) pushdown(p); int mid = tr[p].l + tr[p].r >> 1; if(l <= mid) res = max(res, query_max(p << 1, l, r)); if(r > mid) res = max(res, query_max(p << 1 | 1, l, r)); pushup(p); return res; } int query_min(int p, int l, int r) {
int res = INF; if(tr[p].l >= l && tr[p].r <= r) return tr[p].minn; if(tr[p].tag) pushdown(p); int mid = tr[p].l + tr[p].r >> 1; if(l <= mid) res = min(res, query_min(p << 1, l, r)); if(r > mid) res = min(res, query_min(p << 1 | 1, l, r)); pushup(p); return res; } }ST;
模板题与代码:
最后,我们给几道模板题,再贴上代码:
单点修改,区间查询:洛谷树状数组模板1
P3374 【模板】树状数组 1
#include
#include
#include
#include
#include
#include
#include
#include
#include
using namespace std; #define MAXN #define INF #define MOD #define LL long long #define in(a) a=read() #define REP(i,k,n) for(long long i=k;i<=n;i++) #define DREP(i,k,n) for(long long i=k;i>=n;i--) #define cl(a) memset(a,0,sizeof(a)) inline long long read(){
long long x=0,f=1;char ch=getchar(); for(;!isdigit(ch);ch=getchar()) if(ch=='-') f=-1; for(;isdigit(ch);ch=getchar()) x=x*10+ch-'0'; return x*f; } inline void out(long long x){
if(x<0) putchar('-'),x=-x; if(x>9) out(x/10); putchar(x%10+'0'); } long long n,m,p; long long input[MAXN]; struct node{
long long l,r; long long sum,mlz,plz; }tree[4*MAXN]; inline void build(long long i,long long l,long long r){
tree[i].l=l; tree[i].r=r; tree[i].mlz=1; if(l==r){
tree[i].sum=input[l]%p; return ; } long long mid=(l+r)>>1; build(i<<1,l,mid); build(i<<1|1,mid+1,r); tree[i].sum=(tree[i<<1].sum+tree[i<<1|1].sum)%p; return ; } inline void pushdown(long long i){
long long k1=tree[i].mlz,k2=tree[i].plz; tree[i<<1].sum=(tree[i<<1].sum*k1+k2*(tree[i<<1].r-tree[i<<1].l+1))%p; tree[i<<1|1].sum=(tree[i<<1|1].sum*k1+k2*(tree[i<<1|1].r-tree[i<<1|1].l+1))%p; tree[i<<1].mlz=(tree[i<<1].mlz*k1)%p; tree[i<<1|1].mlz=(tree[i<<1|1].mlz*k1)%p; tree[i<<1].plz=(tree[i<<1].plz*k1+k2)%p; tree[i<<1|1].plz=(tree[i<<1|1].plz*k1+k2)%p; tree[i].plz=0; tree[i].mlz=1; return ; } inline void mul(long long i,long long l,long long r,long long k){
if(tree[i].r<l || tree[i].l>r) return ; if(tree[i].l>=l && tree[i].r<=r){
tree[i].sum=(tree[i].sum*k)%p; tree[i].mlz=(tree[i].mlz*k)%p; tree[i].plz=(tree[i].plz*k)%p; return ; } pushdown(i); if(tree[i<<1].r>=l) mul(i<<1,l,r,k); if(tree[i<<1|1].l<=r) mul(i<<1|1,l,r,k); tree[i].sum=(tree[i<<1].sum+tree[i<<1|1].sum)%p; return ; } inline void add(long long i,long long l,long long r,long long k){
if(tree[i].r<l || tree[i].l>r) return ; if(tree[i].l>=l && tree[i].r<=r){
tree[i].sum+=((tree[i].r-tree[i].l+1)*k)%p; tree[i].plz=(tree[i].plz+k)%p; return ; } pushdown(i); if(tree[i<<1].r>=l) add(i<<1,l,r,k); if(tree[i<<1|1].l<=r) add(i<<1|1,l,r,k); tree[i].sum=(tree[i<<1].sum+tree[i<<1|1].sum)%p; return ; } inline long long search(long long i,long long l,long long r){
if(tree[i].r<l || tree[i].l>r) return 0; if(tree[i].l>=l && tree[i].r<=r) return tree[i].sum; pushdown(i); long long sum=0; if(tree[i<<1].r>=l) sum+=search(i<<1,l,r)%p; if(tree[i<<1|1].l<=r) sum+=search(i<<1|1,l,r)%p; return sum%p; } int main(){
in(n); in(m);in(p); REP(i,1,n) in(input[i]); build(1,1,n); REP(i,1,m){
long long fl,a,b,c; in(fl); if(fl==1){
in(a);in(b);in(c); c%=p; mul(1,a,b,c); } if(fl==2){
in(a);in(b);in(c); c%=p; add(1,a,b,c); } if(fl==3){
in(a);in(b); printf("%lld\n",search(1,a,b)); } } return 0; } /* 5 4 1000 1 2 3 4 5 3 1 5 2 1 5 1 1 1 5 2 3 1 5 */
区间修改,单点查询:洛谷树状数组模板2
P3368 【模板】树状数组 2
#include
#include
#include
#include
#include
#include
using namespace std; int n,m; int ans; int input[]; struct node {
int left,right; int num; }tree[]; void build(int left,int right,int index) {
tree[index].num=0; tree[index].left=left; tree[index].right=right; if(left==right) return ; int mid=(right+left)/2; build(left,mid,index*2); build(mid+1,right,index*2+1); } void pls(int index,int l,int r,int k) {
if(tree[index].left>=l && tree[index].right<=r) {
tree[index].num+=k; return ; } if(tree[index*2].right>=l) pls(index*2,l,r,k); if(tree[index*2+1].left<=r) pls(index*2+1,l,r,k); } void search(int index,int dis) {
ans+=tree[index].num; if(tree[index].left==tree[index].right) return ; if(dis<=tree[index*2].right) search(index*2,dis); if(dis>=tree[index*2+1].left) search(index*2+1,dis); } int main() {
int n,m; cin>>n>>m; build(1,n,1); for(int i=1;i<=n;i++) scanf("%d",&input[i]); for(int i=1;i<=m;i++) {
int a; scanf("%d",&a); if(a==1) {
int x,y,z; scanf("%d%d%d",&x,&y,&z); pls(1,x,y,z); } if(a==2) {
ans=0; int x; scanf("%d",&x); search(1,x); printf("%d\n",ans+input[x]); } } }
区间加法,洛谷线段树模板1
P3372 【模板】线段树 1
#include
using namespace std; typedef long long ll; const ll N=1e6+7; const ll mod=; ll n,m; struct node {
ll l,r,sum,lz; }tree[N]; ll arr[N]; void build(ll i,ll l,ll r,ll arr[]) {
tree[i].lz=0;//初始化的时候肯定都是0 tree[i].l=l; tree[i].r=r; if(l==r) {
tree[i].sum=arr[l];//到达底部单节点才把输入的值赋给你 return ; } ll mid=(l+r)/2; build(i*2,l,mid,arr); build(i*2+1,mid+1,r,arr); tree[i].sum=tree[i*2].sum+tree[i*2+1].sum;//树已经全部建完了,再从下往上+++,使得上层的树都有了值 return ; } inline void push_down(ll i) {
if(tree[i].lz!=0) {
tree[i*2].lz+=tree[i].lz; tree[i*2+1].lz+=tree[i].lz; ll mid=(tree[i].l+tree[i].r)/2; tree[i*2].sum+=tree[i].lz*(mid-tree[i*2].l+1); tree[i*2+1].sum+=tree[i].lz*(tree[i*2+1].r-mid); tree[i].lz=0; } return ; } inline void add(ll i,ll l,ll r,ll k) {
if(tree[i].l>=l&&tree[i].r<=r) {
tree[i].sum+=k*(tree[i].r-tree[i].l+1); tree[i].lz+=k; return ; } push_down(i); if(tree[i*2].r>=l) add(i*2,l,r,k); if(tree[i*2+1].l<=r) add(i*2+1,l,r,k); tree[i].sum=tree[i*2].sum+tree[i*2+1].sum; return ; } inline ll searchs(ll i,ll l, ll r) {
if(tree[i].l>=l&&tree[i].r<=r) return tree[i].sum; push_down(i); ll num=0; if(tree[i*2].r>=l) num+=searchs(i*2,l,r); if(tree[i*2+1].l<=r) num+=searchs(i*2+1,l,r); return num; } int main() {
ios::sync_with_stdio(false); cin.tie(0),cout.tie(0); cin>>n>>m; for(int i=1;i<=n;++i) cin>>arr[i]; build(1,1,n,arr);//从根节点开始建树 for(int i=1;i<=m;++i) {
ll tmp; cin>>tmp; if(tmp==1) {
ll a,b,c; cin>>a>>b>>c; add(1,a,b,c);//每次修改都是从编号为1开始的,因为编号1是树的顶端,往下分叉 } if(tmp==2) {
ll a,b; cin>>a>>b; printf("%lld\n",searchs(1,a,b));//编号i的话,每次都是从1开始 } } return 0; }
区间乘法:洛谷线段树模板2
P3373 【模板】线段树 2
#include
using namespace std; typedef long long ll; const ll N=1e6+7; template<typename T>void read(T &x) {
x=0;char ch=getchar();ll f=1; while(!isdigit(ch)){
if(ch=='-')f=-1;ch=getchar();} while(isdigit(ch)){
x=(x<<1)+(x<<3)+(ch^48);ch=getchar();}x*=f; } ll n,m,arr[N],mod,flag,cn,cm,cw; struct node {
ll l,r,sum,mul,add;//有乘有加,先乘后加 }tree[N]; inline void build(ll i,ll l,ll r) {
tree[i].l=l; tree[i].r=r; tree[i].mul=1; if(l==r) {
tree[i].sum=arr[l]%mod; return ; } ll mid=(l+r)>>1; build(i*2,l,mid); build(i*2+1,mid+1,r); tree[i].sum=(tree[i*2].sum+tree[i*2+1].sum)%mod; } inline void push_down(ll i) {
tree[i*2].sum=(ll)(tree[i].mul*tree[i*2].sum+((tree[i*2].r-tree[i*2].l+1)*tree[i].add)%mod)%mod; tree[i*2+1].sum=(ll)(tree[i].mul*tree[i*2+1].sum+((tree[i*2+1].r-tree[i*2+1].l+1)*tree[i].add)%mod)%mod; tree[i*2].mul=(ll)(tree[i*2].mul*tree[i].mul)%mod; tree[i*2+1].mul=(ll)(tree[i*2+1].mul*tree[i].mul)%mod; tree[i*2].add=(ll)(tree[i*2].add*tree[i].mul+tree[i].add)%mod; tree[i*2+1].add=(ll)(tree[i*2+1].add*tree[i].mul+tree[i].add)%mod; tree[i].mul=1;tree[i].add=0; } inline void add(ll i,ll l,ll r,ll k) {
if(tree[i].l>=l&&tree[i].r<=r) {
tree[i].add=(ll)(tree[i].add+k)%mod; tree[i].sum=(ll)(tree[i].sum+k*(tree[i].r-tree[i].l+1))%mod; return ; } push_down(i); if(tree[i*2].r>=l) add(i*2,l,r,k); if(tree[i*2+1].l<=r) add(i*2+1,l,r,k); //ll mid=(tree[i].l+tree[i].r)>>1; //if(l<=mid)add(i*2,l,r,k); //if(mid
tree
[i
]
.sum
=
(tree
[i
*
2
]
.sum
+tree
[i
*
2
+
1
]
.sum
)
%mod
;
}
inline
void
mult
(ll i
,ll l
,ll r
,ll k
)
{
if
(tree
[i
]
.l
>=l
&&tree
[i
]
.r
<=r
)
{
tree
[i
]
.mul
=
(tree
[i
]
.mul
*k
)
%mod
; tree
[i
]
.add
=
(tree
[i
]
.add
*k
)
%mod
; tree
[i
]
.sum
=
(tree
[i
]
.sum
*k
)
%mod
;
return
;
}
push_down
(i
)
;
if
(tree
[i
*
2
]
.r
>=l
)
mult
(i
*
2
,l
,r
,k
)
;
if
(tree
[i
*
2
+
1
]
.l
<=r
)
mult
(i
*
2
+
1
,l
,r
,k
)
;
//ll mid=(tree[i].l+tree[i].r)>>1;
//if(l<=mid)mult(i*2,l,r,k);
//if(mid
tree
[i
]
.sum
=
(tree
[i
*
2
]
.sum
+tree
[i
*
2
+
1
]
.sum
)
%mod
;
}
inline ll
query
(ll i
,ll l
,ll r
)
{
if
(tree
[i
]
.l
>=l
&&tree
[i
]
.r
<=r
)
return tree
[i
]
.sum
;
push_down
(i
)
; ll num
=
0
;
if
(tree
[i
*
2
]
.r
>=l
) num
=
(num
+
query
(i
*
2
,l
,r
)
)
%mod
;
if
(tree
[i
*
2
+
1
]
.l
<=r
) num
=
(num
+
query
(i
*
2
+
1
,l
,r
)
)
%mod
;
return num
;
//ll val=0;
//ll mid=(tree[i].l+tree[i].r)>>1;
//if(l<=mid)val=(val+query(i*2,l,r))%mod;
//if(mid
//return val;
}
int
main
(
)
{
read
(n
)
,
read
(m
)
,
read
(mod
)
;
for
(
int i
=
1
;i
<=n
;
++i
)
read
(arr
[i
]
)
;
build
(
1
,
1
,n
)
;
for
(
int i
=
1
;i
<=m
;
++i
)
{
read
(flag
)
;
if
(flag
==
1
)
{
read
(cn
)
,
read
(cm
)
,
read
(cw
)
;
mult
(
1
,cn
,cm
,cw
)
;
}
else
if
(flag
==
2
)
{
read
(cn
)
,
read
(cm
)
,
read
(cw
)
;
add
(
1
,cn
,cm
,cw
)
;
}
else
{
read
(cn
)
,
read
(cm
)
; cout
<<
query
(
1
,cn
,cm
)
<<endl
;
}
}
}
/* 5 4 1000 1 2 3 4 5 3 1 5 2 1 5 1 1 1 5 2 3 1 5 */
区间根号,bzoj3211
#include
#include
#include
#include
#include
#define MAXN #define REP(i,k,n) for(int i=k;i<=n;i++) #define in(a) a=read() using namespace std; int read(){
int x=0,f=1; char ch=getchar(); for(;!isdigit(ch);ch=getchar()) if(ch=='-') f=-1; for(;isdigit(ch);ch=getchar()) x=x*10+ch-'0'; return x*f; } struct node{
int l,r; long long lz,sum,maxx,minn; }tree[MAXN<<2]; int n,m,input[MAXN]; inline void build(int i,int l,int r){
tree[i].l=l;tree[i].r=r; if(l==r){
tree[i].sum=tree[i].minn=tree[i].maxx=input[l]; return ; } int mid=(l+r)>>1; build(i*2,l,mid); build(i*2+1,mid+1,r); tree[i].sum=tree[i*2].sum+tree[i*2+1].sum; tree[i].minn=min(tree[i*2].minn,tree[i*2+1].minn); tree[i].maxx=max(tree[i*2].maxx,tree[i*2+1].maxx); return ; } inline void push_down(int i){
if(!tree[i].lz) return ; long long k=tree[i].lz; tree[i*2].lz+=k; tree[i*2+1].lz+=k; tree[i*2].sum-=(tree[i*2].r-tree[i*2].l+1)*k; tree[i*2+1].sum-=(tree[i*2+1].r-tree[i*2+1].l+1)*k; tree[i*2].minn-=k; tree[i*2+1].minn-=k; tree[i*2].maxx-=k; tree[i*2+1].maxx-=k; tree[i].lz=0; return ; } inline void Sqrt(int i,int l,int r){
if(tree[i].l>=l && tree[i].r<=r && (tree[i].minn-(long long)sqrt(tree[i].minn))==(tree[i].maxx-(long long)sqrt(tree[i].maxx))){
long long u=tree[i].minn-(long long)sqrt(tree[i].minn); tree[i].lz+=u; tree[i].sum-=(tree[i].r-tree[i].l+1)*u; tree[i].minn-=u; tree[i].maxx-=u; //cout<<"i"<
return
;
}
if
(tree
[i
]
.r
<l
|| tree
[i
]
.l
>r
)
return
;
push_down
(i
)
;
if
(tree
[i
*
2
]
.r
>=l
)
Sqrt
(i
*
2
,l
,r
)
;
if
(tree
[i
*
2
+
1
]
.l
<=r
)
Sqrt
(i
*
2
+
1
,l
,r
)
; tree
[i
]
.sum
=tree
[i
*
2
]
.sum
+tree
[i
*
2
+
1
]
.sum
; tree
[i
]
.minn
=
min
(tree
[i
*
2
]
.minn
,tree
[i
*
2
+
1
]
.minn
)
; tree
[i
]
.maxx
=
max
(tree
[i
*
2
]
.maxx
,tree
[i
*
2
+
1
]
.maxx
)
;
//cout<<"i"<
return
;
}
inline
long
long
search
(
int i
,
int l
,
int r
)
{
if
(tree
[i
]
.l
>=l
&& tree
[i
]
.r
<=r
)
return tree
[i
]
.sum
;
if
(tree
[i
]
.r
<l
|| tree
[i
]
.l
>r
)
return
0
;
push_down
(i
)
;
long
long s
=
0
;
if
(tree
[i
*
2
]
.r
>=l
) s
+=
search
(i
*
2
,l
,r
)
;
if
(tree
[i
*
2
+
1
]
.l
<=r
) s
+=
search
(i
*
2
+
1
,l
,r
)
;
return s
;
}
int
main
(
)
{
in
(n
)
;
REP
(i
,
1
,n
)
in
(input
[i
]
)
;
build
(
1
,
1
,n
)
;
in
(m
)
;
int a
,b
,c
;
REP
(i
,
1
,m
)
{
in
(a
)
;
in
(b
)
;
in
(c
)
;
if
(a
==
1
)
printf
(
"%lld\n"
,
search
(
1
,b
,c
)
)
;
if
(a
==
2
)
{
Sqrt
(
1
,b
,c
)
;
//for(int i=1;i<=7;i++)
// cout<
// cout<
}
}
}
发布者:全栈程序员-站长,转载请注明出处:https://javaforall.net/178009.html原文链接:https://javaforall.net
