oj.uz

Submission #721099

# Submission time Handle Problem Language Result Execution time Memory
721099 2023-04-10T10:06:36 Z drdilyor Distributing Candies (IOI21_candies) C++17
29 / 100
 316 ms 34152 KB 
#include <bits/stdc++.h>
#ifdef ONPC
    #include "t_debug.cpp"
#else
    #define debug(...) 42
#endif
using namespace std;
//namespace pbds = __gnu_pbds;
using ll = long long;
const int inf = 1e9;
const ll infl = 1e18;
const int RANDOM = chrono::high_resolution_clock::now().time_since_epoch().count();
mt19937 rng(RANDOM);
template<typename T, typename U> istream& operator>>(istream& is, pair<T, U>& p) { return is >> p.first >> p.second; }
template<typename Cont> int sz(const Cont& cont) { return int(cont.size()); }
const string fileio = "";
constexpr int tests = 1, nmax = 2e5, nlog = __lg(nmax), mod = 1e9+7;
#define int ll

struct segtree {
    struct node{
        ll sum=0;
        ll mnp=0, mxp=0;
        static node single(int x) {
            return {x, min((int)0, x), max((int)0, x)};
        }
        node operator+(const node& rhs) {
            return node{
                sum + rhs.sum,
                min(mnp, sum + rhs.mnp),
                max(mxp, sum + rhs.mxp),
            };
        }
    };
    int n;
    vector<node> t;
    segtree(int m) : n(m), t(m*4) {}
    void update(int i, int x) {
        t[i += n] = node::single(x);
        for (i /= 2; i >= 1; i /= 2)
            t[i] = t[i*2] + t[i*2+1];
    }
    node query(int l, int r) {
        l += n;
        r += n;
        node nl, nr;
        while (l <= r) {
            if (l%2==1) nl = nl + t[l++];
            if (r%2==0) nr = t[r--] + nr;
            l /= 2;
            r /= 2;
        }
        return nl + nr;
    }
};

vector<int32_t> distribute_candies(vector<int32_t> c, vector<int32_t> l, vector<int32_t> r, vector<int32_t> v) {
    int n = sz(c), q = sz(l);
 
    vector<int> res(n), real(n + 1);
    for (int i = 0; i < q; i++) {
        real[l[i]] += v[i];
        real[r[i]+1] -= v[i];
    }

    for (int i = 1; i < n; i++)
        real[i] += real[i-1];

    reverse(l.begin(), l.end());
    reverse(r.begin(), r.end());
    reverse(v.begin(), v.end());

 
    segtree st(q);
    for (int j = 0; j < q; j++)
        st.update(j, -v[j]);
    for (int i = 0; i < n; i++) {
        int cur = 0, low = 0, high = 0;
        int lb = -1, rb = q-1;
        auto check = [&](int j) {
            auto node = st.query(0, j);
            cur = node.sum;
            low = node.mnp;
            high = node.mxp;
            if (high - low >= c[i]) {
                if (cur == high)
                    res[i] = -low;
                else
                    res[i] = c[i] - high;
                return true;
            }
            return false;
        };
        while (lb < rb-1) {
            int mid = (lb + rb) / 2;
            if (check(mid))
                rb = mid;
            else lb = mid;
        }
        if (!check(rb)) {
            res[i] = -low;
        }
    }
    return vector<int32_t>(res.begin(), res.end());
}
/*
     █████               █████  ███  ████                               
    ▒▒███               ▒▒███  ▒▒▒  ▒▒███                               
  ███████  ████████   ███████  ████  ▒███  █████ ████  ██████  ████████ 
 ███▒▒███ ▒▒███▒▒███ ███▒▒███ ▒▒███  ▒███ ▒▒███ ▒███  ███▒▒███▒▒███▒▒███
▒███ ▒███  ▒███ ▒▒▒ ▒███ ▒███  ▒███  ▒███  ▒███ ▒███ ▒███ ▒███ ▒███ ▒▒▒ 
▒███ ▒███  ▒███     ▒███ ▒███  ▒███  ▒███  ▒███ ▒███ ▒███ ▒███ ▒███     
▒▒████████ █████    ▒▒████████ █████ █████ ▒▒███████ ▒▒██████  █████    
 ▒▒▒▒▒▒▒▒ ▒▒▒▒▒      ▒▒▒▒▒▒▒▒ ▒▒▒▒▒ ▒▒▒▒▒   ▒▒▒▒▒███  ▒▒▒▒▒▒  ▒▒▒▒▒     
                                            ███ ▒███                    
                                           ▒▒██████                     
                                            ▒▒▒▒▒▒
*/

Subtask #1
0 / 3.0

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

Subtask #2
0 / 8.0

# Verdict Execution time Memory Grader output
1 Incorrect 314 ms 29268 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 212 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #4
29.0 / 29.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 212 KB Output is correct
2 Correct 1 ms 212 KB Output is correct
3 Correct 71 ms 23784 KB Output is correct
4 Correct 117 ms 5964 KB Output is correct
5 Correct 302 ms 29268 KB Output is correct
6 Correct 284 ms 33448 KB Output is correct
7 Correct 245 ms 34152 KB Output is correct
8 Correct 312 ms 32668 KB Output is correct
9 Correct 316 ms 34148 KB Output is correct

Subtask #5
0 / 33.0

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