【模板】线段树区间修改

区间修改:

区间修改过程类似于区间询问,例如将[ul, ur]内的所有元素都加上v,则进行如下操作:

当当前区间被区间[ul, ur]所包含时,

  当前的节点值加上区间长度(r - l  + 1)乘以v

  对当前节点的lazy-tag加上v,修改结束

否则,将当前节点的lazy-tag下传,分别修改左孩子和右孩子(一定条件下),然后更新此节点的值

lazy-tag下传:

如果当前节点没有lazy-tag,直接return

否则:1.  将左孩子右孩子分别加上lazy-tag * 区间长度的值

    2.  左孩子和右孩子的懒惰标记加上lazy-tag

    3.  当前节点的lazy-tag清零

注意区间询问的时候也要lazy-tag下传

【代码:】

 1 //线段树区间修改
 2 #include<iostream>
 3 #include<cstdio>
 4 using namespace std;
 5
 6 const int MAXN = 100000 + 1;
 7
 8 int n, m, a[MAXN];
 9 struct Segment {
10     long long sum, lazy;
11     Segment() {
12         sum = lazy = 0;
13     }
14 }t[MAXN << 2];
15
16 void Build(int o, int l, int r) {
17     if(l == r) t[o].sum = a[l];
18     else {
19         int mid = (l + r) >> 1;
20         Build(o << 1, l, mid);
21         Build(o << 1|1, mid + 1, r);
22         t[o].sum = t[o << 1].sum + t[o << 1|1].sum;
23     }
24 }
25 void down(int o, int len) {
26     if(!t[o].lazy) return;
27     t[o << 1].sum += t[o].lazy * (len - (len >> 1));
28     t[o << 1|1].sum += t[o].lazy * (len >> 1);
29     t[o << 1].lazy += t[o].lazy;
30     t[o << 1|1].lazy += t[o].lazy;
31     t[o].lazy = 0;
32 }
33 void Update(int o, int l, int r, int ul, int ur, int v) {
34     if(ul <= l && r <= ur) {
35         t[o].sum += (r - l + 1) * v;
36         t[o].lazy += v;
37     }
38     else {
39         down(o, r - l + 1);
40         int mid = (l + r) >> 1;
41         if(ul <= mid) Update(o << 1, l, mid, ul, ur, v);
42         if(ur > mid) Update(o << 1|1, mid + 1, r, ul, ur, v);
43         t[o].sum = t[o << 1].sum + t[o << 1|1].sum;
44     }
45 }
46 long long Query(int o, int l, int r, int ql, int qr) {
47     if(ql <= l && r <= qr) {
48         return t[o].sum;
49     }
50     int mid = (l + r) >> 1;
51     long long ret = 0;
52     down(o, r - l + 1);
53     if(ql <= mid) ret += Query(o << 1, l, mid, ql, qr);
54     if(qr > mid) ret += Query(o << 1|1, mid + 1, r, ql, qr);
55     return ret;
56 }
57 int main() {
58     scanf("%d%d", &n, &m);
59     for(int i = 1; i <= n; ++i)
60         scanf("%d", &a[i]);
61     Build(1, 1, n);
62     while(m--) {
63         int fl, x, y;
64         scanf("%d%d%d", &fl, &x, &y);
65         if(fl == 1) {
66             int k;
67             scanf("%d", &k);
68             Update(1, 1, n, x, y, k);
69         }
70         else printf("%lld\n", Query(1, 1, n, x, y));
71     }
72 }

模板测试题目传送门

原文地址:https://www.cnblogs.com/devilk-sjj/p/9044841.html

时间: 2024-10-01 12:35:16

【模板】线段树区间修改的相关文章

算法模板——线段树区间修改区间求和

