結果

問題 No.1861 Required Number
ユーザー tamato
提出日時 2022-03-04 21:41:22
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 592 ms / 2,500 ms
コード長 1,111 bytes
コンパイル時間 284 ms
コンパイル使用メモリ 82,396 KB
実行使用メモリ 234,356 KB
最終ジャッジ日時 2024-07-18 22:37:50
合計ジャッジ時間 13,068 ms
ジャッジサーバーID
(参考情報)
judge5 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 42 MLE * 4
権限があれば一括ダウンロードができます

ソースコード

diff #

mod = 1000000007
eps = 10**-9


def main():
    import sys
    input = sys.stdin.readline

    N, K = map(int, input().split())
    A = list(map(int, input().split()))

    dp = [[0] * (K+1) for _ in range(N+1)]
    dp[0][0] = 1
    for i, a in enumerate(A):
        for k in range(K + 1):
            dp[i + 1][k] = (dp[i + 1][k] + dp[i][k])%mod
            if k + a <= K:
                dp[i + 1][k + a] = (dp[i + 1][k + a] + dp[i][k])%mod

    if dp[-1][-1] == 0:
        print(-1)
    else:
        dp2 = [[0] * (K + 1) for _ in range(N + 1)]
        dp2[0][0] = 1
        for i, a in enumerate(A[::-1]):
            for k in range(K + 1):
                dp2[i + 1][k] = (dp2[i + 1][k] + dp2[i][k]) % mod
                if k + a <= K:
                    dp2[i + 1][k + a] = (dp2[i + 1][k + a] + dp2[i][k]) % mod

        ans = 0
        for i in range(N):
            ok = 1
            for k in range(K + 1):
                if dp[i][k] * dp2[N - i - 1][K - k] != 0:
                    ok = 0
                    break
            ans += ok

        print(ans)


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