結果

問題 No.1677 mæx
ユーザー SSRSSSRS
提出日時 2021-09-10 21:39:25
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 21 ms / 2,000 ms
コード長 1,049 bytes
コンパイル時間 1,891 ms
コンパイル使用メモリ 174,908 KB
実行使用メモリ 11,136 KB
最終ジャッジ日時 2024-06-11 22:54:00
合計ジャッジ時間 3,217 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
const long long MOD = 998244353;
vector<vector<int>> mex = {{1, 2, 1}, {2, 0, 0}, {1, 0, 0}};
vector<long long> eval(string &S, int &p){
  if (S[p] == '0'){
    p++;
    return {1, 0, 0};
  } else if (S[p] == '1'){
    p++;
    return {0, 1, 0};
  } else if (S[p] == '2'){
    p++;
    return {0, 0, 1};
  } else if (S[p] == '?'){
    p++;
    return {1, 1, 1};
  } else {
    string op = S.substr(p, 3);
    vector<long long> ans(3, 0);
    p += 4;
    vector<long long> A = eval(S, p);
    p++;
    vector<long long> B = eval(S, p);
    p++;
    for (int i = 0; i < 3; i++){
      for (int j = 0; j < 3; j++){
        if (op != "mex"){
          ans[max(i, j)] += A[i] * B[j];
          ans[max(i, j)] %= MOD;
        }
        if (op != "max"){
          ans[mex[i][j]] += A[i] * B[j];
          ans[mex[i][j]] %= MOD;
        }
      }
    }
    return ans;
  }
}
int main(){
  string S;
  cin >> S;
  int K;
  cin >> K;
  int p = 0;
  vector<long long> ans = eval(S, p);
  cout << ans[K] << endl;
}
0