该模板实现的功能——进行区间的乘法和加法,以及区间的求和(1:乘法 2:加法 3:求和)详见BZOJ1798 1 type 2 vet=record 3 a0,a1:int64; 4 end; 5 var 6 i,j,k,l,m,n,a2,a3,a4:longint; 7 p:int64; 8 d1,d2,d:vet; 9 a,c:array[0..1000000] of int64; 10 b:array[0..1000000] of vet; 11 function min(x,y:long

线段树区间修改模板

本来打算把大白书第三章一口气攻下来的,但是这个线段树也是卡了好久. 不敢过题太快,怕自己走马观花到头来结果什么都不会. 可也不能再拖了,在做题中也许有更多的体会. 模板一:1 L R v 表示区间[L, R]所有元素都加上v2 L R   表示查询区间[L, R]的sum, min, maxsumv[o]的定义为:如果只执行节点o及其子孙节点的中的add操作,节点o对应区间中所有数之和 1 //线段树区间修改 2 //1 L R v 表示区间[L, R]所有元素都加上v 3 //2 L R 表示

hdu1698 Just a Hook(线段树+区间修改+区间查询+模板)

Time Limit: 4000/2000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)Total Submission(s): 54923    Accepted Submission(s): 25566 Problem Description In the game of DotA, Pudge’s meat hook is actually the most horrible thing for most of

线段树区间修改 P3372 【模板】线段树 1

题目描述 如题,已知一个数列,你需要进行下面两种操作: 1.将某区间每一个数加上x 2.求出某区间每一个数的和 输入输出格式 输入格式: 第一行包含两个整数N.M,分别表示该数列数字的个数和操作的总个数. 第二行包含N个用空格分隔的整数,其中第i个数字表示数列第i项的初始值. 接下来M行每行包含3或4个整数,表示一个操作,具体如下: 操作1: 格式:1 x y k 含义:将区间[x,y]内每个数加上k 操作2: 格式:2 x y 含义:输出区间[x,y]内每个数的和 输出格式: 输出包含若干行整

poj 2777 Count Color(线段树区间修改)

题目链接:http://poj.org/problem?id=2777 题目意思:就是问你在询问的区间里有几种不同的颜色 思路:这题和一般的区间修改差不多,但是唯一不同的就是我们要怎么计算有种颜色,所以这时候我们就需要把延时标记赋予不同的意义,当某段区间有多种颜色时就赋值为-1,当为一种颜色时就把它赋值为这个颜色的号数.这儿我们要怎么统计询问区间不同的颜色数叻,为了不重复计算同一种颜色,那么我们就需要用一个数组来标记计算过的颜色,当我们下次遇到时就不需要再次计算了.... 代码核心处就在计数那儿

hdu-5023 A Corrupt Mayor&#39;s Performance Art (线段树区间修改)

今天集训队打比赛的一道题,很明显是个线段树,我们队照着lrj蓝书敲了一通,机智的将修改值和加和改成了位运算:|= 但是好像哪里出了点小问题,就是不对,赛后又水了一遍,竟然过了...发现还是lrj的书好啊,市面上的模板一点也不好用,连区间修改都没有 . 等集训完了要静心好好系统的学习一下线段树 . 多看多刷lrj的书 . 细节参见代码: #include<bits/stdc++.h> using namespace std; const int maxn = 1000000 + 5; int n

【线段树区间修改】fzu2105Digits Count

/* 题意: 给出数组A,有以下几个操作: 1: AND(opn, L, R):把区间[L, R]中的元素A[i]改为A[i] & opn;;;;;; 2: OR(opn, L, R) :把区间[L, R]中的元素A[i]改为A[i] | opn;;;;;;; 3: XOR(opn, L, R):把区间[L, R]中的元素A[i]改为A[i] ^ opn;;;;;;; 4: SUM(L, R) :对区间[L, R]中的元素求和:::: ------------------------------

Wikilo 1191线段树区间修改单点查询

这题也算比较容易的了. 如果哪个区间已经没有黑色的话,就不用update了,就是因为这个原因WA了2发,唉-- #include <iostream> #include <cstdio> #include <algorithm> #include <cmath> #include <deque> #include <vector> #include <queue> #include <string> #incl

HDU - 1698 Just a Hook (线段树区间修改)

Description In the game of DotA, Pudge's meat hook is actually the most horrible thing for most of the heroes. The hook is made up of several consecutive metallic sticks which are of the same length. Now Pudge wants to do some operations on the hook.