結果

問題 No.2378 Cards and Subsequences
ユーザー poyonpoyon
提出日時 2023-06-20 19:09:21
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 180 ms / 2,000 ms
コード長 1,161 bytes
コンパイル時間 149 ms
コンパイル使用メモリ 82,668 KB
実行使用メモリ 108,032 KB
最終ジャッジ日時 2024-07-21 16:27:09
合計ジャッジ時間 6,122 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 6
other AC * 35
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys

input = sys.stdin.readline

MOD = 998244353


def main():
    N, M, K = map(int, input().split())
    S = list(map(int, input().split()))
    A = list(map(int, input().split()))
    B = list(map(int, input().split()))

    # 部分列 dp (累積和で O(σ + N) に高速化した版)
    # dp[i][k] := 整数列 (S[1], ... , S[i]) から作れる部分列のうち、
    #             総和が k である部分列の種類数（空列を含む）
    #             (1-based indexing)
    # idx[s] := 種類 s のカードが最後に登場した位置
    #           未登場なら dummy の N + 1 で常に dp[N + 1] = 0 (場合分けを無くすテク)
    dp = [[0] * (K + 1) for _ in range(N + 2)]
    idx = [N + 1] * M
    dp[0][0] = 1
    for i in range(N):
        s = S[i] - 1
        j = idx[s]
        for k in range(K + 1):
            dp[i + 1][k] += dp[i][k]
            dp[i + 1][k] %= MOD
            if k + A[s] <= K: dp[i + 1][k + A[s]] += dp[i][k] - dp[j][k]
            if k + B[s] <= K: dp[i + 1][k + B[s]] += dp[i][k] - dp[j][k]
        idx[s] = i
    ans = dp[N][K]
    print(ans)


if __name__ == "__main__":
    main()
0