結果

問題 No.1677 mæx
コンテスト
ユーザー monnu
提出日時 2021-09-16 21:49:05
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 12 ms / 2,000 ms
コード長 1,134 bytes
コンパイル時間 3,680 ms
コンパイル使用メモリ 231,004 KB
実行使用メモリ 7,804 KB
最終ジャッジ日時 2024-06-29 17:15:03
合計ジャッジ時間 5,018 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
#include <atcoder/all>
using namespace atcoder;
using ll=long long;
using Graph=vector<vector<int>>;
#define MAX 2000005
//#define MOD 1000000007
#define MOD 998244353
#define INF 1000000000
//#define INF 1000000000000000000

int mex(int a,int b){
  int ret=0;
  while(ret==a||ret==b){
    ++ret;
  }
  return ret;
}

vector<ll> dfs(string &S,int &i){
  if(S[i]=='0'){
    ++i;
    return {1,0,0};
  }else if(S[i]=='1'){
    ++i;
    return {0,1,0};
  }else if(S[i]=='2'){
    ++i;
    return {0,0,1};
  }else if(S[i]=='?'){
    ++i;
    return {1,1,1};
  }

  char c=S[i+1];
  i+=4;
  vector<ll> a=dfs(S,i);
  assert(S[i]==',');
  ++i;
  vector<ll> b=dfs(S,i);
  ++i;

  vector<ll> ret(3,0);
  for(int j=0;j<3;j++){
    for(int k=0;k<3;k++){
      if(c=='a'||c=='?'){
        ret[max(j,k)]+=a[j]*b[k];
        ret[max(j,k)]%=MOD;
      }
      if(c=='e'||c=='?'){
        ret[mex(j,k)]+=a[j]*b[k];
        ret[mex(j,k)]%=MOD;
      }
    }
  }
  return ret;
}

int main(){
  string S;
  int K;
  cin>>S>>K;
  int N=S.size();
  int i=0;
  vector<ll> ans=dfs(S,i);
  cout<<ans[K]<<'\n';
}
0