結果

問題 No.3053 $((0 \And 1)\mathop{|}2)\oplus 3$
コンテスト
ユーザー KumaTachiRen
提出日時 2025-03-07 22:18:24
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
(最新)
AC  
(最初)
実行時間 -
コード長 1,500 bytes
コンパイル時間 3,527 ms
コンパイル使用メモリ 279,956 KB
実行使用メモリ 12,180 KB
最終ジャッジ日時 2025-03-08 00:11:07
合計ジャッジ時間 13,798 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 34 WA * 1
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <atcoder/modint>

using namespace std;
using mint = atcoder::modint998244353;
using mat = array<mint, 4>;

mat Mul(const mat& a, const mat& b) {
    return {
    	a[0] * b[0] + a[1] * b[2],
    	a[0] * b[1] + a[1] * b[3],
        a[2] * b[0] + a[3] * b[2],
        a[2] * b[1] + a[3] * b[3]
    };
}

mat Pow(mat a, long long n) {
    mat r = {1, 0, 0, 1};
    while (n > 0) {
        if (n % 2 == 1) r = Mul(r, a);
        a = Mul(a, a);
        n /= 2;
    }
    return r;
}

void Solve() {
    string s;
    cin >> s;
    mint ans = 0;
    
    mat m0 = {3, 1, 0, 2}, m1 = {1, 1, 2, 2};
    mat q0 = {3, 1, 0, 2}, q1 = {1, 1, 2, 2};
    vector<long long> ls(s.size() + 1, 0);
    constexpr long long m = 998244353 - 1;
    for (size_t i = 0; i < s.size(); i++)
        ls[i + 1] = (ls[i] * 2 + (s[i] - '0')) % m;
    
    mint p = 1, inv = mint(1) / 6;
    for (int i = s.size() - 1; i >= 0; i--) {
        mat c = {2 * inv, -1 * inv, 0 * inv, 3 * inv};
        c = Mul(Pow(Mul(m1, m0), ls[i]), c);
        if (s[i] == '0') {
            c = Mul(q0, c);
        } else {
            c = Mul(m0, c);
            c = Mul(q1, c);
        }
        ans += c[2] * p;
        p *= 2;
        if (s[i] == '1') {
            q0 = Mul(q0, m0);
            q1 = Mul(q1, m1);
        }
        m0 = Mul(m0, m0);
        m1 = Mul(m1, m1);
    }
    cout << ans.val() << '\n';
}

int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    Solve();
    return 0;
}
0