結果

問題 No.2378 Cards and Subsequences
コンテスト
ユーザー yamate11
提出日時 2023-06-13 20:59:14
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 930 bytes
コンパイル時間 259 ms
コンパイル使用メモリ 82,040 KB
実行使用メモリ 95,920 KB
最終ジャッジ日時 2024-07-21 16:26:11
合計ジャッジ時間 13,130 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 6
other AC * 4 TLE * 3 -- * 28
権限があれば一括ダウンロードができます

ソースコード

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

        def op(a):
            debug("  called", a, v)
            if a <= K:
                acc[i + 1][a] = (acc[i + 1][a] + v) % primeB
        
        op(k + A[S[i]])
        op(k + B[S[i]])
    debug(acc[i + 1])

print(acc[N][K])
0