結果
問題 | No.2611 Count 01 |
ユーザー | 👑 hos.lyric |
提出日時 | 2024-01-19 21:40:02 |
言語 | C++14 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 1,563 ms / 6,000 ms |
コード長 | 8,665 bytes |
コンパイル時間 | 1,064 ms |
コンパイル使用メモリ | 117,132 KB |
実行使用メモリ | 77,500 KB |
最終ジャッジ日時 | 2024-09-28 04:12:36 |
合計ジャッジ時間 | 33,906 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
6,816 KB |
testcase_01 | AC | 1 ms
6,940 KB |
testcase_02 | AC | 2 ms
6,944 KB |
testcase_03 | AC | 1,538 ms
77,344 KB |
testcase_04 | AC | 1,557 ms
77,472 KB |
testcase_05 | AC | 1,558 ms
77,460 KB |
testcase_06 | AC | 1,562 ms
77,440 KB |
testcase_07 | AC | 1,560 ms
77,500 KB |
testcase_08 | AC | 1,557 ms
77,432 KB |
testcase_09 | AC | 1,533 ms
77,424 KB |
testcase_10 | AC | 1,555 ms
77,404 KB |
testcase_11 | AC | 1,549 ms
77,320 KB |
testcase_12 | AC | 1,552 ms
77,448 KB |
testcase_13 | AC | 1,537 ms
77,404 KB |
testcase_14 | AC | 1,563 ms
77,496 KB |
testcase_15 | AC | 1,534 ms
77,452 KB |
testcase_16 | AC | 1,529 ms
77,444 KB |
testcase_17 | AC | 1,555 ms
77,304 KB |
testcase_18 | AC | 1,549 ms
77,392 KB |
testcase_19 | AC | 1,549 ms
77,404 KB |
testcase_20 | AC | 1,550 ms
77,356 KB |
testcase_21 | AC | 1,537 ms
77,348 KB |
testcase_22 | AC | 1,538 ms
77,464 KB |
ソースコード
#include <cassert> #include <cmath> #include <cstdint> #include <cstdio> #include <cstdlib> #include <cstring> #include <algorithm> #include <bitset> #include <complex> #include <deque> #include <functional> #include <iostream> #include <limits> #include <map> #include <numeric> #include <queue> #include <random> #include <set> #include <sstream> #include <string> #include <unordered_map> #include <unordered_set> #include <utility> #include <vector> using namespace std; using Int = long long; template <class T1, class T2> ostream &operator<<(ostream &os, const pair<T1, T2> &a) { return os << "(" << a.first << ", " << a.second << ")"; }; template <class T> ostream &operator<<(ostream &os, const vector<T> &as) { const int sz = as.size(); os << "["; for (int i = 0; i < sz; ++i) { if (i >= 256) { os << ", ..."; break; } if (i > 0) { os << ", "; } os << as[i]; } return os << "]"; } template <class T> void pv(T a, T b) { for (T i = a; i != b; ++i) cerr << *i << " "; cerr << endl; } template <class T> bool chmin(T &t, const T &f) { if (t > f) { t = f; return true; } return false; } template <class T> bool chmax(T &t, const T &f) { if (t < f) { t = f; return true; } return false; } #define COLOR(s) ("\x1b[" s "m") //////////////////////////////////////////////////////////////////////////////// template <unsigned M_> struct ModInt { static constexpr unsigned M = M_; unsigned x; constexpr ModInt() : x(0U) {} constexpr ModInt(unsigned x_) : x(x_ % M) {} constexpr ModInt(unsigned long long x_) : x(x_ % M) {} constexpr ModInt(int x_) : x(((x_ %= static_cast<int>(M)) < 0) ? (x_ + static_cast<int>(M)) : x_) {} constexpr ModInt(long long x_) : x(((x_ %= static_cast<long long>(M)) < 0) ? (x_ + static_cast<long long>(M)) : x_) {} ModInt &operator+=(const ModInt &a) { x = ((x += a.x) >= M) ? (x - M) : x; return *this; } ModInt &operator-=(const ModInt &a) { x = ((x -= a.x) >= M) ? (x + M) : x; return *this; } ModInt &operator*=(const ModInt &a) { x = (static_cast<unsigned long long>(x) * a.x) % M; return *this; } ModInt &operator/=(const ModInt &a) { return (*this *= a.inv()); } ModInt pow(long long e) const { if (e < 0) return inv().pow(-e); ModInt a = *this, b = 1U; for (; e; e >>= 1) { if (e & 1) b *= a; a *= a; } return b; } ModInt inv() const { unsigned a = M, b = x; int y = 0, z = 1; for (; b; ) { const unsigned q = a / b; const unsigned c = a - q * b; a = b; b = c; const int w = y - static_cast<int>(q) * z; y = z; z = w; } assert(a == 1U); return ModInt(y); } ModInt operator+() const { return *this; } ModInt operator-() const { ModInt a; a.x = x ? (M - x) : 0U; return a; } ModInt operator+(const ModInt &a) const { return (ModInt(*this) += a); } ModInt operator-(const ModInt &a) const { return (ModInt(*this) -= a); } ModInt operator*(const ModInt &a) const { return (ModInt(*this) *= a); } ModInt operator/(const ModInt &a) const { return (ModInt(*this) /= a); } template <class T> friend ModInt operator+(T a, const ModInt &b) { return (ModInt(a) += b); } template <class T> friend ModInt operator-(T a, const ModInt &b) { return (ModInt(a) -= b); } template <class T> friend ModInt operator*(T a, const ModInt &b) { return (ModInt(a) *= b); } template <class T> friend ModInt operator/(T a, const ModInt &b) { return (ModInt(a) /= b); } explicit operator bool() const { return x; } bool operator==(const ModInt &a) const { return (x == a.x); } bool operator!=(const ModInt &a) const { return (x != a.x); } friend std::ostream &operator<<(std::ostream &os, const ModInt &a) { return os << a.x; } }; //////////////////////////////////////////////////////////////////////////////// constexpr unsigned MO = 998244353; using Mint = ModInt<MO>; // T: monoid representing information of an interval. // T() should return the identity. // T(S s) should represent a single element of the array. // T::pull(const T &l, const T &r) should pull two intervals. template <class T> struct SegmentTreePoint { int logN, n; vector<T> ts; SegmentTreePoint() : logN(0), n(0) {} explicit SegmentTreePoint(int n_) { for (logN = 0, n = 1; n < n_; ++logN, n <<= 1) {} ts.resize(n << 1); } template <class S> explicit SegmentTreePoint(const vector<S> &ss) { const int n_ = ss.size(); for (logN = 0, n = 1; n < n_; ++logN, n <<= 1) {} ts.resize(n << 1); for (int i = 0; i < n_; ++i) at(i) = T(ss[i]); build(); } T &at(int i) { return ts[n + i]; } void build() { for (int u = n; --u; ) pull(u); } inline void pull(int u) { ts[u].pull(ts[u << 1], ts[u << 1 | 1]); } // Changes the value of point a to s. template <class S> void change(int a, const S &s) { assert(0 <= a); assert(a < n); ts[a += n] = T(s); for (; a >>= 1; ) pull(a); } // Applies T::f(args...) to point a. template <class F, class... Args> void ch(int a, F f, Args &&... args) { assert(0 <= a); assert(a < n); (ts[a += n].*f)(args...); for (; a >>= 1; ) pull(a); } // Calculates the product for [a, b). T get(int a, int b) { assert(0 <= a); assert(a <= b); assert(b <= n); if (a == b) return T(); a += n; b += n; T prodL, prodR, t; for (int aa = a, bb = b; aa < bb; aa >>= 1, bb >>= 1) { if (aa & 1) { t.pull(prodL, ts[aa++]); prodL = t; } if (bb & 1) { t.pull(ts[--bb], prodR); prodR = t; } } t.pull(prodL, prodR); return t; } // Calculates T::f(args...) of a monoid type for [a, b). // op(-, -) should calculate the product. // e() should return the identity. template <class Op, class E, class F, class... Args> #if __cplusplus >= 201402L auto #else decltype((std::declval<T>().*F())()) #endif get(int a, int b, Op op, E e, F f, Args &&... args) { assert(0 <= a); assert(a <= b); assert(b <= n); if (a == b) return e(); a += n; b += n; auto prodL = e(), prodR = e(); for (int aa = a, bb = b; aa < bb; aa >>= 1, bb >>= 1) { if (aa & 1) prodL = op(prodL, (ts[aa++].*f)(args...)); if (bb & 1) prodR = op((ts[--bb].*f)(args...), prodR); } return op(prodL, prodR); } // Find min b s.t. T::f(args...) returns true, // when called for the partition of [a, b) from left to right. // Returns n + 1 if there is no such b. template <class F, class... Args> int findRight(int a, F f, Args &&... args) { assert(0 <= a); assert(a <= n); if ((T().*f)(args...)) return a; if (a == n) return n + 1; a += n; for (; ; a >>= 1) if (a & 1) { if ((ts[a].*f)(args...)) { for (; a < n; ) { if (!(ts[a <<= 1].*f)(args...)) ++a; } return a - n + 1; } ++a; if (!(a & (a - 1))) return n + 1; } } // Find max a s.t. T::f(args...) returns true, // when called for the partition of [a, b) from right to left. // Returns -1 if there is no such a. template <class F, class... Args> int findLeft(int b, F f, Args &&... args) { assert(0 <= b); assert(b <= n); if ((T().*f)(args...)) return b; if (b == 0) return -1; b += n; for (; ; b >>= 1) if ((b & 1) || b == 2) { if ((ts[b - 1].*f)(args...)) { for (; b <= n; ) { if (!(ts[(b <<= 1) - 1].*f)(args...)) --b; } return b - n - 1; } --b; if (!(b & (b - 1))) return -1; } } }; //////////////////////////////////////////////////////////////////////////////// constexpr int U = 6; struct Node { Mint a[U][U]; Node() : a{} { for (int u = 0; u < U; ++u) a[u][u] = 1; } Node(char c) : a{} { for (int u = 0; u < U; ++u) a[u][u] = 1; a[0][1] = a[4][5] = 1; if (c == '0') { a[0][2] = a[1][2] = a[3][4] = a[3][5] = 1; } else if (c == '1') { a[0][3] = a[1][3] = a[2][4] = a[2][5] = 1; } else { assert(false); } } void pull(const Node &l, const Node &r) { memset(a, 0, sizeof(a)); for (int u = 0; u < U; ++u) for (int w = 0; w < U; ++w) for (int v = 0; v < U; ++v) { a[u][v] += l.a[u][w] * r.a[w][v]; } } }; int N, Q; char S[150'010]; int main() { for (; ~scanf("%d%d", &N, &Q); ) { scanf("%s", S); SegmentTreePoint<Node> seg(vector<char>(S, S + N)); for (; Q--; ) { int O; scanf("%d", &O); if (O == 1) { int I; scanf("%d", &I); --I; seg.change(I, S[I] ^= 1); } else if (O == 2) { int L, R; scanf("%d%d", &L, &R); --L; const auto res = seg.get(L, R); printf("%u\n", res.a[0][5].x); } else { assert(false); } } cerr<<"========"<<endl; } return 0; }