結果
問題 | No.2219 Re:010 |
ユーザー | r1933 |
提出日時 | 2023-02-17 23:27:30 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
WA
|
実行時間 | - |
コード長 | 10,370 bytes |
コンパイル時間 | 2,379 ms |
コンパイル使用メモリ | 208,416 KB |
実行使用メモリ | 6,948 KB |
最終ジャッジ日時 | 2024-07-19 14:39:40 |
合計ジャッジ時間 | 3,735 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
6,812 KB |
testcase_01 | AC | 2 ms
6,940 KB |
testcase_02 | AC | 2 ms
6,944 KB |
testcase_03 | WA | - |
testcase_04 | WA | - |
testcase_05 | WA | - |
testcase_06 | WA | - |
testcase_07 | WA | - |
testcase_08 | WA | - |
testcase_09 | WA | - |
testcase_10 | WA | - |
testcase_11 | WA | - |
testcase_12 | WA | - |
testcase_13 | WA | - |
testcase_14 | WA | - |
testcase_15 | WA | - |
testcase_16 | WA | - |
testcase_17 | WA | - |
testcase_18 | AC | 8 ms
6,944 KB |
testcase_19 | AC | 27 ms
6,940 KB |
testcase_20 | AC | 55 ms
6,944 KB |
testcase_21 | AC | 2 ms
6,940 KB |
testcase_22 | AC | 2 ms
6,944 KB |
testcase_23 | AC | 2 ms
6,940 KB |
ソースコード
#include "bits/stdc++.h" #pragma GCC optimize("Ofast") // Begin Header {{{ #pragma region using namespace std; #ifndef DEBUG #define dump(...) #endif #define all(x) begin(x), end(x) #define rall(x) rbegin(x), rend(x) #define rep(i, b, e) for (intmax_t i = (b), i##_limit = (e); i < i##_limit; ++i) #define repc(i, b, e) for (intmax_t i = (b), i##_limit = (e); i <= i##_limit; ++i) #define repr(i, b, e) for (intmax_t i = (b), i##_limit = (e); i >= i##_limit; --i) #define var(Type, ...) \ Type __VA_ARGS__; \ input(__VA_ARGS__) #define let const auto constexpr size_t operator""_zu(unsigned long long value) { return value; }; constexpr intmax_t operator""_jd(unsigned long long value) { return value; }; constexpr uintmax_t operator""_ju(unsigned long long value) { return value; }; constexpr int INF = 0x3f3f3f3f; constexpr intmax_t LINF = 0x3f3f3f3f3f3f3f3f_jd; using usize = size_t; using imax = intmax_t; using uimax = uintmax_t; using ld = long double; template <class T, class Compare = less<>> using MaxHeap = priority_queue<T, vector<T>, Compare>; template <class T, class Compare = greater<>> using MinHeap = priority_queue<T, vector<T>, Compare>; inline void input() {} template <class Head, class... Tail> inline void input(Head&& head, Tail&&... tail) { cin >> head; input(forward<Tail>(tail)...); } template <class Container, class Value = typename Container::value_type, enable_if_t<!is_same<Container, string>::value, nullptr_t> = nullptr> inline istream& operator>>(istream& is, Container& vs) { for (auto& v: vs) is >> v; return is; } template <class T, class U> inline istream& operator>>(istream& is, pair<T, U>& p) { is >> p.first >> p.second; return is; } inline void output() { cout << "\n"; } template <class Head, class... Tail> inline void output(Head&& head, Tail&&... tail) { cout << head; if (sizeof...(tail)) cout << " "; output(forward<Tail>(tail)...); } template <class Container, class Value = typename Container::value_type, enable_if_t<!is_same<Container, string>::value, nullptr_t> = nullptr> inline ostream& operator<<(ostream& os, const Container& vs) { static constexpr const char* delim[] = {" ", ""}; for (auto it = begin(vs); it != end(vs); ++it) { os << delim[it == begin(vs)] << *it; } return os; } template <class Iterator> inline void join(const Iterator& Begin, const Iterator& End, const string& delim = "\n", const string& last = "\n") { for (auto it = Begin; it != End; ++it) { cout << ((it == Begin) ? "" : delim) << *it; } cout << last; } template <class T> inline vector<T> makeVector(const T& init_value, size_t sz) { return vector<T>(sz, init_value); } template <class T, class... Args> inline auto makeVector(const T& init_value, size_t sz, Args... args) { return vector<decltype(makeVector<T>(init_value, args...))>( sz, makeVector<T>(init_value, args...)); } template <class Func> class FixPoint : Func { public: explicit constexpr FixPoint(Func&& f) noexcept : Func(forward<Func>(f)) {} template <class... Args> constexpr decltype(auto) operator()(Args&&... args) const { return Func::operator()(*this, std::forward<Args>(args)...); } }; template <class Func> static inline constexpr decltype(auto) makeFixPoint(Func&& f) noexcept { return FixPoint<Func> {forward<Func>(f)}; } template <class Container> struct reverse_t { Container& c; reverse_t(Container& c) : c(c) {} auto begin() { return c.rbegin(); } auto end() { return c.rend(); } }; template <class Container> auto reversed(Container& c) { return reverse_t<Container>(c); } template <class T> inline bool chmax(T& a, const T& b) noexcept { return b > a && (a = b, true); } template <class T> inline bool chmin(T& a, const T& b) noexcept { return b < a && (a = b, true); } template <class T> inline T diff(const T& a, const T& b) noexcept { return a < b ? b - a : a - b; } template <class T> inline T sigma(const T& l, const T& r) noexcept { return (l + r) * (r - l + 1) / 2; } template <class T> inline T ceildiv(const T& n, const T& d) noexcept { return (n + d - 1) / d; } void operator|=(vector<bool>::reference lhs, const bool rhs) { lhs = lhs | rhs; } void operator&=(vector<bool>::reference lhs, const bool rhs) { lhs = lhs & rhs; } void operator^=(vector<bool>::reference lhs, const bool rhs) { lhs = lhs ^ rhs; } void ioinit() { ios_base::sync_with_stdio(false); cin.tie(nullptr); cout << fixed << setprecision(10); cerr << fixed << setprecision(10); clog << fixed << setprecision(10); } #pragma endregion // }}} End Header // FenwickTree {{{ template <class T> struct FenwickTree { vector<T> ft; const size_t SIZE; explicit FenwickTree(size_t n) : ft(n + 5, 0), SIZE(1 << (__lg(n + 5) + 1)) {} void add(int i, const T& v) { for (++i; i < ft.size(); i += i & -i) ft[i] += v; } // [0, i] T sum(int i) const { T ret = 0; for (++i; i > 0; i -= i & -i) ret += ft[i]; return ret; } // [s, t] T sum(int s, int t) const { if (s > t) swap(s, t); return sum(t) - sum(s - 1); } size_t lower_bound(T v) const { if (v <= 0) return 0; T x = 0; for (size_t k = SIZE; k > 0; k >>= 1) { if (x + k < ft.size() && ft[x + k] < v) { v -= ft[x + k]; x += k; } } return x; } }; // }}} // ModInt {{{ template <intmax_t Modulo> class ModInt { public: using value_type = intmax_t; private: static constexpr value_type cmod = Modulo; // compile-time static value_type rmod; // runtime value_type value = 0; static constexpr value_type inverse(value_type n, value_type m) { value_type a = n; value_type b = m; value_type x = 0; value_type y = 1; for (value_type u = y, v = x; a;) { const value_type t = b / a; swap(x -= t * u, u); swap(y -= t * v, v); swap(b -= t * a, a); } if ((x %= m) < 0) x += m; return x; } static value_type normalize(intmax_t n, value_type m) { if (n >= m) { n %= m; } else if (n < 0) { if ((n %= m) < 0) n += m; } return n; } public: ModInt() = default; ModInt(intmax_t n) : value(normalize(n, getModulo())) {} template <typename T> constexpr explicit operator T() const { return static_cast<T>(value); } ModInt& operator=(intmax_t n) { value = normalize(n, getModulo()); return *this; } ModInt& operator+=(const ModInt& other) { if ((value += other.value) >= getModulo()) value -= getModulo(); return *this; } ModInt& operator-=(const ModInt& other) { if ((value -= other.value) < 0) value += getModulo(); return *this; } ModInt& operator*=(const ModInt& other) { value = (value * other.value) % getModulo(); return *this; } ModInt& operator/=(const ModInt& other) { value = (value * inverse(other.value, getModulo())) % getModulo(); return *this; } ModInt& operator++() { if (++value == getModulo()) value = 0; return *this; } ModInt& operator--() { if (value-- == 0) value = getModulo() - 1; return *this; } ModInt operator++(int) { const ModInt tmp(*this); ++*this; return tmp; } ModInt operator--(int) { const ModInt tmp(*this); --*this; return tmp; } friend ModInt operator+(ModInt lhs, const ModInt& rhs) { return lhs += rhs; } friend ModInt operator-(ModInt lhs, const ModInt& rhs) { return lhs -= rhs; } friend ModInt operator*(ModInt lhs, const ModInt& rhs) { return lhs *= rhs; } friend ModInt operator/(ModInt lhs, const ModInt& rhs) { return lhs /= rhs; } ModInt operator+() const { return *this; } ModInt operator-() const { if (value == 0) return *this; return ModInt(getModulo() - value); } friend bool operator==(const ModInt& lhs, const ModInt& rhs) { return lhs.value == rhs.value; } friend bool operator!=(const ModInt& lhs, const ModInt& rhs) { return !(lhs == rhs); } friend ostream& operator<<(ostream& os, const ModInt& n) { return os << n.value; } friend istream& operator>>(istream& is, ModInt& n) { is >> n.value; n.value = normalize(n.value, getModulo()); return is; } static value_type getModulo() { return ((cmod > 0) ? cmod : rmod); } template <int M = Modulo, typename T = typename enable_if<(M <= 0)>::type> static T setModulo(value_type m) { rmod = m; } }; template <intmax_t M> constexpr typename ModInt<M>::value_type ModInt<M>::cmod; template <intmax_t M> typename ModInt<M>::value_type ModInt<M>::rmod; // }}} // constexpr intmax_t MOD = intmax_t(1e9) + 7; constexpr intmax_t MOD = 998244353; using Mint = ModInt<MOD>; template <class T> T power(const T& b, const intmax_t& e) { T ret = 1; T n = b; for (intmax_t p = abs(e); p > 0; p >>= 1) { if (p & 1) ret *= n; n *= n; } if (e < 0) return T(1) / ret; return ret; } int main() { ioinit(); var(string, S); const size_t N = S.size(); FenwickTree<intmax_t> ft0(N + 1), ftq(N + 1); rep(i, 0, N) { if (S[i] == '0') ft0.add(i + 1, 1); if (S[i] == '?') ftq.add(i + 1, 1); } Mint res = 0; rep(i, 0, N) { if (S[i] == '1' || S[i] == '?') { const auto psum0 = ft0.sum(0, i); const auto psumq = ftq.sum(0, i); const auto ssum0 = ft0.sum(i + 2, N + 1); const auto ssumq = ftq.sum(i + 2, N + 1); dump(i, psum0, psumq, ssum0, ssumq); res += Mint(Mint(psum0) * Mint(psumq + 1) + power<Mint>(2, psumq - 1) * Mint(psumq)) * Mint(Mint(ssum0) * Mint(ssumq + 1) + power<Mint>(2, ssumq - 1) * Mint(ssumq)); } } output(res); return 0; }