結果

問題 No.1677 mæx
コンテスト
ユーザー googol_S0
提出日時 2021-09-10 21:44:08
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,149 ms / 2,000 ms
コード長 1,214 bytes
コンパイル時間 229 ms
コンパイル使用メモリ 82,220 KB
実行使用メモリ 89,760 KB
最終ジャッジ日時 2024-06-11 23:10:50
合計ジャッジ時間 19,729 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

S=input()
K=int(input())
mod=998244353
def mAx(a,b):
  r=[0]*3
  for i in range(3):
    for j in range(3):
      r[max(i,j)]=(r[max(i,j)]+a[i]*b[j])%mod
  return tuple(r)

def mex(a,b):
  if min(a,b)>0:
    return 0
  elif a==0 and b==0:
    return 1
  elif max(a,b)<2:
    return 2
  else:
    return 1

MEX=[[mex(i,j) for j in range(3)]for i in range(3)]

def mEx(a,b):
  r=[0]*3
  for i in range(3):
    for j in range(3):
      r[MEX[i][j]]=(r[MEX[i][j]]+a[i]*b[j])%mod
  return tuple(r)

def m_x(a,b):
  r=[0]*3
  for i in range(3):
    for j in range(3):
      r[MEX[i][j]]=(r[MEX[i][j]]+a[i]*b[j])%mod
      r[max(i,j)]=(r[max(i,j)]+a[i]*b[j])%mod
  return tuple(r)

Q=[]
N=len(S)
for i in range(N):
  if S[i]=='m':
    Q.append(['m'])
  elif S[i]=='?':
    if Q[-1]==['m']:
      Q[-1].append('_')
    else:
      Q.append((1,1,1))
  elif S[i]=='a' or S[i]=='e':
    Q[-1].append(chr(ord(S[i])-32))
  elif S[i]=='x':
    Q[-1].append('x')
    Q[-1]=''.join(Q[-1])
  elif S[i]=='(':
    continue
  elif S[i]==')':
    Q[-3]=eval(Q[-3]+'('+str(Q[-2])+','+str(Q[-1])+')')
    del Q[-1]
    del Q[-1]
  elif S[i]==',':
    continue
  else:
    x=[0,0,0]
    x[int(S[i])]=1
    Q.append(tuple(x))
print(Q[0][K])
0