Sums

#include <iostream>
 
using namespace std;
typedef long long ll;
 
const int N = 1<<19;
 
int n, q;
ll st[2*N];
ll lzset[2*N];
bool hasset[2*N];
ll lzadd[2*N];
bool hasadd[2*N];
 
void push(int s, int l, int r) {
    if (hasset[s]) st[s] = lzset[s]*(r-l+1);
    if (hasadd[s]) st[s] += lzadd[s]*(r-l+1);
    
    if (s < N) {
        if (hasset[s]) {
            lzset[2*s] = lzset[s];
            lzset[2*s+1] = lzset[s];
            hasset[2*s] = true;
            hasset[2*s+1] = true;
            lzadd[2*s] = 0;
            lzadd[2*s+1] = 0;
            hasadd[2*s] = false;
            hasadd[2*s+1] = false;
        }
        if (hasadd[s]) {
            lzadd[2*s] += lzadd[s];
            lzadd[2*s+1] += lzadd[s];
            hasadd[2*s] = true;
            hasadd[2*s+1] = true;
        }
    }
    lzset[s] = 0;
    hasset[s] = false;
    lzadd[s] = 0;
    hasadd[s] = false;
}
 
ll summa(int ql, int qr, int l = 1, int r = N, int s = 1) {
    push(s, l, r);
    if (ql <= l && r <= qr) return st[s];
    if (l > qr || r < ql) return 0;
    int mid = (l+r)/2;
    return summa(ql, qr, l, mid, 2*s) + summa(ql, qr, mid+1, r, 2*s+1); 
}
 
void aseta(int ql, int qr, ll x, int l = 1, int r = N, int s = 1) {
    push(s, l, r);
    if (ql <= l && r <= qr) {
        lzset[s] = x;
        hasset[s] = true;
        lzadd[s] = 0;
        hasadd[s] = false;
        push(s, l, r);
        return;
    }
    if (l > qr || r < ql) return;
    int mid = (l+r)/2;
    aseta(ql, qr, x, l, mid, 2*s);
    aseta(ql, qr, x, mid+1, r, 2*s+1);
    st[s] = st[2*s] + st[2*s+1];
}
 
void lisaa(int ql, int qr, ll x, int l = 1, int r = N, int s = 1) {
    push(s, l, r);
    if (ql <= l && r <= qr) {
        lzadd[s] += x;
        hasadd[s] = true;
        push(s, l, r);
        return;
    }
    if (l > qr || r < ql) return;
    int mid = (l+r)/2;
    lisaa(ql, qr, x, l, mid, 2*s);
    lisaa(ql, qr, x, mid+1, r, 2*s+1);
    st[s] = st[2*s] + st[2*s+1];
}
 
int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(0);
    cin >> n;
    for (int i = 1; i <= n; ++i) {
        ll cn;
        cin >> cn;
        aseta(i, i, cn);
    }
    
    cin >> q;
    
    for (int i = 0; i < q; ++i) {
        int tp;
        cin >> tp;
        if (tp == 1) {
            int a, b, x;
            cin >> a >> b >> x;
            lisaa(a, b, x);
        }
        if (tp == 2) {
            int a, b, x;
            cin >> a >> b >> x;
            aseta(a, b, x);
        }
        if (tp == 3) {
            int a, b;
            cin >> a >> b;
            cout << summa(a, b) << "\n";
        }
    }
    return 0;
}

Test 1

Verdict: ACCEPTED

Test 2

Verdict: ACCEPTED

Test 3

Verdict: ACCEPTED

Test 4

Verdict: ACCEPTED

Test 5

Verdict: ACCEPTED

Test 6

Verdict: ACCEPTED

Test 7

Verdict: ACCEPTED

Test 8

Verdict: ACCEPTED

Test 9

Verdict: ACCEPTED

Test 10

Verdict: ACCEPTED