結果

問題 No.1677 mæx
コンテスト
ユーザー umnc
提出日時 2021-09-22 17:26:36
言語 C++17(clang)
(17.0.6 + boost 1.87.0)
結果
AC  
実行時間 11 ms / 2,000 ms
コード長 1,862 bytes
コンパイル時間 4,891 ms
コンパイル使用メモリ 163,048 KB
実行使用メモリ 7,668 KB
最終ジャッジ日時 2024-07-05 08:16:09
合計ジャッジ時間 6,252 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
// #include <atcoder/all>
// using namespace atcoder;
#define rep(i, n) for (ll i = 0; i < (n); ++i)
#define rep1(i, n) for (ll i = 1; i <= n; ++i)
#define reps(i, s, e) for (ll i = s; i <= e; ++i)
#define all(v) v.begin(), v.end()
using ll = long long;
using ld = long double;
using cp = complex<ld>;
using pa = pair<ll, ll>;
using tup = tuple<ll, ll, ll>;
using vp = vector<pair<ll, ll>>;
using vtup = vector<tuple<ll, ll, ll>>;
using st = string;
using vs = vector<string>;
using vc = vector<char>;
using vvi = vector<vector<ll>>;
using vvc = vector<vector<char>>;
using vi = vector<ll>;
const ll MOD1 = 1000000007;
const ll MOD2 = 998244353;
const ll INF = (1LL << 60);

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)の場合
  st 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)] %= MOD2;
    }
  }

  // mex
  if (type & 2) {
    rep(i, 3) rep(j, 3) {
      res[mex[i][j]] += a[i] * b[j];
      res[mex[i][j]] %= MOD2;
    }
  }
  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;

  st s;
  ll k;
  cin >> s >> k;

  int i = 0;
  array<ll, 3> a = f(i, s);
  cout << a[k] << endl;
}
0