线段树(区间更新求和)

//poj 3468
#include <iostream>
using namespace std;
#define MAX 100100

struct node
{
    int left, right;
    //区间和
    long long sum;
    //增加量
    long long add;
    //求分割点
    int get_mid() {
        return (left + right) / 2;
    }
    //求区间长度
    int get_dis() {
        return right - left + 1;
    }
};

node tree[MAX * 3];
long long d[MAX];

//建树
long long build(int left, int right, int loc)
{
    int LL = loc << 1;
    int RR = LL | 1;
    tree[loc].left = left;
    tree[loc].right = right;
    //初始增量为零
    tree[loc].add = 0;
    if (left == right) {
        return tree[loc].sum = d[left];       
    }
    int mid = tree[loc].get_mid();
    return tree[loc].sum = build(left, mid, LL) + build(mid + 1, right, RR);
}

//更新节点
void update(int left, int right, long long value, int loc)
{
    int LL = loc << 1;
    int RR = LL | 1;
    //如果是某个节点,直接增加增量。
    if (tree[loc].left == left && tree[loc].right == right)
    {
        tree[loc].add += value;
        return ;
    }
   
    //不管要更新的区间位于什么位置上,根节点的值必定增加((right - left + 1) * value);
    tree[loc].sum += (long long)(right - left + 1) * value;
   
    //递归更新左右节点
    int mid = tree[loc].get_mid();
    if (right <= mid) {
        update(left, right, value, LL);
    } else if (mid < left) {
        update(left, right, value, RR);
    } else {
        update(left, mid, value, LL);
        update(mid + 1, right, value, RR);
    }
}

//查询区间值
long long query(int left, int right, int loc)
{
    int LL = loc << 1;
    int RR = LL | 1;
    //刚好查询的区间和是一个节点的区间和
    if (tree[loc].left == left && tree[loc].right == right) {
        return tree[loc].sum + (long long)tree[loc].get_dis() * tree[loc].add;
    }
   
    //区间有增量
    if (tree[loc].add) {
        //先更新本节点的区间和
        tree[loc].sum += (long long)tree[loc].get_dis() * tree[loc].add;
        //增量向左右节点传递
        tree[LL].add += tree[loc].add;
        tree[RR].add += tree[loc].add;
        //本节点增量设置为零
        tree[loc].add = 0;
    }
    //递归查询左右子树
    int mid = tree[loc].get_mid();
    if (right <= mid) {
        return query(left, right, LL);
    } else if (mid < left) {
        return query(left, right, RR);
    } else {
        return query(left, mid, LL) + query(mid + 1, right, RR);
    }
}


int main()
{
#ifndef ONLINE_JUDGE
    freopen("3468.txt", "r", stdin);
#endif
    int n, q;
    while (cin >> n >> q)
    {
        int i;
        for (i = 1; i <= n; i++) {
            scanf("%lld", &d[i]);
        }
        build(1, n, 1);
        char op[5];
        int a, b, c;
        while (q--) {
            scanf("%s", &op);
            if (op[0] == 'Q') {
                scanf("%d %d", &a, &b);
                if (a > b) {
                    int temp = a;
                    a = b;
                    b = temp;
                }
                printf("%lld\n", query(a, b, 1));
            } else {
                scanf("%d %d %lld", &a, &b, &c);
                if (a > b) {
                    int temp = a;
                    a = b;
                    b = temp;
                }
                update(a, b, c, 1);
            }
        }
    }
    return 0;
}

  • 2
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
电子图书资源服务系统是一款基于 Java Swing 的 C-S 应用,旨在提供电子图书资源一站式服务,可从系统提供的图书资源中直接检索资源并进行下载。.zip优质项目,资源经过严格测试可直接运行成功且功能正常的情况才上传,可轻松copy复刻,拿到资料包后可轻松复现出一样的项目。 本人系统开发经验充足,有任何使用问题欢迎随时与我联系,我会及时为你解惑,提供帮助。 【资源内容】:包含完整源码+工程文件+说明(若有),项目具体内容可查看下方的资源详情。 【附带帮助】: 若还需要相关开发工具、学习资料等,我会提供帮助,提供资料,鼓励学习进步。 【本人专注计算机领域】: 有任何使用问题欢迎随时与我联系,我会及时解答,第一时间为你提供帮助,CSDN博客端可私信,为你解惑,欢迎交流。 【适合场景】: 相关项目设计中,皆可应用在项目开发、毕业设计、课程设计、期末/期中/大作业、工程实训、大创等学科竞赛比赛、初期项目立项、学习/练手等方面中 可借鉴此优质项目实现复刻,也可以基于此项目进行扩展来开发出更多功能 【无积分此资源可联系获取】 # 注意 1. 本资源仅用于开源学习和技术交流。不可商用等,一切后果由使用者承担。 2. 部分字体以及插图等来自网络,若是侵权请联系删除。积分/付费仅作为资源整理辛苦费用。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值