結果

問題 No.2378 Cards and Subsequences
ユーザー yamate11
提出日時 2023-06-13 21:05:15
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 600 ms / 2,000 ms
コード長 941 bytes
コンパイル時間 205 ms
コンパイル使用メモリ 81,920 KB
実行使用メモリ 140,708 KB
最終ジャッジ日時 2024-07-21 16:26:46
合計ジャッジ時間 17,811 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 6
other AC * 35
権限があれば一括ダウンロードができます

ソースコード

diff # 

#!/usr/bin/env pypy3

import sys

def debug(*msg):
    print(*msg, file=sys.stderr)

def intdec(x):
    return int(x) - 1

N, M, K = map(int, input().strip().split())
S = list(map(intdec, input().strip().split()))
A = list(map(int, input().strip().split()))
B = list(map(int, input().strip().split()))

rec = [-1 for _ in range(M + 1)]
prev = [0 for _ in range(N)]
for i in range(N):
    prev[i] = rec[S[i]]
    rec[S[i]] = i
debug(prev)

primeB = 998244353

acc = [[0 for k in range(K + 1)] for n in range(N + 1)]
acc[0][0] = 1
for i in range(N):
    acc[i + 1] = acc[i][:]
    for k in range(K + 1):
        exc = 0 if prev[i] < 0 else acc[prev[i]][k]
        v = (acc[i][k] - exc) % primeB
        a1 = k + A[S[i]]
        if a1 <= K:
            acc[i + 1][a1] = (acc[i + 1][a1] + v) % primeB
        b1 = k + B[S[i]]
        if b1 <= K:
            acc[i + 1][b1] = (acc[i + 1][b1] + v) % primeB
    debug(acc[i + 1])

print(acc[N][K])
0