結果
問題 | No.2488 Mod Sum Maximization |
ユーザー |
👑 ![]() |
提出日時 | 2023-09-29 23:59:20 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 489 ms / 2,000 ms |
コード長 | 4,768 bytes |
コンパイル時間 | 762 ms |
コンパイル使用メモリ | 95,848 KB |
最終ジャッジ日時 | 2025-02-17 03:41:20 |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 38 |
ソースコード
#line 1 "..\\Main.cpp"#include <iostream>#include <string>#include <vector>#include <numeric>#include <algorithm>#include <atcoder/modint>#line 3 "D:\\Programming\\VSCode\\competitive-cpp\\nachia\\array\\segtree.hpp"namespace nachia{template<class S,S op(S l, S r)>struct Segtree {private:int N;std::vector<S> A;int xN;void mergev(int i){if(i < N) A[i] = op(A[i*2], A[i*2+1]);}template<class E>int minLeft2(int r, E cmp, int a = 0, int b = 0, int i = -1){static S x;if(i == -1){ a=0; b=N; i=1; x=A[0]; }if(r <= a) return a;if(b <= r){S nx = op(A[i], x);if(cmp(nx)){ x = nx; return a; }}if(b - a == 1) return b;int q = minLeft2(r, cmp, (a+b)/2, b, i*2+1);if(q > (a+b)/2) return q;return minLeft2(r, cmp, a, (a+b)/2, i*2);}template<class E>int maxRight2(int l, E cmp, int a = 0, int b = 0, int i = -1){static S x;if(i == -1){ a=0; b=N; i=1; x=A[0]; }if(b <= l) return b;if(l <= a){S nx = op(x, A[i]);if(cmp(nx)){ x = nx; return b; }}if(b - a == 1) return a;int q = maxRight2(l, cmp, a, (a+b)/2, i*2);if(q < (a+b)/2) return q;return maxRight2(l, cmp, (a+b)/2, b, i*2+1);}public:Segtree() : N(0) {}Segtree(int n, S e) : xN(n) {N = 1; while (N < n) N *= 2;A.assign(N * 2, e);}Segtree(const std::vector<S>& a, S e) : Segtree(a.size(), e){for(int i=0; i<(int)a.size(); i++) A[i + N] = a[i];for(int i=N-1; i>=1; i--) mergev(i);}void set(int p, S x){p += N; A[p] = x;for(int d=1; (1<<d)<=N; d++) mergev(p>>d);}S get(int p) const { return A[N+p]; }S prod(int l, int r) const {l += N; r += N;S ql = A[0], qr = A[0];while(l<r){if(l&1) ql = op(ql, A[l++]);if(r&1) qr = op(A[--r], qr);l /= 2;r /= 2;}return op(ql, qr);}S allProd() const { return A[1]; }// bool cmp(S)template<class E>int minLeft(int r, E cmp){return minLeft2(r, cmp);}// bool cmp(S)template<class E>int maxRight(int l, E cmp){int x = maxRight2(l, cmp);return x > xN ? xN : x;}};} // namespace nachia#line 3 "D:\\Programming\\VSCode\\competitive-cpp\\nachia\\range-query\\point-set-range-min.hpp"#include <functional>namespace nachia {template<class T, class Cmp = std::less<T>>struct PointSetRangeMin{private:static T minop(T l, T r){ return std::min(l, r, Cmp()); }using Base = Segtree<T, minop>;Base base;Cmp cmpx;public:PointSetRangeMin() {}PointSetRangeMin(int len, T INF): base(len, INF){}PointSetRangeMin(const std::vector<T>& init, T INF): base(init, INF){}T min(int l, int r){ return base.prod(l, r); }T min(){ return base.allProd(); }void set(int pos, T val){ base.set(pos, val); }T get(int pos){ return base.get(pos); }int lBoundLeft(int from, T val){ return base.minLeft(from, [this,val](const T& x){ return cmpx(val, x); }); }int uBoundLeft(int from, T val){ return base.minLeft(from, [this,val](const T& x){ return !cmpx(x, val); }); }int lBoundRight(int from, T val){ return base.maxRight(from, [this,val](const T& x){ return cmpx(val, x); }); }int uBoundRight(int from, T val){ return base.maxRight(from, [this,val](const T& x){ return !cmpx(x, val); }); }template<class E>int minLeft(int r, E cmp){ return base.minLeft(r, cmp); }template<class E>int maxRight(int l, E cmp){ return base.maxRight(l, cmp); }};} // namespace nachia#line 9 "..\\Main.cpp"using Modint = atcoder::static_modint<998244353>;using namespace std;using i64 = long long;using u64 = unsigned long long;#define rep(i,n) for(int i=0; i<(int)(n); i++)const i64 INF = 1001001001001001001;void testcase(){int N; cin >> N;vector<int> A(N); rep(i,N) cin >> A[i];i64 ans = 0;rep(i,N) ans += A[i];nachia::PointSetRangeMin<i64> ds(N,INF);ds.set(0, 0);rep(i,N){int a = A[i];i64 x = ds.min(i, N);for(int t=2; ; t++){int pos = lower_bound(A.begin(), A.end(), A[i]*t) - A.begin();ds.set(pos-1, min(ds.get(pos-1), x+a*(t-1)));if(pos == N) break;}}ans -= ds.get(N-1);cout << ans << endl;}int main(){ios::sync_with_stdio(false);cin.tie(nullptr);testcase();return 0;}