Submission #1329028

#TimeUsernameProblemLanguageResultExecution timeMemory
1329028kawhietAddk (eJOI21_addk)C++20
100 / 100
216 ms12076 KiB
#include <bits/stdc++.h>
using namespace std;

#define int long long

struct SegmentTree {
    int n;
    vector<int> t;

    SegmentTree(int _n) {
        n = _n;
        t.resize(4 * n);
    }

    int merge(int x, int y) {
        return x + y;
    }

    void update(int id, int tl, int tr, int i, int v) {
        if (tl == tr) {
            t[id] = v;
            return;
        }
        int x = (id << 1) + 1, y = x + 1, tm = (tl + tr) >> 1;
        if (i <= tm) {
            update(x, tl, tm, i, v);
        } else {
            update(y, tm + 1, tr, i, v);
        }
        t[id] = merge(t[x], t[y]);
    }

    int get(int id, int tl, int tr, int l, int r) {
        if (r < tl || tr < l) return 0LL;
        if (l <= tl && tr <= r) return t[id];
        int x = (id << 1) + 1, y = x + 1, tm = (tl + tr) >> 1;
        return merge(get(x, tl, tm, l, r), get(y, tm + 1, tr, l, r));
    }

    void update(int i, int v) { update(0, 0, n - 1, i, v); }
    int get(int l, int r) { return get(0, 0, n - 1, l, r); }
};

signed main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    int n, k;
    cin >> n >> k;
    vector<int> a(n + 1);
    for (int i = 1; i <= n; i++) {
        cin >> a[i];
    }
    SegmentTree p(n + 1), s(n + 1), sum(n + 1);
    for (int i = 1; i <= n; i++) {
        sum.update(i, a[i]);
        p.update(i, a[i] * i);
        s.update(i, a[i] * (n - i + 1));
    }
    int q;
    cin >> q;
    while (q--) {
        int t;
        cin >> t;
        if (t == 1) {
            vector<int> pos(k);
            for (int i = 0; i < k; i++) {
                cin >> pos[i];
            }
            for (int i = 1; i < k; i++) {
                swap(a[pos[i]], a[pos[i - 1]]);
            }
            for (auto i : pos) {
                p.update(i, a[i] * i);
                s.update(i, a[i] * (n - i + 1));
                sum.update(i, a[i]);
            }
        } else {
            int l, r, m;
            cin >> l >> r >> m;
            int x = min(r - l - m + 2, m);
            int ans = 0;
            ans += p.get(l, l + x - 2) - sum.get(l, l + x - 2) * (l - 1);
            ans += s.get(r - x + 2, r) - sum.get(r - x + 2, r) * (n - r);
            ans += sum.get(l + x - 1, r - x + 1) * x;
            cout << ans << '\n';
        }
    }
    return 0;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...