oj.uz

Submission #1062656

# Submission time Handle Problem Language Result Execution time Memory
1062656 2024-08-17T09:36:29 Z j_vdd16 Distributing Candies (IOI21_candies) C++17
0 / 100
 90 ms 17880 KB 
#include "candies.h"

#include <algorithm>
#include <bitset>
#include <cstdint>
#include <cstring>
#include <iostream>
#include <limits.h>
#include <math.h>
#include <map>
#include <numeric>
#include <queue>
#include <set>
#include <stack>
#include <string>
#include <vector>

#define ll long long
#define loop(X, N) for(int X = 0; X < (N); X++)
#define all(V) V.begin(), V.end()
#define rall(V) V.rbegin(), V.rend()

using namespace std;

typedef vector<int> vi;
typedef vector<vi> vvi;
typedef pair<int, int> ii;
typedef vector<ii> vii;
typedef vector<vector<ii>> vvii;
typedef vector<bool> vb;
typedef vector<vector<bool>> vvb;


// struct SegTree {
//     int n, N;
//     vector<set<int>> tree;

//     SegTree(int _n) {
//         int n = _n;
//         int N = 1;
//         while (N < n) N *= 2;

//         tree.resize(2 * N);

//     }
// };

vi distribute_candies(vi c, vi ls, vi rs, vi vs) {
    int n = c.size();
    int q = ls.size();

    vector<ll> cumul(q, vs[0]);
    vector<pair<ll, int>> sufMax(q), sufMin(q);

    for (int i = 1; i < q; i++) {
        ll v = vs[i];

        cumul[i] = cumul[i - 1] + v;
        cumul[i] = max((ll)0, cumul[i]);
    }

    sufMax.back() = {cumul.back(), q - 1};
    sufMin.back() = {cumul.back(), q - 1};
    for (int i = q - 2; i >= 0; i--) {
        sufMin[i] = sufMin[i + 1];
        sufMax[i] = sufMax[i + 1];
        if (cumul[i] < sufMin[i + 1].first) {
            sufMin[i].first = cumul[i];
            sufMin[i].second = i;
        }
        if (cumul[i] > sufMax[i + 1].first) {
            sufMax[i].first = cumul[i];
            sufMax[i].second = i;
        }
    }

    vi s(n);
    loop(i, n) {
        int l = 0, r = q - 1;
        while (l < r) {
            int m = (l + r + 1) / 2;

            ll diff = sufMax[m].first - sufMin[m].first;
            if (diff < c[i]) {
                r = m - 1;
            }
            else { 
                l = m;
            }
        }

        int lastMin = sufMin[l].second;
        int lastMax = sufMax[l].second;
        if (sufMax[0].first - sufMin[0].first <= c[i] && l == 0) {
            s[i] = cumul.back();
            continue;
        }

        if (lastMin >= lastMax) {
            l = lastMin;
            s[i] = 0;
        }
        else {
            l = lastMax;
            s[i] = c[i];
        }

        s[i] += cumul.back() - cumul[l];
    }

    return s;
}

Subtask #1
0 / 3.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 348 KB Output is correct
2 Incorrect 1 ms 432 KB Output isn't correct
3 Halted 0 ms 0 KB -

Subtask #2
0 / 8.0

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

Subtask #3
0 / 27.0

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

Subtask #4
0 / 29.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 348 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Incorrect 37 ms 14304 KB Output isn't correct
4 Halted 0 ms 0 KB -

Subtask #5
0 / 33.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 348 KB Output is correct
2 Incorrect 1 ms 432 KB Output isn't correct
3 Halted 0 ms 0 KB -