洛谷P3372【模板】线段树1(范围增加,维护区间和)
题意: 维护区间和,简单区间修改(范围增加)。输入 1 为修改操作,输入 2 为查询操作。
思路: 直接建立线段树维护区间和,使用lazy tag维护线段树。
注意: 在进行区间修改时记得下传lazy tag并修改子区间范围的值,否则up上传信息修改父区间范围值时可能会产生错误。
代码:
// LH
#include<bits/stdc++.h>
using namespace std;
#define int long long
typedef long long ll;
typedef pair<int,int> PII;
typedef pair<ll,ll> PLL;
typedef vector<int> VI;
typedef vector<ll> VL;
typedef vector<PII> VPII;
typedef vector<PLL> VPLL;
//线段树1(洛谷p3372)
//维护区间和 ,范围增加 , lazy tag
const int N = 1e5 + 5;
int ls(int x) {return x << 1;}
int rs(int x) {return x << 1 | 1;}
int tree[N<<2] , tag[N<<2];
void up(int p) {tree[p] = tree[ls(p)] + tree[rs(p)];}//信息汇总,在修改了子区间范围的值后导致父范围的值发生改变应该将子范围的信息重新汇总到父范围
void down(int p , int l , int r) { //传递lazy tag父范围已经改变过,当需要查询或改变子区间范围值时,将父范围的lazy tag下传,并修改子区间范围的值
if(tag[p]) {
int mid = l + r >> 1;
tree[ls(p)] += tag[p] * (mid - l + 1);
tree[rs(p)] += tag[p] * (r - mid);
tag[ls(p)] += tag[p];
tag[rs(p)] += tag[p];
tag[p] = 0;
}
}
void build(int l , int r , int p) { //建树
if(l == r) {
cin >> tree[p];
return;
}
int mid = l + r >> 1;
build(l , mid , ls(p));
build(mid + 1 , r , rs(p));
up(p);
}
int query(int l , int r , int p , int lx , int rx) { //查询时若需查询子区间应将lazy tag下传,并根据lazy tag修改子区间范围的值
if(lx <= l && r <= rx) {
return tree[p];
}
down(p , l , r);
int mid = l + r >> 1;
int sum = 0;
if(lx <= mid) sum += query(l , mid , ls(p) , lx , rx);
if(rx > mid) sum += query(mid+1 , r , rs(p) , lx , rx);
return sum;
}
void add(int l , int r , int p , int lx , int rx , int k) { //修改(此处为范围增加),在需要修改子区间范围时,记得下传lazy tag
if(lx <= l && r <= rx) {
tree[p] += (r - l + 1) * k;
tag[p] += k;
return;
}
down(p , l , r); //不要忘了下传lazy tag否则可能会导致子范围未正确修改,up函数上传错误信息至父范围
int mid = l + r >> 1;
if(lx <= mid) add(l , mid , ls(p) , lx , rx , k);
if(rx > mid) add(mid+1 , r , rs(p) , lx , rx , k);
up(p);
}
void solve(){
int n , m;
cin >> n >> m;
build(1 , n , 1);
while(m --) {
int op;
cin >> op;
if(op == 1) {
int x , y , k;
cin >> x >> y >> k;
add(1 , n , 1 , x , y , k);
}else {
int x , y;
cin >> x >> y;
cout << query(1 , n , 1 , x , y) << '\n';
}
}
}
signed main() {
std::ios::sync_with_stdio(false);
cin.tie(0);cout.tie(0);
int t = 1;
//cin >> t;
while(t--){
solve();
}
return 0;
}