結果
問題 |
No.1861 Required Number
|
ユーザー |
|
提出日時 | 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 |
ソースコード
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()