結果

問題 No.1861 Required Number
ユーザー shiomusubi496
提出日時 2022-03-04 16:07:50
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
RE  
実行時間 -
コード長 752 bytes
コンパイル時間 168 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 48,472 KB
最終ジャッジ日時 2024-07-18 22:26:03
合計ジャッジ時間 24,807 ms
ジャッジサーバーID
(参考情報)
judge4 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 14 RE * 32
権限があれば一括ダウンロードができます

ソースコード

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][:-A[i]]
    for i in reversed(range(N)):
        dpr[i] = dpr[i + 1]
        dpr[i][A[i]:] |= dpr[i + 1][:-A[i]]
    if not dpl[-1][-1]:
        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