結果

問題 No.1677 mæx
ユーザー titiatitia
提出日時 2022-10-04 04:42:44
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
AC  
実行時間 664 ms / 2,000 ms
コード長 3,351 bytes
コンパイル時間 476 ms
コンパイル使用メモリ 11,632 KB
実行使用メモリ 74,868 KB
最終ジャッジ日時 2023-08-28 10:28:48
合計ジャッジ時間 10,889 ms
ジャッジサーバーID
（参考情報） 		judge11 / judge13
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 22 ms
9,324 KB
testcase_01 AC 22 ms
9,288 KB
testcase_02 AC 23 ms
9,340 KB
testcase_03 AC 18 ms
8,812 KB
testcase_04 AC 554 ms
67,748 KB
testcase_05 AC 549 ms
67,856 KB
testcase_06 AC 549 ms
67,612 KB
testcase_07 AC 554 ms
67,824 KB
testcase_08 AC 559 ms
67,896 KB
testcase_09 AC 549 ms
67,900 KB
testcase_10 AC 559 ms
67,768 KB
testcase_11 AC 561 ms
68,036 KB
testcase_12 AC 554 ms
67,740 KB
testcase_13 AC 552 ms
67,712 KB
testcase_14 AC 664 ms
68,792 KB
testcase_15 AC 662 ms
68,824 KB
testcase_16 AC 661 ms
68,484 KB
testcase_17 AC 653 ms
68,592 KB
testcase_18 AC 660 ms
68,716 KB
testcase_19 AC 18 ms
8,896 KB
testcase_20 AC 17 ms
8,904 KB
testcase_21 AC 658 ms
74,868 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

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

mod=998244353

if len(S)==1:
    if S=="?":
        print(1)
    else:
        x=int(S)

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

    exit()

now=0
X=[[-1,-1] for i in range(len(S))]

IND=[]
for i in range(len(S)):
    if S[i]=="m":
        IND.append(i)
    elif S[i]==",":
        X[IND[-1]][0]=i
    elif S[i]==")":
        X[IND[-1]][1]=i
        IND.pop()

from functools import lru_cache
@lru_cache(maxsize=None)
def calc(fr,to,x):
    #print(fr,to,x)
    if fr==to:
        if S[fr]=="?":
            return 1
        y=int(S[fr])

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

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

print(calc(0,X[0][1],K)%mod)
0