Submission #470774

#	Time	Username	Problem	Language	Result	Execution time	Memory
470774		nobik	Safety (NOI18_safety)	C++14	100 / 100	279 ms	21680 KiB

safety

// https://oj.uz/problem/view/NOI18_safety
#include <bits/stdc++.h>
// #include <atcoder/all>

using namespace std;
// using namespace atcoder;

#define REP(i, n) for (int i = 0; i < (n); ++i)
#define SZ(a) ((int)(a).size())
#define ALL(a) (a).begin(), (a).end()

using ll = long long;
using vi = vector<int>;
using vvi = vector<vi>;
// using mint = modint998244353;

constexpr int kInf = 1000 * 1000 * 1000 + 7;
constexpr long long kInfLL = 1e18;

struct TaggedSet {
  multiset<long long> s;
  long long add{};

  void Insert(long long x) { s.insert(x - add); }
  void Erase(long long x) { s.erase(s.find(x - add)); }
  long long Min() { return *s.begin() + add; }
  long long Max() { return *s.rbegin() + add; }
  void Add(long long x) { add += x; }
  bool Empty() { return s.empty(); }
};

int main() {
  ios::sync_with_stdio(false);
  cin.tie(nullptr);

  int n, h;
  cin >> n >> h;

  vi a(n);
  REP(i, n) cin >> a[i];

  TaggedSet neg, pos;
  neg.Insert(a[0]);
  pos.Insert(a[0]);

  long long result{};
  for (int i = 1; i < n; ++i) {
    neg.Add(-h); pos.Add(h);
    while (!neg.Empty() && neg.Min() < 0) neg.Erase(neg.Min());
    if (neg.Empty()) neg.Insert(0);
    if (pos.Empty()) pos.Insert(0);
    long long mn = pos.Min();
    int l = a[i], r = a[i];
    if (l <= mn) {
      neg.Insert(l);
    } else {
      pos.Insert(l);
      pos.Erase(mn);
      neg.Insert(mn);
      result += l - mn;
    }
    long long mx = neg.Max();
    if (mx <= r) {
      pos.Insert(r);
    } else {
      neg.Insert(r);
      neg.Erase(mx);
      pos.Insert(mx);
      result += mx - r;
    }
  }

  cout << result << '\n';
}

• Subtask 1: 3 / 3
• Subtask 2: 4 / 4
• Subtask 3: 9 / 9
• Subtask 4: 5 / 5
• Subtask 5: 6 / 6
• Subtask 6: 11 / 11
• Subtask 7: 11 / 11
• Subtask 8: 22 / 22
• Subtask 9: 29 / 29