Submission #435823

# Submission time Handle Problem Language Result Execution time Memory
435823 2021-06-23T18:51:03 Z bicsi Distributing Candies (IOI21_candies) C++17
100 / 100
3841 ms 490212 KB
#include "candies.h"
#include <bits/stdc++.h>
 
using namespace std;
using ll = long long;
const int BS = 333;
 
 
vector<ll> compute_function(const vector<int>& ops) {
  vector<ll> stk;
  stk.push_back(4e18);
  stk.push_back(2e18);
  for (auto delta : ops) {
    if (delta == 0) continue;
    if (delta > 0) {
      ll rem = delta, px = 0;
      while (true) {
        ll x = stk.back();
        if (x - px > rem) {
          stk.push_back(px + rem);
          stk.push_back(0);
          break;
        } 
        rem -= (x - px);
        stk.pop_back();
        px = stk.back();
        stk.pop_back();
      }
    } else {
      ll rem = -delta;
      while (true) {
        ll px = stk.back(); stk.pop_back();
        ll x = stk.back();
        if (x - px > rem) {
          stk.push_back(px + rem);
          break;
        }
        rem -= (x - px);
        stk.pop_back();
      }
    }
  }
  
  stk.push_back(0);
  stk.push_back(0);
  reverse(stk.begin(), stk.end());
  return stk;
}
 
vector<int> distribute_candies(
    vector<int> c, vector<int> l,
    vector<int> r, vector<int> v) {
  int n = c.size();
 
  
  // Sort each bucket increasingly by c[i] to facilitate easy evaluation of f.
  vector<int> order(n);
  iota(order.begin(), order.end(), 0);
  for (int i = 0; i < n; i += BS) {
    int j = min(i + BS, n);
    sort(order.begin() + i, order.begin() + j, [&](int a, int b) {
      return c[a] < c[b];
    });
  }
 
  vector<ll> stk;
  vector<int> a(n, 0), na(n, 0);
  vector<vector<int>> ops(n / BS + 1);
 
  auto commit = [&](int bi) {
    if (ops[bi].empty()) return;
    int l = bi * BS, r = min(n, (bi + 1) * BS);
 
    // compute f(.) and evaluate it in all c[i] of this bucket
    auto stk = compute_function(ops[bi]);
    ll value = 0;
    int ptr = 0;
    for (int i = l; i < r; ++i) {
      int ci = c[order[i]];
      while (stk[ptr + 1] < ci) {
        if (ptr % 2 == 0)
          value += stk[ptr + 1] - stk[ptr];
        ++ptr;
      }
      if (ptr % 2 == 0) {
        na[order[i]] = value + (ci - stk[ptr]);
      } else {
        na[order[i]] = value;
      }
    }
  
    // Adapt for non-zero initial values.
    ll mind = 0, maxd = 0, nowd = 0;
    for (auto x : ops[bi]) {
      nowd += x;
      mind = min(mind, nowd);
      maxd = max(maxd, nowd);
    }
    for (int i = l; i < r; ++i) {
      ll hi = min(c[i] - maxd, (ll)a[i]);
      ll lo = -mind;
      if (lo < hi) 
        na[i] += hi - lo; 
    }
 
    // Housekeeping.
    for (int i = l; i < r; ++i) 
      a[i] = na[i];
    ops[bi].clear();
  };
 
  int q = l.size();
  for (int i = 0; i < q; ++i) {
    int bl = l[i] / BS, br = r[i] / BS;
    commit(bl); commit(br); 
    for (int j = l[i]; j <= r[i]; ) {
      if (j % BS == 0 && (j + BS - 1) <= r[i]) {
        ops[j / BS].push_back(v[i]);
        j += BS;
      } else {
        a[j] = max(0, min(a[j] + v[i], c[j]));
        j += 1;
      }
    }
  }
 
  for (int i = 0; i < n; i += BS)
    commit(i / BS);
  return a;
}
# Verdict Execution time Memory Grader output
1 Correct 1 ms 256 KB Output is correct
2 Correct 1 ms 204 KB Output is correct
3 Correct 1 ms 332 KB Output is correct
4 Correct 2 ms 332 KB Output is correct
5 Correct 9 ms 372 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 2280 ms 11644 KB Output is correct
2 Correct 2557 ms 11732 KB Output is correct
3 Correct 2475 ms 11660 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 1 ms 204 KB Output is correct
2 Correct 559 ms 5036 KB Output is correct
3 Correct 106 ms 5176 KB Output is correct
4 Correct 2020 ms 11700 KB Output is correct
5 Correct 2007 ms 11780 KB Output is correct
6 Correct 2088 ms 11640 KB Output is correct
7 Correct 1904 ms 11648 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 1 ms 204 KB Output is correct
2 Correct 1 ms 204 KB Output is correct
3 Correct 613 ms 9340 KB Output is correct
4 Correct 109 ms 9136 KB Output is correct
5 Correct 3841 ms 490196 KB Output is correct
6 Correct 3731 ms 490212 KB Output is correct
7 Correct 3762 ms 490172 KB Output is correct
8 Correct 3759 ms 490164 KB Output is correct
9 Correct 3199 ms 490140 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 1 ms 256 KB Output is correct
2 Correct 1 ms 204 KB Output is correct
3 Correct 1 ms 332 KB Output is correct
4 Correct 2 ms 332 KB Output is correct
5 Correct 9 ms 372 KB Output is correct
6 Correct 2280 ms 11644 KB Output is correct
7 Correct 2557 ms 11732 KB Output is correct
8 Correct 2475 ms 11660 KB Output is correct
9 Correct 1 ms 204 KB Output is correct
10 Correct 559 ms 5036 KB Output is correct
11 Correct 106 ms 5176 KB Output is correct
12 Correct 2020 ms 11700 KB Output is correct
13 Correct 2007 ms 11780 KB Output is correct
14 Correct 2088 ms 11640 KB Output is correct
15 Correct 1904 ms 11648 KB Output is correct
16 Correct 1 ms 204 KB Output is correct
17 Correct 1 ms 204 KB Output is correct
18 Correct 613 ms 9340 KB Output is correct
19 Correct 109 ms 9136 KB Output is correct
20 Correct 3841 ms 490196 KB Output is correct
21 Correct 3731 ms 490212 KB Output is correct
22 Correct 3762 ms 490172 KB Output is correct
23 Correct 3759 ms 490164 KB Output is correct
24 Correct 3199 ms 490140 KB Output is correct
25 Correct 1 ms 204 KB Output is correct
26 Correct 96 ms 5168 KB Output is correct
27 Correct 640 ms 5032 KB Output is correct
28 Correct 2734 ms 11608 KB Output is correct
29 Correct 2732 ms 11720 KB Output is correct
30 Correct 2617 ms 11712 KB Output is correct
31 Correct 2319 ms 11616 KB Output is correct
32 Correct 2195 ms 11628 KB Output is correct