結果

問題 No.1861 Required Number
ユーザー Chippppp
提出日時 2022-03-05 14:37:36
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 713 ms / 2,500 ms
コード長 763 bytes
コンパイル時間 140 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 63,480 KB
最終ジャッジ日時 2024-07-19 19:21:12
合計ジャッジ時間 31,816 ms
ジャッジサーバーID
(参考情報)
judge4 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 42 RE * 4
権限があれば一括ダウンロードができます

ソースコード

diff #

def main():
    import sys
    readline = sys.stdin.readline
    import numpy as np

    N, K = map(int, readline().split())
    A = np.fromstring(readline(), dtype = np.int64, sep = ' ')

    dpl = np.zeros((N + 1, K + 1), dtype = bool)
    dpr = np.zeros((N + 1, K + 1), dtype = bool)
    dpl[0][0] = dpr[N][0] = True

    for i in range(N):
        dpl[i + 1] = dpl[i]
        dpl[i + 1][A[i]:] |= dpl[i][:K - A[i] + 1]
    for i in reversed(range(N)):
        dpr[i] = dpr[i + 1]
        dpr[i][A[i]:] |= dpr[i + 1][:K - A[i] + 1]
    if not dpl[N][K]:
        print(-1)
    else:
        ans = 0
        for i in range(N):
            if not np.any(dpl[i] & dpr[i + 1][::-1]):
                ans += 1
        print(ans)
if __name__ == '__main__':
    main()
0