Submission #668424

#	Time	Username	Problem	Language	Result	Execution time	Memory
668424		600Mihnea	Safety (NOI18_safety)	C++17	100 / 100	207 ms	21108 KiB

safety

#include <bits/stdc++.h>


using namespace std;

#define int long long


signed main()
{
#ifdef ONPC
  freopen ("input.txt", "r", stdin);
#endif // ONPC

#ifndef ONPC
  ios::sync_with_stdio(0); cin.tie(0); cout.tie(0);
#endif // ONPC

  int n, dmax, sol = 0, lazy = 0;
  cin >> n >> dmax;

  multiset<int> a, b;

  for (int i = 0; i < n; i++)
  {
    int x;
    cin >> x;
    if (i == 0)
    {
      a.insert(x);
      b.insert(x);
      continue;
    }
    lazy += dmax;

    auto ita = a.end();
    ita--;

    if (x <= *ita - lazy)
    {
      sol += *ita - lazy - x;
      b.insert(*ita - lazy - lazy);
      a.erase(ita);
      a.insert(x + lazy);
      a.insert(x + lazy);
    }
    else
    {
      auto itb = b.begin();
      if (x >= *itb + lazy)
      {
        sol += x - (*itb + lazy);
        a.insert(*itb + lazy + lazy);
        b.erase(itb);
        b.insert(x - lazy);
        b.insert(x - lazy);
      }
      else
      {
        a.insert(x + lazy);
        b.insert(x - lazy);
      }
    }
  }
  cout << sol << "\n";


  return 0;
}

• 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