結果

問題 No.1677 mæx
ユーザー titia
提出日時 2021-09-10 23:34:30
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 2,048 bytes
コンパイル時間 662 ms
コンパイル使用メモリ 82,172 KB
実行使用メモリ 307,980 KB
最終ジャッジ日時 2024-06-12 05:00:11
合計ジャッジ時間 4,519 ms
ジャッジサーバーID
(参考情報)
judge5 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other TLE * 1 -- * 17
権限があれば一括ダウンロードができます

ソースコード

diff #

import sys
input = sys.stdin.readline
sys.setrecursionlimit(2*10**5)

S=input().strip()
K=int(input())

mod=998244353

if len(S)==1:
    x=int(S)

    if K==x:
        print(1)
    else:
        print(0)

    exit()


from functools import lru_cache
@lru_cache(maxsize=None)
def calc(S,x):
    #print(S,x)

    if len(S)==1:
        if S=="?":
            return 1
        y=int(S)

        if y==x:
            return 1
        else:
            return 0

    now=0
    for i in range(4,len(S)):
        if S[i]=="(":
            now+=1
        elif S[i]==")":
            now-=1
        elif S[i]=="," and now==0:
            kom=i
            break

    T=S[4:kom]
    U=S[kom+1:-1]
        
    if S[0:3]=="max":
        if x==0:
            return calc(T,0)*calc(U,0)%mod
        elif x==1:
            return (calc(T,0)*calc(U,1)%mod+calc(T,1)*calc(U,0)%mod+calc(T,1)*calc(U,1)%mod)%mod
        else:
            return (calc(T,0)*calc(U,2)%mod+calc(T,1)*calc(U,2)%mod+calc(T,2)*calc(U,0)%mod+calc(T,2)*calc(U,1)%mod+calc(T,2)*calc(U,2)%mod)%mod
    elif S[0:3]=="mex":
        if x==0:
            return (calc(T,1)*calc(U,1)%mod + calc(T,1)*calc(U,2)%mod +calc(T,2)*calc(U,1)%mod + calc(T,2)*calc(U,2)%mod )%mod
        elif x==1:
            return (calc(T,0)*calc(U,0)%mod + calc(T,0)*calc(U,2)%mod + calc(T,2)*calc(U,0)%mod)%mod
        else:
            return (calc(T,0)*calc(U,1)%mod+calc(T,1)*calc(U,0)%mod)%mod
    else:
        if x==0:
            return (calc(T,0)*calc(U,0)%mod +calc(T,1)*calc(U,1)%mod + calc(T,1)*calc(U,2)%mod +calc(T,2)*calc(U,1)%mod + calc(T,2)*calc(U,2)%mod )%mod
        elif x==1:
            return ((calc(T,0)*calc(U,1)%mod+calc(T,1)*calc(U,0)%mod+calc(T,1)*calc(U,1)%mod)%mod + calc(T,0)*calc(U,0)%mod + calc(T,0)*calc(U,2)%mod + calc(T,2)*calc(U,0)%mod)%mod
        else:
            return ((calc(T,0)*calc(U,2)%mod+calc(T,1)*calc(U,2)%mod+calc(T,2)*calc(U,0)%mod+calc(T,2)*calc(U,1)%mod+calc(T,2)*calc(U,2)%mod)%mod + (calc(T,0)*calc(U,1)%mod+calc(T,1)*calc(U,0)%mod)%mod)%mod

print(calc(S,K)%mod)
0