UESToj 1057 秋实大哥与花 ( 线段树

秋实大哥与花

Description

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

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

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

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

Input

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

第二行包含nn个整数aiai，表示第ii朵花初始的愉悦值。

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

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

1≤n，m，ai，|v|≤1000001≤n，m，ai，|v|≤100000，1≤l≤r≤n。

Output

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

Sample Input

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

Sample Out

2
0
3

题解：

线段树的区间求和板子题，b站上qsc学姐的算法讲堂上有 掏出祖传的线段树板子抄一抄QAQ

AC代码

#include <cstdio>
#include <algorithm>
using namespace std;
#define LL long long
const int N = 1000000*4+10;
int arr[N];
struct node {
    int left, right;
    LL sum, num;
}tree[N];
void build(int l,int r,int step)
{
    tree[step].left = l;
    tree[step].right = r;
    tree[step].num = 0;
    if(l == r) {
        tree[step].sum = arr[l];
        return ;    
    }
    int mid = (l+r) >> 1;
    build(l,mid,step<<1);
    build(mid+1,r,step<<1|1);
    tree[step].sum = tree[step<<1].sum + tree[step<<1|1].sum;
}
void update(int l,int r,LL value,int step)
{
    if(tree[step].left==l&&r==tree[step].right) {
        tree[step].num += value;
        return;
    }
    tree[step].sum += value*(r-l+1); 
    int mid = (tree[step].left+tree[step].right) >> 1;
    if(r <= mid) update(l,r,value,step<<1);
    else if(l > mid) update(l,r,value,step<<1|1);
    else {
        update(l,mid,value,step<<1);
        update(mid+1,r,value,step<<1|1);
    }
}

LL query(int l,int r,int step)
{
    if(l==tree[step].left && r==tree[step].right) {
        return  tree[step].sum+(r-l+1)*tree[step].num;
    }
    int mid = (tree[step].left+tree[step].right) >> 1;
    tree[step].sum += (tree[step].right-tree[step].left+1)*tree[step].num;
    update(tree[step].left,mid,tree[step].num,step<<1);
    update(mid+1,tree[step].right,tree[step].num,step<<1|1);
    tree[step].num = 0;
    if(r <= mid) return query(l,r,step<<1);
    else if(l > mid) return query(l,r,step<<1|1);
    else return query(l,mid,step<<1)+query(mid+1,r,step<<1|1); 
}

int main()
{
    int n, q, x;
    scanf("%d",&n);
    for(int i = 1;i <= n; i++) {
        scanf("%d",&arr[i]);
    }
    build(1,n,1);
    scanf("%d",&q); 
    while(q--) {
        char ch;
        int u, v, z; 
            scanf("%d%d%d",&u,&v,&z);
            update(u,v,z,1);
            printf("%lld\n",query(u,v,1));
    }
return 0;
}