結果

問題 No.705 ゴミ拾い Hard
ユーザー 0w1
提出日時 2018-10-19 00:46:05
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 168 ms / 1,500 ms
コード長 2,080 bytes
コンパイル時間 1,998 ms
コンパイル使用メモリ 207,656 KB
最終ジャッジ日時 2025-01-06 14:34:30
ジャッジサーバーID
(参考情報)
judge2 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 40
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
using namespace std;

template<typename R> // return_type
struct MongeDP {
  vector<R> dp;
  vector<int> pre;
  function<bool(R, R)> cmp; // true is left better
  function<R(const vector<R> &dp, int, int)> get_cost;
  MongeDP(int n, function<bool(R, R)> c, function<R(const vector<R> &dp, int, int)> gc)
      : dp(n + 1), pre(n + 1, -1), cmp(c), get_cost(gc) {
    deque<tuple<int, int, int>> dcs; // decision
    dcs.emplace_back(0, 1, n); // transition from dp[0] is effective for [1, N]
    for (int i = 1; i <= n; ++i) {
      while (get<2>(dcs.front()) < i) dcs.pop_front(); // right bound is out-dated
      dp[i] = get_cost(dp, pre[i] = get<0>(dcs.front()), i); // best t is A[dcs.top(), i)
      while (dcs.size()) {
        int x, lb, rb;
        tie(x, lb, rb) = dcs.back();
        if (lb <= i) break; // will be pop_fronted soon anyway
        if (!cmp(get_cost(dp, x, lb), get_cost(dp, i, lb))) {
          dcs.pop_back();
          if (dcs.size()) get<2>(dcs.back()) = n;
        } else break;
      }
      int best = -1;
      for (int lb = i + 1, rb = n, x = get<0>(dcs.back()); lb <= rb; ) {
        int mb = lb + rb >> 1;
        if (cmp(get_cost(dp, i, mb), get_cost(dp, x, mb))) {
          best = mb;
          rb = mb - 1;
        } else lb = mb + 1;
      }
      if (~best) {
        get<2>(dcs.back()) = best - 1;
        dcs.emplace_back(i, best, n);
      }
    }
  }
  R operator[](int x) { return dp[x]; }
};

signed main() {
  ios::sync_with_stdio(false);
  
  int N;
  cin >> N;

  vector<int> A(N);
  for (int i = 0; i < N; ++i) cin >> A[i];

  vector<int> X(N);
  for (int i = 0; i < N; ++i) cin >> X[i];

  vector<int> Y(N);
  for (int i = 0; i < N; ++i) cin >> Y[i];

  auto abscub = [](int64_t x) { return abs(x * x * x); };

  MongeDP<int64_t> mdp(N, [](int64_t x, int64_t y) { return x < y; },
                       [&](const vector<int64_t> &dp, int x, int rb) {
                         return dp[x] + abscub(A[rb - 1] - X[x]) + abscub(Y[x]);
                       });

  cout << mdp[N] << endl;

  return 0;
}
0