結果
問題 |
No.324 落ちてた閉路グラフ
|
ユーザー |
|
提出日時 | 2020-10-19 08:54:30 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 525 ms / 5,000 ms |
コード長 | 891 bytes |
コンパイル時間 | 196 ms |
コンパイル使用メモリ | 12,672 KB |
実行使用メモリ | 44,228 KB |
最終ジャッジ日時 | 2024-07-21 07:48:14 |
合計ジャッジ時間 | 19,027 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 34 |
ソースコード
import numpy as np N, M = map(int, input().split()) W = tuple(map(int, input().split())) MAX = 1_000_000 if M < 2: print(0) exit() def dp(last, non_last): for w in W[:-1]: new_last = np.full(M + 1, -MAX) new_last[1:] = np.maximum(last[:-1] + w, non_last[:-1]) new_non_last = np.maximum(last, non_last) last, non_last = new_last, new_non_last return last, non_last # dp[i] = iコの点を打ち終えている # 一番目の点を含む dp_last = np.full(M + 1, -MAX) dp_non_last = np.full(M + 1, -MAX) dp_last[1] = 0 dp_last, dp_non_last = dp(dp_last, dp_non_last) ans = max(dp_last[-1] + W[-1], dp_non_last[-1]) # 一番目の点を含まない dp_last = np.full(M + 1, -MAX) dp_non_last = np.full(M + 1, -MAX) dp_non_last[0] = 0 dp_last, dp_non_last = dp(dp_last, dp_non_last) ans = max(ans, dp_last[-1], dp_non_last[-1]) print(ans)