結果

問題 No.1677 mæx
ユーザー titiatitia
提出日時 2021-09-10 23:34:16
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
TLE  
実行時間 -
コード長 2,048 bytes
コンパイル時間 77 ms
コンパイル使用メモリ 12,928 KB
実行使用メモリ 16,384 KB
最終ジャッジ日時 2024-06-12 04:59:15
合計ジャッジ時間 3,788 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 31 ms
16,384 KB
testcase_01 AC 31 ms
11,008 KB
testcase_02 AC 28 ms
10,880 KB
testcase_03 AC 26 ms
11,008 KB
testcase_04 TLE -
testcase_05 -- -
testcase_06 -- -
testcase_07 -- -
testcase_08 -- -
testcase_09 -- -
testcase_10 -- -
testcase_11 -- -
testcase_12 -- -
testcase_13 -- -
testcase_14 -- -
testcase_15 -- -
testcase_16 -- -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
testcase_20 -- -
testcase_21 -- -
権限があれば一括ダウンロードができます

ソースコード

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