Submission #531138

#	Time	Username	Problem	Language	Result	Execution time	Memory
531138		fabijan_cikac	Addk (eJOI21_addk)	C++17	100 / 100	81 ms	4104 KiB

Main

#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 (stderr)

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

• Subtask 1: 36 / 36
• Subtask 2: 56 / 56
• Subtask 3: 8 / 8