結果
問題 | No.1677 mæx |
ユーザー | 沙耶花 |
提出日時 | 2021-09-10 22:02:33 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 26 ms / 2,000 ms |
コード長 | 1,446 bytes |
コンパイル時間 | 4,088 ms |
コンパイル使用メモリ | 251,980 KB |
最終ジャッジ日時 | 2025-01-24 10:48:21 |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 18 |
ソースコード
#include <stdio.h> #include <bits/stdc++.h> #include <atcoder/all> using namespace atcoder; using mint = modint998244353; using namespace std; #define rep(i,n) for (int i = 0; i < (n); ++i) #define Inf 1000000000000000000 using Data = array<mint,3>; string S; int N; Data MAX(Data a,Data b){ Data ret; rep(i,3)ret[i] = 0; rep(i,3){ rep(j,3){ ret[max(i,j)] += a[i] * b[j]; } } return ret; } Data MEX(Data a,Data b){ Data ret; rep(i,3)ret[i] = 0; rep(i,3){ rep(j,3){ if(i!=0&&j!=0){ ret[0] += a[i]*b[j]; } else if(i!=1&&j!=1){ ret[1] += a[i]*b[j]; } else{ ret[2] += a[i]*b[j]; } } } return ret; } Data eq(int &pos){ char c; if(S[pos]=='m'){ c = S[pos+1]; pos += 3; pos++; Data x = eq(pos); pos++; Data y = eq(pos); pos++; Data ret; rep(i,3){ ret[i] = 0; } /* cout<<c<<endl; rep(i,3){ cout<<x[i].val()<<','; } cout<<endl; rep(i,3){ cout<<y[i].val()<<','; } cout<<endl; */ if(c!='e'){ Data temp = MAX(x,y); rep(i,3){ ret[i] += temp[i]; } } if(c!='a'){ Data temp = MEX(x,y); rep(i,3){ ret[i] += temp[i]; } } return ret; } else{ c = S[pos]; pos++; Data ret; if(c=='?'){ ret = {1,1,1}; } else{ rep(i,3)ret[i] = 0; ret[c-'0']++; } return ret; } } int main(){ cin>>S; int K; cin>>K; N = S.size(); int pos = 0; Data ans = eq(pos); cout<<ans[K].val()<<endl; return 0; }