码迷,mamicode.com
首页 > 其他好文 > 详细

秋实大哥与花 线段树模板

时间:2019-11-29 19:43:38      阅读:66      评论:0      收藏:0      [点我收藏+]

标签:tor   val   llb   query   ons   pre   inpu   update   模板   

地址 http://121.48.165.90/problem/qiu-shi-da-ge-yu-hua/description

秋实大哥是一个儒雅之人,昼听笙歌夜醉眠,若非月下即花前。

所以秋实大哥精心照料了很多花朵。现在所有的花朵排成了一行,每朵花有一个愉悦值。

秋实大哥每天要对着某一段连续的花朵歌唱,然后这些花朵的愉悦值都会增加一个相同的值vv(vv可能为负)。

同时他想知道每次他唱完歌后这一段连续的花朵的愉悦值总和是多少。

Standard Input

第一行有一个整数nn,表示花朵的总数目。

第二行包含nn个整数a\_ia_i,表示第ii朵花初始的愉悦值。

第三行包含一个整数mm,表示秋实大哥唱了mm天的歌。

接下来mm行,每行包含三个整数ll rr vv,表示秋实大哥对着[l, r][l,r]这个区间内的花朵歌唱,每朵花的愉悦值增加了vv

1\leq n,m,a\_i,|v|\leq 1000001nma_iv1000001\leq l\leq r\leq n。1lrn

Standard Output

输出共mm行,第ii行表示秋实大哥完成第ii天的歌唱后,那一段花朵的愉悦值总和。

Samples

3
0 0 0
3
1 2 1
1 2 -1
1 3 1




2
0
3

线段树 模板

#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;

const int maxn = 1e5 + 6;
int n;
int a[maxn];
int q;



struct node {
    int l, r;
    long long sum, lazy;
    void update(long long x)
    {
        sum += 1ll * (r - l + 1)*x;
        lazy += x;
    }
}tree[maxn*4];

void push_up(int x)
{
    tree[x].sum = tree[x << 1].sum + tree[x << 1 | 1].sum;
}



void push_down(int x)
{
    int lazyval = tree[x].lazy;
    if (lazyval) {
        tree[x << 1].update(lazyval);
        tree[x << 1 | 1].update(lazyval);
        tree[x].lazy = 0;
    }

}




void build(int x, int l, int r) {
    tree[x].l = l; tree[x].r = r;
    tree[x].sum = tree[x].lazy = 0;
    if (l == r) {
        tree[x].sum = a[l];
    }
    else {
        int mid = (l + r) / 2;
        build(x << 1, l, mid);
        build(x << 1 | 1, mid + 1, r);
        push_up(x);
    }
}

void update(int x, int l, int r, long long val)
{
    int L = tree[x].l, R = tree[x].r;
    if (l <= L && R <= r) {
        tree[x].update(val);
    }
    else {
        push_down(x);
        int mid = (L + R) / 2;
        if (mid >= l)  update(x << 1, l, r, val);
        if (r > mid) update(x << 1| 1, l, r, val);
        push_up(x);
    }
}


long long query(int x, int l, int r)
{
    int L = tree[x].l, R = tree[x].r;
    if (l <= L && R <= r) {
        return tree[x].sum;
    }
    else {
        push_down(x);
        long long ans = 0;
        int mid = (L + R) / 2;
        if (mid >= l)  ans += query(x << 1, l, r);
        if (r > mid)  ans += query(x << 1 | 1, l, r);
        push_up(x);
        return ans;
    }



}





int main()
{
    cin >> n;
    for (int i = 1; i <= n; i++) {
        cin >> a[i];
    }
    build(1, 1, n);
    cin >> q;

    for (int i = 1; i <= q; i++) {
        int l, r, val;
        cin >> l >> r >> val;
        update(1, l, r, val);
        cout << query(1, l, r) << endl;
    }

    return 0;
}

 

秋实大哥与花 线段树模板

标签:tor   val   llb   query   ons   pre   inpu   update   模板   

原文地址:https://www.cnblogs.com/itdef/p/11959822.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!