线段树--区间乘法和区间加法

假设有编号从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;
}
  • 2
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值