結果

問題 No.2783 4-33 Easy
ユーザー flippergo
提出日時 2024-09-19 16:08:37
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
TLE  
実行時間 -
コード長 1,052 bytes
コンパイル時間 150 ms
コンパイル使用メモリ 12,928 KB
実行使用メモリ 87,460 KB
最終ジャッジ日時 2024-09-19 16:08:44
合計ジャッジ時間 6,065 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 2 TLE * 1 -- * 23
権限があれば一括ダウンロードができます

ソースコード

diff # 

MOD = 998244353
N = int(input())
A1 = list(map(int,input().split()))
B1 = list(input().split())
A = []
B = []
C = []
NUM = [str(i) for i in range(33+1)]
for i in range(N):
    if B1[i] in NUM:
        A.append(A1[i])
        B.append(int(B1[i]))
    elif B1[i]=="X":
        C.append(A1[i])
N = len(A)
A = [0]+A
B = [0]+B
dp = [[[[0 for _ in range(33+1)] for _ in range(4+1)] for _ in range(9+1)] for _ in range(N+1)]
dp[1][1][A[1]][B[1]] = 1
for i in range(2,N+1):
    for j in range(1,8+1):
        for a in range(4+1):
            for b in range(33+1):
                dp[i][j][a][b] = dp[i-1][j][a][b]
                if j==1 and a==A[i] and b==B[i]:
                    dp[i][j][a][b] = (dp[i][j][a][b]+1)%MOD
                elif j>=2 and a>=A[i] and b>=B[i]:
                    dp[i][j][a][b] = (dp[i][j][a][b]+dp[i-1][j-1][a-A[i]][b-B[i]])%MOD
for k in range(len(C)):
    for a in range(4+1):
        for b in range(33+1):
            if a>=C[k]:
                dp[N][9][a][b] = (dp[N][9][a][b]+dp[N][8][a-C[k]][b])%MOD
print(dp[N][9][4][33])
0