結果

問題 No.1677 mæx
ユーザー stoqstoq
提出日時 2021-08-02 04:17:24
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 12 ms / 2,000 ms
コード長 1,321 bytes
コンパイル時間 2,203 ms
コンパイル使用メモリ 199,864 KB
実行使用メモリ 8,428 KB
最終ジャッジ日時 2023-09-02 15:02:13
合計ジャッジ時間 4,080 ms
ジャッジサーバーID
（参考情報） 		judge11 / judge13
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
4,384 KB
testcase_01 AC 1 ms
4,376 KB
testcase_02 AC 1 ms
4,376 KB
testcase_03 AC 1 ms
4,384 KB
testcase_04 AC 10 ms
6,736 KB
testcase_05 AC 10 ms
6,684 KB
testcase_06 AC 10 ms
6,712 KB
testcase_07 AC 11 ms
6,868 KB
testcase_08 AC 10 ms
6,688 KB
testcase_09 AC 11 ms
6,752 KB
testcase_10 AC 10 ms
6,752 KB
testcase_11 AC 10 ms
6,712 KB
testcase_12 AC 10 ms
6,684 KB
testcase_13 AC 10 ms
6,764 KB
testcase_14 AC 11 ms
6,824 KB
testcase_15 AC 11 ms
7,012 KB
testcase_16 AC 11 ms
6,776 KB
testcase_17 AC 10 ms
6,760 KB
testcase_18 AC 10 ms
6,792 KB
testcase_19 AC 2 ms
4,380 KB
testcase_20 AC 2 ms
4,380 KB
testcase_21 AC 12 ms
8,428 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
#define rep(i, n) for (int i = 0; i < int(n); i++)
constexpr ll MOD = 998244353;

int mex[3][3];

array<ll, 3> f(int &i, string &s) {
  // 0, 1, 2 の場合
  if (s[i] != 'm') {
    array<ll, 3> res = {};
    if (isdigit(s[i]))
      res[s[i] - '0'] = 1;
    else
      res[0] = res[1] = res[2] = 1;
    i++;
    return res;
  }

  // m?x(A,B) の場合
  string m = s.substr(i, 3);
  i += 3;    // "m?x"
  int type;  // 1:max 2:mex 3:m?x
  if (m == "max")
    type = 1;
  else if (m == "mex")
    type = 2;
  else
    type = 3;
  i++;  // '('
  array<ll, 3> a = f(i, s);
  i++;  // ','
  array<ll, 3> b = f(i, s);
  i++;  // ')'

  array<ll, 3> res = {};
  // max
  if (type & 1) {
    rep(i, 3) rep(j, 3) {
      res[max(i, j)] += a[i] * b[j];
      res[max(i, j)] %= MOD;
    }
  }
  // mex
  if (type & 2) {
    rep(i, 3) rep(j, 3) {
      res[mex[i][j]] += a[i] * b[j];
      res[mex[i][j]] %= MOD;
    }
  }
  return res;
}

int main() {
  mex[0][0] = 1, mex[0][1] = 2, mex[0][2] = 1;
  mex[1][0] = 2, mex[1][1] = 0, mex[1][2] = 0;
  mex[2][0] = 1, mex[2][1] = 0, mex[2][2] = 0;
  string s;
  int k;
  cin >> s >> k;
  int i = 0;
  array<ll, 3> a = f(i, s);
  cout << a[k] << "\n";
}
0