結果
問題 | No.2697 Range LIS Query |
ユーザー | 👑 emthrm |
提出日時 | 2024-03-22 22:27:55 |
言語 | C++23 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 425 ms / 10,000 ms |
コード長 | 11,852 bytes |
コンパイル時間 | 3,475 ms |
コンパイル使用メモリ | 270,564 KB |
実行使用メモリ | 27,812 KB |
最終ジャッジ日時 | 2024-09-30 11:58:11 |
合計ジャッジ時間 | 8,915 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 15 |
ソースコード
#include <bits/stdc++.h> using namespace std; #define FOR(i,m,n) for(int i=(m);i<(n);++i) #define REP(i,n) FOR(i,0,n) using ll = long long; constexpr int INF = 0x3f3f3f3f; constexpr long long LINF = 0x3f3f3f3f3f3f3f3fLL; constexpr double EPS = 1e-8; constexpr int MOD = 998244353; // constexpr int MOD = 1000000007; constexpr int DY4[]{1, 0, -1, 0}, DX4[]{0, -1, 0, 1}; constexpr int DY8[]{1, 1, 0, -1, -1, -1, 0, 1}; constexpr int DX8[]{0, -1, -1, -1, 0, 1, 1, 1}; template <typename T, typename U> inline bool chmax(T& a, U b) { return a < b ? (a = b, true) : false; } template <typename T, typename U> inline bool chmin(T& a, U b) { return a > b ? (a = b, true) : false; } struct IOSetup { IOSetup() { std::cin.tie(nullptr); std::ios_base::sync_with_stdio(false); std::cout << fixed << setprecision(20); } } iosetup; template <typename T> requires requires { typename T::Monoid; typename T::OperatorMonoid; {T::m_id()} -> std::same_as<typename T::Monoid>; {T::o_id()} -> std::same_as<typename T::OperatorMonoid>; {T::m_merge(std::declval<typename T::Monoid>(), std::declval<typename T::Monoid>())} -> std::same_as<typename T::Monoid>; {T::o_merge(std::declval<typename T::OperatorMonoid>(), std::declval<typename T::OperatorMonoid>())} -> std::same_as<typename T::OperatorMonoid>; {T::apply(std::declval<typename T::Monoid>(), std::declval<typename T::OperatorMonoid>())} -> std::same_as<typename T::Monoid>; } struct LazySegmentTree { using Monoid = typename T::Monoid; using OperatorMonoid = typename T::OperatorMonoid; explicit LazySegmentTree(const int n) : LazySegmentTree(std::vector<Monoid>(n, T::m_id())) {} explicit LazySegmentTree(const std::vector<Monoid>& a) : n(a.size()), height(std::countr_zero(std::bit_ceil(a.size()))), p2(1 << height) { lazy.assign(p2, T::o_id()); data.assign(p2 << 1, T::m_id()); std::copy(a.begin(), a.end(), data.begin() + p2); for (int i = p2 - 1; i > 0; --i) { data[i] = T::m_merge(data[i << 1], data[(i << 1) + 1]); } } void set(int idx, const Monoid val) { idx += p2; for (int i = height; i > 0; --i) { propagate(idx >> i); } data[idx] = val; for (int i = 1; i <= height; ++i) { const int current_idx = idx >> i; data[current_idx] = T::m_merge(data[current_idx << 1], data[(current_idx << 1) + 1]); } } void apply(int idx, const OperatorMonoid val) { idx += p2; for (int i = height; i > 0; --i) { propagate(idx >> i); } data[idx] = T::apply(data[idx], val); for (int i = 1; i <= height; ++i) { const int current_idx = idx >> i; data[current_idx] = T::m_merge(data[current_idx << 1], data[(current_idx << 1) + 1]); } } void apply(int left, int right, const OperatorMonoid val) { if (right <= left) [[unlikely]] return; left += p2; right += p2; const int ctz_left = std::countr_zero(static_cast<unsigned int>(left)); for (int i = height; i > ctz_left; --i) { propagate(left >> i); } const int ctz_right = std::countr_zero(static_cast<unsigned int>(right)); for (int i = height; i > ctz_right; --i) { propagate(right >> i); } for (int l = left, r = right; l < r; l >>= 1, r >>= 1) { if (l & 1) apply_sub(l++, val); if (r & 1) apply_sub(--r, val); } for (int i = left >> (ctz_left + 1); i > 0; i >>= 1) { data[i] = T::m_merge(data[i << 1], data[(i << 1) + 1]); } for (int i = right >> (ctz_right + 1); i > 0; i >>= 1) { data[i] = T::m_merge(data[i << 1], data[(i << 1) + 1]); } } Monoid get(int left, int right) { if (right <= left) [[unlikely]] return T::m_id(); left += p2; right += p2; const int ctz_left = std::countr_zero(static_cast<unsigned int>(left)); for (int i = height; i > ctz_left; --i) { propagate(left >> i); } const int ctz_right = std::countr_zero(static_cast<unsigned int>(right)); for (int i = height; i > ctz_right; --i) { propagate(right >> i); } Monoid res_l = T::m_id(), res_r = T::m_id(); for (; left < right; left >>= 1, right >>= 1) { if (left & 1) res_l = T::m_merge(res_l, data[left++]); if (right & 1) res_r = T::m_merge(data[--right], res_r); } return T::m_merge(res_l, res_r); } Monoid operator[](const int idx) { const int node = idx + p2; for (int i = height; i > 0; --i) { propagate(node >> i); } return data[node]; } template <typename G> int find_right(int left, const G g) { if (left >= n) [[unlikely]] return n; left += p2; for (int i = height; i > 0; --i) { propagate(left >> i); } Monoid val = T::m_id(); do { while (!(left & 1)) left >>= 1; Monoid nxt = T::m_merge(val, data[left]); if (!g(nxt)) { while (left < p2) { propagate(left); left <<= 1; nxt = T::m_merge(val, data[left]); if (g(nxt)) { val = nxt; ++left; } } return left - p2; } val = nxt; ++left; } while (!std::has_single_bit(static_cast<unsigned int>(left))); return n; } template <typename G> int find_left(int right, const G g) { if (right <= 0) [[unlikely]] return -1; right += p2; for (int i = height; i > 0; --i) { propagate((right - 1) >> i); } Monoid val = T::m_id(); do { --right; while (right > 1 && (right & 1)) right >>= 1; Monoid nxt = T::m_merge(data[right], val); if (!g(nxt)) { while (right < p2) { propagate(right); right = (right << 1) + 1; nxt = T::m_merge(data[right], val); if (g(nxt)) { val = nxt; --right; } } return right - p2; } val = nxt; } while (!std::has_single_bit(static_cast<unsigned int>(right))); return -1; } private: const int n, height, p2; std::vector<Monoid> data; std::vector<OperatorMonoid> lazy; void apply_sub(const int idx, const OperatorMonoid& val) { data[idx] = T::apply(data[idx], val); if (idx < p2) lazy[idx] = T::o_merge(lazy[idx], val); } void propagate(const int idx) { // assert(1 <= idx && idx < p2); apply_sub(idx << 1, lazy[idx]); apply_sub((idx << 1) + 1, lazy[idx]); lazy[idx] = T::o_id(); } }; namespace monoid { template <typename T> struct RangeMinimumAndUpdateQuery { using Monoid = T; using OperatorMonoid = T; static constexpr Monoid m_id() { return std::numeric_limits<Monoid>::max(); } static constexpr OperatorMonoid o_id() { return std::numeric_limits<OperatorMonoid>::max(); } static Monoid m_merge(const Monoid& a, const Monoid& b) { return std::min(a, b); } static OperatorMonoid o_merge(const OperatorMonoid& a, const OperatorMonoid& b) { return b == o_id() ? a : b; } static Monoid apply(const Monoid& a, const OperatorMonoid& b) { return b == o_id() ? a : b; } }; template <typename T> struct RangeMaximumAndUpdateQuery { using Monoid = T; using OperatorMonoid = T; static constexpr Monoid m_id() { return std::numeric_limits<Monoid>::lowest(); } static constexpr OperatorMonoid o_id() { return std::numeric_limits<OperatorMonoid>::lowest(); } static Monoid m_merge(const Monoid& a, const Monoid& b) { return std::max(a, b); } static OperatorMonoid o_merge(const OperatorMonoid& a, const OperatorMonoid& b) { return b == o_id() ? a : b; } static Monoid apply(const Monoid& a, const OperatorMonoid& b) { return b == o_id()? a : b; } }; template <typename T, T Inf> struct RangeMinimumAndAddQuery { using Monoid = T; using OperatorMonoid = T; static constexpr Monoid m_id() { return Inf; } static constexpr OperatorMonoid o_id() { return 0; } static Monoid m_merge(const Monoid& a, const Monoid& b) { return std::min(a, b); } static OperatorMonoid o_merge(const OperatorMonoid& a, const OperatorMonoid& b) { return a + b; } static Monoid apply(const Monoid& a, const OperatorMonoid& b) { return a + b; } }; template <typename T, T Inf> struct RangeMaximumAndAddQuery { using Monoid = T; using OperatorMonoid = T; static constexpr Monoid m_id() { return -Inf; } static constexpr OperatorMonoid o_id() { return 0; } static Monoid m_merge(const Monoid& a, const Monoid& b) { return std::max(a, b); } static OperatorMonoid o_merge(const OperatorMonoid& a, const OperatorMonoid& b) { return a + b; } static Monoid apply(const Monoid& a, const OperatorMonoid& b) { return a + b; } }; template <typename T> struct RangeSumAndUpdateQuery { using Monoid = struct { T sum; int len; }; using OperatorMonoid = T; static std::vector<Monoid> init(const int n) { return std::vector<Monoid>(n, Monoid{0, 1}); } static constexpr Monoid m_id() { return {0, 0}; } static constexpr OperatorMonoid o_id() { return std::numeric_limits<OperatorMonoid>::max(); } static Monoid m_merge(const Monoid& a, const Monoid& b) { return Monoid{a.sum + b.sum, a.len + b.len}; } static OperatorMonoid o_merge(const OperatorMonoid& a, const OperatorMonoid& b) { return b == o_id() ? a : b; } static Monoid apply(const Monoid& a, const OperatorMonoid& b) { return Monoid{b == o_id() ? a.sum : b * a.len, a.len}; } }; template <typename T> struct RangeSumAndAddQuery { using Monoid = struct { T sum; int len; }; using OperatorMonoid = T; static std::vector<Monoid> init(const int n) { return std::vector<Monoid>(n, Monoid{0, 1}); } static constexpr Monoid m_id() { return {0, 0}; } static constexpr OperatorMonoid o_id() { return 0; } static Monoid m_merge(const Monoid& a, const Monoid& b) { return Monoid{a.sum + b.sum, a.len + b.len}; } static OperatorMonoid o_merge(const OperatorMonoid& a, const OperatorMonoid& b) { return a + b; } static Monoid apply(const Monoid& a, const OperatorMonoid& b) { return Monoid{a.sum + b * a.len, a.len}; } }; } // namespace monoid int main() { constexpr int M = 4; struct S { using Monoid = pair<array<array<int, M>, M>, int>; using OperatorMonoid = int; static constexpr Monoid m_id() { return Monoid{}; } static constexpr OperatorMonoid o_id() { return -1; } static Monoid m_merge(Monoid a, const Monoid& b) { Monoid res{}; REP(i, M) FOR(j, i, M) res.first[i][j] = max(a.first[i][j], b.first[i][j]); REP(i, M) FOR(j, i, M) FOR(k, j, M) chmax(res.first[i][k], a.first[i][j] + b.first[j][k]); res.second = a.second + b.second; return res; } static OperatorMonoid o_merge(const OperatorMonoid& a, const OperatorMonoid& b) { return b == -1 ? a : b; } static Monoid apply(const Monoid& a, const OperatorMonoid& b) { if (b == -1) return a; Monoid res{}; REP(i, b + 1) FOR(j, b, M) res.first[i][j] = a.second; res.second = a.second; return res; } }; int n; cin >> n; LazySegmentTree<S> a(vector(n, S::Monoid{array<array<int, M>, M>{}, 1})); REP(i, n) { int a_i; cin >> a_i; --a_i; a.apply(i, a_i); } int q; cin >> q; while (q--) { int type, l, r; cin >> type >> l >> r; --l; --r; if (type == 1) { const auto dp = a.get(l, r + 1).first; int ans = 0; for (const auto& row : dp) chmax(ans, ranges::max(row)); cout << ans << '\n'; } else if (type == 2) { int x; cin >> x; --x; a.apply(l, r + 1, x); } } return 0; }