Submission #146910

#	Time	Username	Problem	Language	Result	Execution time	Memory
146910		bortoz	Potatoes and fertilizers (LMIO19_bulves)	C++17	0 / 100	3 ms	376 KiB

bulves

#include "bits/stdc++.h"
using namespace std;
typedef long long ll;
#define fi first
#define se second

int main() {
  ios::sync_with_stdio(false);
  int N;
  cin >> N;
  vector<int> V(N);
  set<int> L;
  for (int i = 0; i < N; i++) {
    int a, b;
    cin >> a >> b;
    V[i] = a - b;
    if (V[i] > 0) {
      L.insert(i);
    }
  }
  ll res = 0;
  for (int i = 0; i < N; i++) {
    while (V[i] < 0) {
      auto itR = L.lower_bound(i);
      if (itR == L.end() && itR != L.begin()) {
        itR--;
      }
      auto itL = itR;
      if (itL != L.begin()) {
        itL--;
      }
      if (abs(i - *itL) < abs(i - *itR)) {
        int m = min(-V[i], V[*itL]);
        res += 1ll * m * abs(i - *itL);
        V[i] += m;
        V[*itL] -= m;
        if (V[*itL] == 0) {
          L.erase(itL);
        }
      } else {
        int m = min(-V[i], V[*itR]);
        res += 1ll * m * abs(i - *itR);
        V[i] += m;
        V[*itR] -= m;
        if (V[*itR] == 0) {
          L.erase(itR);
        }
      }
    }
  }
  cout << res << endl;

  return 0;
}

• Subtask 1: 0 / 24
• Subtask 2: 0 / 10
• Subtask 3: 0 / 20
• Subtask 4: 0 / 10
• Subtask 5: 0 / 36