假设有编号从1到n的n个点,每个点都存了一些信息,用[L,R]表示下标从L到R的这些点。
线段树的用处就是,对编号连续的一些点进行修改或者统计操作,修改和统计的复杂度都是O(log2(n)).
线段树的原理,就是,将[1,n]分解成若干特定的子区间(数量不超过4*n),然后,将每个区间[L,R]都分解为少量特定的子区间,通过对这些少量子区间的修改或者统计,来实现快速对[L,R]的修改或者统计。
由此看出,用线段树统计的东西,必须符合区间加法,否则,不可能通过分成的子区间来得到[L,R]的统计结果。
符合区间加法的例子:
数字之和 总数字之和 = 左区间数字之和 + 右区间数字之和
最大公因数(GCD) 总GCD = gcd( 左区间GCD , 右区间GCD )
最大值 总最大值=max(左区间最大值,右区间最大值)
不符合区间加法的例子:
众数 只知道左右区间的众数,没法求总区间的众数
01序列的最长连续零 只知道左右区间的最长连续零,没法知道总的最长连续零
结构体线段树实现:
int w[N];//存放节点的权值
struct node {
int l, r;//左右的范围
ll sum;//总和
ll add;//懒标记
} tree[N << 2];//四倍
1.push_up()函数
void push_up(int root) { //利用左右子树信息更新当前节点信息
tree[root].sum = tree[root << 1].sum + tree[root << 1 | 1].sum;
}
2.build()函数建树
void build(int root, int l, int r) {
if (l == r)tree[root] = { l,r,w[r],0 }; //填入四个信息,其中w[r]为值
else {
tree[root] = { l,r }; //初始化信息
int mid = l + r >> 1;
build(root << 1, l, mid);//构建左子树
build(root << 1 | 1, mid + 1, r);//构建右子树
push_up(root);//构建完成左右子树后更新当前节点信息
}
}
3.push_down()函数下放懒标记
void push_down(int root) { //下放当前节点的懒标记
auto& Root = tree[root], & left = tree[root << 1], & right = tree[root << 1 | 1];
if (Root.add) {//下放只是下放到当前节点的左右子节点
left.add += Root.add, left.sum += (ll)(left.r - left.l + 1) * Root.add;//首先是懒标记下放,然后左右节点的和要更新
right.add += Root.add, right.sum += (ll)(right.r - right.l + 1) * Root.add;//都是+=因为可能子节点本身有懒标记
Root.add = 0;//记得将懒标记清0
}
}
4.modify()修改函数
void modify(int root, int l, int r, int num) {
if (l <= tree[root].l && tree[root].r <= r) {//如果当前节点包含的区间在修改区间内,直接修改节点的信息即可
tree[root].sum += (ll)(tree[root].r - tree[root].l + 1) * num;//修改该节点的和为区间长度乘以加数
tree[root].add += num;//修改该节点的懒标记
}
else { //如果不完全包含
push_down(root);//下放懒标记
int mid = tree[root].l + tree[root].r >> 1;//计算当前节点所包含区间的中值
if (l <= mid)modify(root << 1, l, r, num);//节点的左边区间和当前修改区间有交集
if (r > mid)modify(root << 1 | 1, l, r, num);//节点的右边区间和当前修改区间有交集
push_up(root);//记得push_up操作更新当前节点
}
}
5.query()查询函数
ll query(int root, int l, int r) {
//递归尾(查询区间大于节点区间,节点是其中一部分直接返回)
if (l <= tree[root].l && tree[root].r <= r)return tree[root].sum;
//(查询区间小于节点区间,在节点区间内进行细化)
push_down(root);//没有的话就下放懒标记
int mid = tree[root].l + tree[root].r >> 1;//计算当前节点所包含区间的中值
if (r <= mid)return query(root << 1, l, r);//如果查询区间完全在节点的左区间
else if (l > mid)return query(root << 1 | 1, l, r);//如果查询区间完全在节点的右区间
else {//如果左右区间都包含有
ll left = query(root << 1, l, r);
ll right = query(root << 1 | 1, l, r);
ll ans;
//push_up
ans = left + right;
return ans;
}
}
洛谷 P3372 【模板】线段树 1
区间查询,区间修改(加)
结构体版
#include<bits/stdc++.h>
using namespace std;
template<class...Args>
void debug(Args... args) {//Parameter pack
auto tmp = { (cout << args << ' ', 0)... };
cout << "\n";
}
typedef long long ll;
typedef unsigned long long ull;
typedef pair<ll, ll>pll;
typedef pair<int, int>pii;
const int INF = 0x7fffffff;
const int N = 1e5 + 5;
int w[N];//存放权值
struct node {
int l, r;//左右的范围
ll sum;//总和
ll add;//懒标记
} tree[N << 2];
void push_up(int root) { //利用左右子树信息更新当前节点信息
tree[root].sum = tree[root << 1].sum + tree[root << 1 | 1].sum;
}
void push_down(int root) { //下放当前节点的懒标记
auto& Root = tree[root], & left = tree[root << 1], & right = tree[root << 1 | 1];
if (Root.add) {//下放只是下放当前节点的左右子节点
left.add += Root.add, left.sum += (ll)(left.r - left.l + 1) * Root.add;//首先是懒标记下放,然后左右节点的和要更新
right.add += Root.add, right.sum += (ll)(right.r - right.l + 1) * Root.add;//都是+=因为可能子节点本身有懒标记
Root.add = 0;//记得将懒标记清0
}
}
void build(int root, int l, int r) {
if (l == r)tree[root] = { l,r,w[r],0 }; //填入四个信息,其中w[r]为值
else {
tree[root] = { l,r }; //初始化信息
int mid = l + r >> 1;
build(root << 1, l, mid);//构建左子树
build(root << 1 | 1, mid + 1, r);//构建右子树
push_up(root);//构建完成左右子树后更新当前节点信息
}
}
void modify(int root, int l, int r, int num) {
if (l <= tree[root].l && tree[root].r <= r) {//如果当前节点包含的区间在修改区间内,直接修改节点的信息即可
tree[root].sum += (ll)(tree[root].r - tree[root].l + 1) * num;//修改该节点的和为区间长度乘以加数
tree[root].add += num;//修改该节点的懒标记
}
else { //如果不完全包含
push_down(root);//下放懒标记
int mid = tree[root].l + tree[root].r >> 1;//计算当前节点所包含区间的中值
if (l <= mid)modify(root << 1, l, r, num);//节点的左边区间和当前修改区间有交集
if (r > mid)modify(root << 1 | 1, l, r, num);//节点的右边区间和当前修改区间有交集
push_up(root);//记得push_up操作更新当前节点
}
}
ll query(int root, int l, int r) {
//递归尾(查询区间大于节点区间,节点是其中一部分直接返回)
if (l <= tree[root].l && tree[root].r <= r)return tree[root].sum;
//(查询区间小于节点区间,在节点区间内进行细化)
push_down(root);//没有的话就下放懒标记
int mid = tree[root].l + tree[root].r >> 1;//计算当前节点所包含区间的中值
if (r <= mid)return query(root << 1, l, r);//如果查询区间完全在节点的左区间
else if (l > mid)return query(root << 1 | 1, l, r);//如果查询区间完全在节点的右区间
else {//如果左右区间都包含有
ll left = query(root << 1, l, r);
ll right = query(root << 1 | 1, l, r);
ll ans;
//push_up
ans = left + right;
return ans;
}
}
int main() {
ios_base::sync_with_stdio(false), cin.tie(0);
ll n, m;
cin >> n >> m;
for (int i = 1; i <= n; i++)cin >> w[i];
build(1, 1, n);
while (m--) {
string str;
ll a, b, c;
cin >> str;
if (str[0] == '1') {
cin >> a >> b >> c;
modify(1, a, b, c);
}
else {
cin >> a >> b;
cout << query(1, a, b) << "\n";
}
}
return 0;
}
poj 3468 A Simple Problem with Integers
区间查询,区间修改(加)
非结构体版
//#include<bits/stdc++.h>
#include<iostream>
using namespace std;
template <typename T> void debug(string s, T x) { cout << s << "=" << x << "\n"; }
typedef long long ll;
typedef unsigned long long ull;
typedef pair<ll, ll>pll;
const ll N = 1e5 + 5;
const ll MOD = 1e9 + 7;
const ll INF = 0x7fffffff;
long long sum[N << 2], add[N << 2];
void push_up(ll rt) {//向上更新,通过当前结点rt把值递归但父结点
sum[rt] = sum[rt << 1] + sum[rt << 1 | 1];
}
void push_down(ll rt, ll m) {//更新rt的子结点
if (add[rt]) {
add[rt << 1] += add[rt];
add[rt << 1 | 1] += add[rt];
sum[rt << 1] += (m - (m >> 1)) * add[rt];
sum[rt << 1 | 1] += (m >> 1) * add[rt];
add[rt] = 0;//取消本层标记
}
}
void build(ll l, ll r, ll rt) { //用满二叉树建树
add[rt] = 0;
if (l == r) { //叶子结点,赋值
cin >> sum[rt];
return;
}
ll mid = (l + r) >> 1;
build(l, mid, rt << 1);
build(mid + 1, r, rt << 1 | 1);
push_up(rt); //向上更新区间和
}
void update(ll a, ll b, ll c, ll l, ll r, ll rt) {//区间更新
if (a <= l && b >= r) {
sum[rt] += (r - l + 1) * c;
add[rt] += c;
return;
}
push_down(rt, r - l + 1);//向下更新
ll mid = (l + r) >> 1;
if (a <= mid) update(a, b, c, l, mid, rt << 1); //分成两半,继续深入
if (b > mid) update(a, b, c, mid + 1, r, rt << 1 | 1);
push_up(rt);//向上更新
}
ll query(ll a, ll b, ll l, ll r, ll rt) { //区间求和
if (a <= l && b >= r) return sum[rt]; //满足lazy,直接返回值
push_down(rt, r - l + 1); //向下更新
ll mid = (l + r) >> 1;
ll ans = 0;
if (a <= mid) ans += query(a, b, l, mid, rt << 1);
if (b > mid) ans += query(a, b, mid + 1, r, rt << 1 | 1);
return ans;
}
int main() {
ios_base::sync_with_stdio(false), cin.tie(0);
ll n, m;
cin >> n >> m;
build(1, n, 1);
while (m--) {
string str;
ll a, b, c;
cin >> str;
if (str[0] == 'C') {
cin >> a >> b >> c;
update(a, b, c, 1, n, 1);
}
else {
cin >> a >> b;
cout << query(a, b, 1, n, 1)<<"\n";
}
}
return 0;
}
区间加法
P3373 【模板】线段树 2
#include<bits/stdc++.h>
#include <unordered_map>
using namespace std;
template<class...Args>
void debug(Args... args) {//Parameter pack
auto tmp = { (cout << args << ' ', 0)... };
cout << "\n";
}
typedef long long ll;
typedef unsigned long long ull;
typedef pair<ll, ll>pll;
typedef pair<int, int>pii;
const ll N = 1e5 + 5;
//const ll MOD = 998244353;
const ll INF = 0x7fffffff;
int MOD;
int w[N];//存放权值
struct node {
int l, r;//左右的范围
ll sum;//总和
ll mul;//乘法懒标记
ll add;//加法懒标记
} tree[N << 2];
void push_up(int root) { //利用左右子树信息更新当前节点信息
tree[root].sum = (tree[root << 1].sum + tree[root << 1 | 1].sum) % MOD;
}
void push_down(int root) { //下放当前节点的懒标记
auto& Root = tree[root], & left = tree[root << 1], & right = tree[root << 1 | 1];
//例子:(x*a+b)*c+d=x*ac+bc+d
left.sum = (left.sum * Root.mul + (left.r - left.l + 1) * Root.add) % MOD;
left.add = (left.add * Root.mul + Root.add) % MOD;
left.mul = left.mul * Root.mul % MOD;
right.sum = (right.sum * Root.mul + (right.r - right.l + 1) * Root.add) % MOD;
right.add = (right.add * Root.mul + Root.add) % MOD;
right.mul = right.mul * Root.mul % MOD;
//重置懒标记
Root.add = 0;
Root.mul = 1;
}
void build(int root, int l, int r) {
if (l == r)tree[root] = { l,r,w[r],1,0 };
else {
tree[root] = { l,r ,0,1,0}; //初始化信息
int mid = l + r >> 1;
build(root << 1, l, mid);//构建左子树
build(root << 1 | 1, mid + 1, r);//构建右子树
push_up(root);//构建完成左右子树后更新当前节点信息
}
}
void modify(int root, int l, int r, int mul,int add) {
if (l <= tree[root].l && tree[root].r <= r) {//如果当前节点包含的区间在修改区间内,直接修改节点的信息即可
//修改该节点的和
tree[root].sum = (tree[root].sum * mul + (tree[root].r - tree[root].l + 1) * add) % MOD;
//修改该节点的懒标记
tree[root].add = (tree[root].add * mul + add) % MOD;
tree[root].mul = tree[root].mul * mul % MOD;
}
else { //如果不完全包含
push_down(root);//下放懒标记
int mid = tree[root].l + tree[root].r >> 1;//计算当前节点所包含区间的中值
if (l <= mid)modify(root << 1, l, r, mul,add);//节点的左边区间和当前修改区间有交集
if (r > mid)modify(root << 1 | 1, l, r, mul,add);//节点的右边区间和当前修改区间有交集
push_up(root);//记得push_up操作更新当前节点
}
}
ll query(int root, int l, int r) {
//递归尾(查询区间大于节点区间,节点是其中一部分直接返回)
if (l <= tree[root].l && tree[root].r <= r)return tree[root].sum;
//(查询区间小于节点区间,在节点区间内进行细化)
push_down(root);//没有的话就下放懒标记
int mid = tree[root].l + tree[root].r >> 1;//计算当前节点所包含区间的中值
if (r <= mid)return query(root << 1, l, r);//如果查询区间完全在节点的左区间
else if (l > mid)return query(root << 1 | 1, l, r);//如果查询区间完全在节点的右区间
else {//如果左右区间都包含有
ll left = query(root << 1, l, r);
ll right = query(root << 1 | 1, l, r);
ll ans;
//push_up
ans = (left + right) % MOD;
return ans;
}
}
int main() {
ios_base::sync_with_stdio(false); cin.tie(0); cout.tie(0);
int n, m;
cin >> n >> m >> MOD;
for (int i = 1; i <= n; i++)cin >> w[i];
build(1, 1, n);
while (m--) {
int op;
cin >> op;
if (op == 1) {
int l, r, mul;
cin >> l >> r >> mul;
modify(1, l, r, mul, 0);
}
else if (op == 2) {
int l, r, add;
cin >> l >> r >> add;
modify(1, l, r, 1, add);
}
else {
int l, r;
cin >> l >> r;
cout << query(1, l, r) << "\n";
}
}
return 0;
}