oj.uz

Submission #556065

# Submission time Handle Problem Language Result Execution time Memory
556065 2022-05-02T10:05:41 Z Skurrl Addk (eJOI21_addk) C++17
36 / 100
 2000 ms 2448 KB 
#include <bits/stdc++.h>
using namespace std;

#define pb push_back
#define N 100005
#define MOD 1000000007
#define endl '\n'
#define all(x) x.begin(), x.end()

typedef long long int ll;

const ll inf = 1e17;

void solve()
{
    int n, k; cin >> n >> k;

    vector<ll> arr(n + 1), pref(n + 1);

    for (int i = 1; i <= n; ++i)
    {
        cin >> arr[i];
        pref[i] = pref[i - 1] + arr[i];
    }

    int q; cin >> q;

    while(q--)
    {
        int type; cin >> type;
        if(type == 1) for(int i = 1; i <= k; ++i) { int smt; cin >> smt; }
        else
        {
            int l, r, m; cin >> l >> r >> m;
            ll res = 0;

            if(r - l + 2 >= 2 * m) 
                for(int i = l; i <= r - m + 1; ++i) 
                    res += pref[i + m - 1] - pref[i - 1];

            else
            {
                for (int i = l, cnt = 1; i < l + m; ++i, ++cnt) res += cnt * arr[i];
                for(int i = r, cnt = 1; i > r - m; --i, ++cnt) res += cnt * arr[i];

                res += m * (pref[r - m] - pref[l + m - 1]);
            }

            cout << res << endl;
        }
    }
}

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

    //ll t; cin >> t;
    //while(t--)
        solve();
}

/*
1 2 3 4 5 6 7 8 9, m = 4
1 2 3 4
2 3 4 5
3 4 5 6
4 5 6 7
5 6 7 8
6 7 8 9
*/

Subtask #1
36.0 / 36.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 212 KB Output is correct
2 Correct 2 ms 336 KB Output is correct
3 Correct 4 ms 380 KB Output is correct
4 Correct 8 ms 340 KB Output is correct
5 Correct 12 ms 340 KB Output is correct
6 Correct 16 ms 440 KB Output is correct
7 Correct 24 ms 488 KB Output is correct
8 Correct 44 ms 480 KB Output is correct
9 Correct 75 ms 632 KB Output is correct

Subtask #2
0 / 56.0

# Verdict Execution time Memory Grader output
1 Correct 223 ms 844 KB Output is correct
2 Correct 548 ms 1216 KB Output is correct
3 Correct 896 ms 1420 KB Output is correct
4 Execution timed out 2023 ms 2448 KB Time limit exceeded
5 Halted 0 ms 0 KB -

Subtask #3
0 / 8.0

# Verdict Execution time Memory Grader output
1 Incorrect 944 ms 1576 KB Output isn't correct
2 Halted 0 ms 0 KB -