結果

問題 No.1095 Smallest Kadomatsu Subsequence
ユーザー risujirohrisujiroh
提出日時 2020-06-26 21:45:20
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 238 ms / 2,000 ms
コード長 1,968 bytes
コンパイル時間 2,091 ms
コンパイル使用メモリ 205,044 KB
最終ジャッジ日時 2025-01-11 11:11:23
ジャッジサーバーID
(参考情報)
judge3 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff #

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

#ifdef LOCAL
#include "debug.h"
#else
#define DEBUG(...)
#endif

template <class T> struct segtree {
  const int n;
  vector<T> t;
  segtree(int _n = 0) : n(_n), t(2 * n) {}
  T& operator[](int i) { return t[n + i]; }
  void build() { for (int i = n; i--; ) t[i] = t[2 * i] * t[2 * i + 1]; }
  T fold(int l, int r) const {
    T a, b;
    for (l += n, r += n; l < r; l >>= 1, r >>= 1) {
      if (l & 1) a = a * t[l++];
      if (r & 1) b = t[--r] * b;
    }
    return a * b;
  }
  void set(int i, T a) {
    t[i += n] = a;
    while (i >>= 1) t[i] = t[2 * i] * t[2 * i + 1];
  }
};

template <class T> constexpr T inf = numeric_limits<T>::max() / 2.1;

template <class T> struct mini {
  T v = inf<T>;
  mini operator*(mini b) const {
    return b.v < v ? b : *this;
  }
};
template <class T> struct maxi {
  T v = -inf<T>;
  maxi operator*(maxi b) const {
    return v < b.v ? b : *this;
  }
};

int main() {
  cin.tie(nullptr);
  ios::sync_with_stdio(false);
  int n;
  cin >> n;
  vector<int> a(n);
  for (auto&& e : a) cin >> e;
  int res = inf<int>;
  {
    vector<int> ord(n);
    iota(begin(ord), end(ord), 0);
    sort(begin(ord), end(ord), [&](int i, int j) {
      return a[i] < a[j];
    });
    segtree<mini<int>> seg(n);
    for (int i : ord) {
      if (seg.fold(0, i).v != inf<int> and seg.fold(i + 1, n).v != inf<int>) {
        res = min(res, seg.fold(0, i).v + a[i] + seg.fold(i + 1, n).v);
      }
      seg.set(i, {a[i]});
    }
  }
  {
    vector<int> ord(n);
    iota(begin(ord), end(ord), 0);
    sort(begin(ord), end(ord), [&](int i, int j) {
      return a[i] > a[j];
    });
    segtree<mini<int>> seg(n);
    for (int i : ord) {
      if (seg.fold(0, i).v != inf<int> and seg.fold(i + 1, n).v != inf<int>) {
        res = min(res, seg.fold(0, i).v + a[i] + seg.fold(i + 1, n).v);
      }
      seg.set(i, {a[i]});
    }
  }
  if (res == inf<int>) {
    res = -1;
  }
  cout << res << '\n';
}
0