結果

問題 No.1677 mæx
ユーザー 👑 SPD_9X2
提出日時 2021-09-10 22:07:59
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 280 ms / 2,000 ms
コード長 1,645 bytes
コンパイル時間 305 ms
コンパイル使用メモリ 82,368 KB
実行使用メモリ 121,300 KB
最終ジャッジ日時 2024-06-12 00:23:24
合計ジャッジ時間 4,830 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

"""

構文解析だな
カンマを見付けるのがむずい

まず、
m?x( , )
の、mと,を対応付ける前処理をする。

mが来たらpush
,が来たらstackに入れる

"""

import sys
from sys import stdin
import heapq
sys.setrecursionlimit(200000)

mod = 998244353

def solve(l,r):

    # (l,r)
    #print (type(l))

    if S[l] == "0":
        return (1,0,0)
    elif S[l] == "1":
        return (0,1,0)
    elif S[l] == "2":
        return (0,0,1)
    elif S[l] == "?":
        return (1,1,1)
    else:

        #まず、カンマを見付ける

        ltup = solve(l+4,MtoC[l])
        rtup = solve(MtoC[l]+1 , r-1)

        ans = [0,0,0]

        if S[l+1] == "a" or S[l+1] == "?":
            for li in range(3):
                for ri in range(3):
                    ans[max(li,ri)] += ltup[li] * rtup[ri]

        if S[l+1] == "e" or S[l+1] == "?":
            for li in range(3):
                for ri in range(3):

                    if li*ri == 0: #0がある
                        if li+ri == 1:
                            mex = 2
                        else:
                            mex = 1
                    else:
                        mex = 0
                        
                        
                    ans[mex] += ltup[li] * rtup[ri]
        for i in range(3):
            ans[i] %= mod

        return ans
        
    

S = stdin.readline()[:-1]
K = int(stdin.readline())

stk = []
MtoC = {}
for i in range(len(S)):
    if S[i] == "m":
        stk.append(i)
    elif S[i] == ",":
        MtoC[stk[-1]] = i
        del stk[-1]

#print (MtoC)

ans = solve(0,len(S))

print (ans[K])
0