結果
問題 | No.2458 Line Up Charged Balls |
ユーザー |
![]() |
提出日時 | 2023-08-10 10:57:12 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 137 ms / 2,000 ms |
コード長 | 2,646 bytes |
コンパイル時間 | 2,265 ms |
コンパイル使用メモリ | 199,300 KB |
最終ジャッジ日時 | 2025-02-16 00:29:54 |
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 25 |
ソースコード
#include <bits/stdc++.h> using namespace std; using ll = long long; template<typename T> class CHT { private: struct node { node *left, *right; static const T inf = numeric_limits<T>::max(); T a, b; node() : node(0, inf){} node(const T _a, const T _b) : left(nullptr), right(nullptr), a(_a), b(_b){} T f(const T x) const { return a * x + b; } }; static void swap(node *x, node *y){ std::swap(x->a, y->a), std::swap(x->b, y->b); } void _add_line(node *cur, node *nw, T l, T r){ while(true){ if(nw->f(l) < cur->f(l)) swap(cur, nw); if(cur->f(r - 1) <= nw->f(r - 1)) break; const T mid = (l + r) / 2; if(cur->f(mid) <= nw->f(mid)){ if(!cur->right){ cur->right = new node(*nw); break; }else{ cur = cur->right, l = mid; } }else{ swap(cur, nw); if(!cur->left){ cur->left = new node(*nw); break; }else{ cur = cur->left, r = mid; } } } } T query(node *cur, const T k, T l, T r) const { T ans = numeric_limits<T>::max(); while(cur){ ans = min(ans, cur->f(k)); const T mid = (l + r) / 2; if(k < mid){ cur = cur->left, r = mid; }else{ cur = cur->right, l = mid; } } return ans; } void clear(node *cur){ if(cur->left) clear(cur->left); if(cur->right) clear(cur->right); delete cur; } const T lpos, rpos; node *root; public: CHT(const T _lpos, const T _rpos) : lpos(_lpos), rpos(_rpos), root(new node()){ assert(lpos < rpos); } // ~CHT(){ clear(root); } // f(x) = a * x + b を挿入 void add_line(const T a, const T b){ node nw(a, b); return _add_line(root, &nw, lpos, rpos); } // x = k での最小値 T query(const T k) const { return query(root, k, lpos, rpos); } }; int main(){ ll mx=0, N; cin >> N; vector<ll> dp(N+1, -1e18), A(N+1), B(N); for (int i=1; i<=N; i++){ cin >> A[i]; B[i-1] = A[i]; } sort(B.begin(), B.end()); CHT<ll> cht(-1e5-1, 1e5+1); cht.add_line(0, 0); for (int i=1; i<=N; i++){ dp[i] = cht.query(A[i]); cht.add_line(-A[i], dp[i]); mx = max(-dp[i], mx); } cout << mx << endl; return 0; }