結果

問題 No.1677 mæx
コンテスト
ユーザー merom686
提出日時 2021-09-10 22:53:43
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 10 ms / 2,000 ms
コード長 1,462 bytes
コンパイル時間 2,039 ms
コンパイル使用メモリ 203,560 KB
最終ジャッジ日時 2025-01-24 12:00:51
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <atcoder/modint>
using mint = atcoder::modint998244353;
using namespace std;
using ll = long long;

string s;
vector<int> a;

array<mint, 3> dfs(int i0) {
    array<mint, 3> r = {};
    if (s[i0] == 'm') {
        int i2;
        if (s[i0 + 4] == 'm') {
            i2 = a[i0 + 7] + 1;
        } else {
            i2 = i0 + 5;
        }
        array<mint, 3> x = dfs(i0 + 4);
        array<mint, 3> y = dfs(i2 + 1);
        if (s[i0 + 1] != 'e') {
            r[0] = x[0] * y[0];
            r[1] = (x[0] + x[1]) * (y[0] + y[1]) - r[0];
            r[2] = (x[0] + x[1] + x[2]) * (y[0] + y[1] + y[2]) - r[1] - r[0];
        }
        if (s[i0 + 1] != 'a') {
            r[0] += (x[1] + x[2]) * (y[1] + y[2]);
            r[1] += x[0] * y[0] + x[2] * y[0] + x[0] * y[2];
            r[2] += x[0] * y[1] + x[1] * y[0];
        }
    } else {
        if (s[i0] == '?') {
            r[0] = r[1] = r[2] = 1;
        } else {
            r[s[i0] - '0'] = 1;
        }
    }
    //cout << i0 << ' ' << r[0].val() << ' ' << r[1].val() << ' ' << r[2].val() << endl;
    return r;
}

int main() {
    ios::sync_with_stdio(false);
    cin.tie(0);

    int k;
    cin >> s >> k;
    int n = s.size();

    a.resize(n);
    stack<int> st;

    for (int i = 0; i < n; i++) {
        if (s[i] == '(') st.push(i);
        if (s[i] == ')') a[st.top()] = i, st.pop();
    }

    auto x = dfs(0);
    cout << x[k].val() << endl;

    return 0;
}
0