結果

問題 No.1677 mæx
コンテスト
ユーザー titia
提出日時 2021-09-10 23:34:16
言語 Python3
(3.14.2 + numpy 2.4.0 + scipy 1.16.3)
結果
TLE  
実行時間 -
コード長 2,048 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 77 ms
コンパイル使用メモリ 12,928 KB
実行使用メモリ 16,384 KB
最終ジャッジ日時 2024-06-12 04:59:15
合計ジャッジ時間 3,788 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other TLE * 1 -- * 17
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

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