线段树 1

题目描述

如题,已知一个数列,你需要进行下面两种操作:

将某区间每一个数加上 k k k
求出某区间每一个数的和。

输入格式

第一行包含两个整数 n , m n, m n,m,分别表示该数列数字的个数和操作的总个数。

第二行包含 n n n 个用空格分隔的整数,其中第 i i i 个数字表示数列第 i i i 项的初始值。

接下来 m m m 行每行包含 3 3 3 4 4 4 个整数,表示一个操作,具体如下:

1 x y k 1 \quad x\quad y\quad k 1xyk:将区间 [ x , y ] [x, y] [x,y] 内每个数加上 k k k
2 x y 2\quad x\quad y 2xy:输出区间 [ x , y ] [x, y] [x,y] 内每个数的和。

输出格式

输出包含若干行整数,即为所有操作 2 2 2的结果。

输入输出样例

输入

5 5
1 5 4 2 3
2 2 4
1 2 3 2
2 3 4
1 1 5 1
2 1 4

输出

11
8
20

说明/提示

对于 100 % 100\% 100% 的数据: 1 ≤ n , m ≤ 10 5 1 \le n, m \le {10}^5 1n,m105

保证任意时刻数列中任意元素的和在 [ − 2 63 , 2 63 ) [-2^{63}, 2^{63}) [263,263)内。

/***  Amber  ***/
//#pragma GCC optimize(3,"Ofast","inline")
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
using namespace std;
#define ls (rt<<1)
#define rs (rt<<1|1)
typedef long long ll;
typedef pair<int,int> pii;
template <typename T>
inline void read(T &x) {
     
    x = 0;
    static int p;
    p = 1;
    static char c;
    c = getchar();
    while (!isdigit(c)) {
     
        if (c == '-')p = -1;
        c = getchar();
    }
    while (isdigit(c)) {
     
        x = (x << 1) + (x << 3) + (c - 48);
        c = getchar();
    }
    x *= p;
}
template <typename T>
inline void print(T x) {
     
    if (x<0) {
     
        x = -x;
        putchar('-');
    }
    static int cnt;
    static int a[50];
    cnt = 0;
    do {
     
        a[++cnt] = x % 10;
        x /= 10;
    } while (x);
    for (int i = cnt; i >= 1; i--)putchar(a[i] + '0');
    puts("");
}
const double Pi=acos(-1);
const double eps=1e-6;
const int mod = 1e9+7;
const int inf = 0x3f3f3f3f;
const ll Inf = 0x3f3f3f3f3f3f3f3f;
const int maxn = 100010;

int n,m;
ll sum[maxn<<2];
ll lazy[maxn<<2];
ll a[maxn];
int l1[maxn<<2],r1[maxn<<2];
inline void pushup(int rt){
     
        sum[rt] = sum[ls] + sum[rs];
}
void build(int rt,int l,int r) {
     
    l1[rt] = l;
    r1[rt] = r;
    if (l == r) {
     
        sum[rt] = a[l];
        return;
    }
    int mid = l + r >> 1;
    build(ls, l, mid);
    build(rs, mid + 1, r);
    pushup(rt);
}
inline void pushdown(int rt) {
     
    if (lazy[rt]) {
     
        sum[ls] += 1ll * (r1[ls] - l1[ls] + 1) * lazy[rt];
        sum[rs] += 1ll * (r1[rs] - l1[rs] + 1) * lazy[rt];
        lazy[ls] += lazy[rt];
        lazy[rs] += lazy[rt];
        lazy[rt] = 0;
    }
}
void change(int rt,int l,int r,int ql,int qr,int k) {
     
    if (ql <= l && r <= qr) {
     
        sum[rt] += 1ll * (r - l + 1) * k;
        lazy[rt] += k;
        return;
    }
    pushdown(rt);
    int mid = l + r >> 1;
    if (ql <= mid) change(ls, l, mid, ql, qr, k);
    if (qr > mid) change(rs, mid + 1, r, ql, qr, k);
    pushup(rt);
}
ll query(int rt,int l,int r,int ql,int qr) {
     
    if (ql <= l && r <= qr) {
     
        return sum[rt];
    }
    pushdown(rt);
    ll res = 0;
    int mid = l + r >> 1;
    if (ql <= mid) res += query(ls, l, mid, ql, qr);
    if (qr > mid) res += query(rs, mid + 1, r, ql, qr);
    pushup(rt);
    return res;
}
inline void work() {
     
    scanf("%d%d", &n, &m);
    for (int i = 1; i <= n; i++) {
     
        scanf("%lld", &a[i]);
    }
    build(1, 1, n);
    for (int i = 1; i <= m; i++) {
     
        int op, l, r, k;
        scanf("%d", &op);
        if (op == 1) {
     
            int l, r, k;
            scanf("%d%d%d", &l, &r, &k);
            change(1, 1, n, l, r, k);
        } else {
     
            scanf("%d%d", &l, &r);
            printf("%lld\n", query(1, 1, n, l, r));
        }
    }
}
int main() {
     
    //freopen("1.txt","r",stdin);
    int T = 1;
    //read(T);
    while (T--) {
     
        work();
    }
    return 0;
}

你可能感兴趣的:(模板,acm竞赛)