結果
問題 | No.3129 Multiple of Twin Subarray |
ユーザー |
|
提出日時 | 2025-04-25 22:47:57 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 289 ms / 2,000 ms |
コード長 | 4,834 bytes |
コンパイル時間 | 1,941 ms |
コンパイル使用メモリ | 200,000 KB |
実行使用メモリ | 52,080 KB |
最終ジャッジ日時 | 2025-04-25 22:48:10 |
合計ジャッジ時間 | 11,350 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 46 |
ソースコード
#include <bits/stdc++.h> using namespace std; using ll = long long; using pii = pair<int, int>; using pll = pair<ll, ll>; namespace atcoder { namespace internal { #if __cplusplus >= 202002L using std::bit_ceil; #else // @return same with std::bit::bit_ceil unsigned int bit_ceil(unsigned int n) { unsigned int x = 1; while (x < (unsigned int)(n)) x *= 2; return x; } #endif // @param n `1 <= n` // @return same with std::bit::countr_zero int countr_zero(unsigned int n) { #ifdef _MSC_VER unsigned long index; _BitScanForward(&index, n); return index; #else return __builtin_ctz(n); #endif } // @param n `1 <= n` // @return same with std::bit::countr_zero constexpr int countr_zero_constexpr(unsigned int n) { int x = 0; while (!(n & (1 << x))) x++; return x; } } // namespace internal } // namespace atcoder namespace atcoder { #if __cplusplus >= 201703L template <class S, auto op, auto e> struct segtree { static_assert(std::is_convertible_v<decltype(op), std::function<S(S, S)>>, "op must work as S(S, S)"); static_assert(std::is_convertible_v<decltype(e), std::function<S()>>, "e must work as S()"); #else template <class S, S (*op)(S, S), S (*e)()> struct segtree { #endif public: segtree() : segtree(0) {} explicit segtree(int n) : segtree(std::vector<S>(n, e())) {} explicit segtree(const std::vector<S>& v) : _n(int(v.size())) { size = (int)internal::bit_ceil((unsigned int)(_n)); log = internal::countr_zero((unsigned int)size); d = std::vector<S>(2 * size, e()); for (int i = 0; i < _n; i++) d[size + i] = v[i]; for (int i = size - 1; i >= 1; i--) { update(i); } } void set(int p, S x) { assert(0 <= p && p < _n); p += size; d[p] = x; for (int i = 1; i <= log; i++) update(p >> i); } S get(int p) const { assert(0 <= p && p < _n); return d[p + size]; } S prod(int l, int r) const { assert(0 <= l && l <= r && r <= _n); S sml = e(), smr = e(); l += size; r += size; while (l < r) { if (l & 1) sml = op(sml, d[l++]); if (r & 1) smr = op(d[--r], smr); l >>= 1; r >>= 1; } return op(sml, smr); } S all_prod() const { return d[1]; } template <bool (*f)(S)> int max_right(int l) const { return max_right(l, [](S x) { return f(x); }); } template <class F> int max_right(int l, F f) const { assert(0 <= l && l <= _n); assert(f(e())); if (l == _n) return _n; l += size; S sm = e(); do { while (l % 2 == 0) l >>= 1; if (!f(op(sm, d[l]))) { while (l < size) { l = (2 * l); if (f(op(sm, d[l]))) { sm = op(sm, d[l]); l++; } } return l - size; } sm = op(sm, d[l]); l++; } while ((l & -l) != l); return _n; } template <bool (*f)(S)> int min_left(int r) const { return min_left(r, [](S x) { return f(x); }); } template <class F> int min_left(int r, F f) const { assert(0 <= r && r <= _n); assert(f(e())); if (r == 0) return 0; r += size; S sm = e(); do { r--; while (r > 1 && (r % 2)) r >>= 1; if (!f(op(d[r], sm))) { while (r < size) { r = (2 * r + 1); if (f(op(d[r], sm))) { sm = op(d[r], sm); r--; } } return r + 1 - size; } sm = op(d[r], sm); } while ((r & -r) != r); return 0; } private: int _n, size, log; std::vector<S> d; void update(int k) { d[k] = op(d[2 * k], d[2 * k + 1]); } }; } // namespace atcoder #define int long long int n, q; int max5(int a, int b, int c, int d, int e) { return max(a, max(max(b, c), max(d, e))); } struct Data { bool check; int lsum, rsum, msum, sum; Data(int x = 0, bool c = false) { lsum = rsum = msum = sum = x; check = c; } Data(int lsum, int rsum, int msum, int sum): lsum(lsum), rsum(rsum), msum(msum), sum(sum) {} }; Data op(Data a, Data b) { if (!a.check || !b.check) { if (!a.check && !b.check) return Data(); if (!a.check) return b; else if (!b.check) return a; } int lsum = max(a.lsum, a.sum+b.lsum); int rsum = max(b.rsum, a.rsum+b.sum); int msum = max5(a.msum, b.msum, lsum, rsum, a.rsum+b.lsum); int sum = a.sum+b.sum; return Data(lsum, rsum, msum, sum); } Data e() { return Data(); } ll ans = -1e18; int32_t main() { ios::sync_with_stdio(0); cin.tie(0); cout.tie(0); cin >> n; atcoder::segtree<Data, op, e> seg(n), mseg(n); for (int i = 0; i < n; i++) { int x; cin >> x; // x += 1000; seg.set(i, Data(x, true)); mseg.set(i, Data(-x, true)); } for (int i = 1; i < n; i++) { ans = max({ans, seg.prod(0, i).msum*seg.prod(i, n).msum, mseg.prod(0, i).msum*mseg.prod(i, n).msum }); } cout << ans; }