oj.uz

Submission #531138

# Submission time Handle Problem Language Result Execution time Memory
531138 2022-02-27T19:56:56 Z fabijan_cikac Addk (eJOI21_addk) C++17
100 / 100
 81 ms 4104 KB 
#include <iostream>
#include <vector>>

using namespace std;

#define ll long long int
#define F first
#define S second
#define pll pair<ll, ll>

const int N = 1e5 + 10;

int n, k, q;
ll a[N]; pll fen[N];

void update(int x, ll val){
    a[x] += val; ++x; int y = x;
    while (x <= n){
        fen[x] = {fen[x].F + val, fen[x].S + y * val}; x += x & (-x);
    }
}

pll query(int x){
    ++x; pll val = {0, 0};
    while (x > 0){
        val = {val.F + fen[x].F, val.S + fen[x].S}; x -= x & (-x);
    }
    return val;
}

pll range_sum(int l, int r){
    if (l > r) return {0, 0};
    pll val1 = query(l - 1); pll val2 = query(r);
    return {val2.F - val1.F, val2.S - val1.S - l * (val2.F - val1.F)};
}

int main(){
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);

    cin >> n >> k;
    for (int i = 0; i < n; ++i){
        int x; cin >> x; update(i, x);
    }
    cin >> q;
    for (int i = 0; i < q; ++i){
        int u; cin >> u;
        if (u == 1){
            vector<int> v; v.resize(k);
            for (int i = 0; i < k; ++i){
                cin >> v[i]; --v[i];
            }
            ll cp = a[v[0]];
            for (int i = 0; i < k - 1; ++i)
                update(v[i], a[v[i + 1]] - a[v[i]]);
            update(v[k - 1], cp - a[v[k - 1]]);
        }
        else{
            int l, r, m; cin >> l >> r >> m; --l; --r;
            int siz = (r - l + 1);
            if (siz >= 2 * m){
                ll sum1 = range_sum(l, l + m - 1).S;
                ll sum2 = m * range_sum(l + m, r - m).F; pll val = range_sum(r - m + 1, r);
                ll sum3 = (m + 1) * val.F - val.S;
                cout << sum1 + sum2 + sum3;
            }
            else if (m < siz){
                if (siz == 2 * m - 1){
                    ll sum1 = range_sum(l, l + m - 2).S;
                    ll sum2 = m * range_sum(l + m - 1, r - m + 1).F; pll val = range_sum(r - m + 2, r);
                    ll sum3 = m * val.F - val.S;
                    cout << sum1 + sum2 + sum3;
                }
                else{
                    ll sum1 = range_sum(l, l + (siz - m)).S;
                    ll sum2 = (siz - m + 1) * range_sum(l + (siz - m) + 1, r - (siz - m + 1)).F; pll val = range_sum(r - (siz - m), r);
                    ll sum3 = (siz - m + 2) * val.F - val.S;
                    cout << sum1 + sum2 + sum3;
                }
            }
            else cout << range_sum(l, r).F;
            cout << '\n';
        }
    }

    return 0;
}

Compilation message

Main.cpp:2:18: warning: extra tokens at end of #include directive
    2 | #include <vector>>
      |                  ^

Subtask #1
36.0 / 36.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 204 KB Output is correct
2 Correct 1 ms 332 KB Output is correct
3 Correct 2 ms 332 KB Output is correct
4 Correct 2 ms 332 KB Output is correct
5 Correct 2 ms 460 KB Output is correct
6 Correct 3 ms 460 KB Output is correct
7 Correct 3 ms 460 KB Output is correct
8 Correct 4 ms 460 KB Output is correct
9 Correct 5 ms 588 KB Output is correct

Subtask #2
56.0 / 56.0

# Verdict Execution time Memory Grader output
1 Correct 11 ms 972 KB Output is correct
2 Correct 16 ms 1356 KB Output is correct
3 Correct 23 ms 1740 KB Output is correct
4 Correct 39 ms 2944 KB Output is correct
5 Correct 53 ms 4104 KB Output is correct
6 Correct 60 ms 3944 KB Output is correct
7 Correct 51 ms 3908 KB Output is correct

Subtask #3
8.0 / 8.0

# Verdict Execution time Memory Grader output
1 Correct 37 ms 1984 KB Output is correct
2 Correct 50 ms 3192 KB Output is correct
3 Correct 81 ms 3292 KB Output is correct
4 Correct 58 ms 4084 KB Output is correct