結果
問題 | 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 npN, M = map(int, input().split())W = tuple(map(int, input().split()))MAX = 1_000_000if 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_lastreturn last, non_last# dp[i] = iコの点を打ち終えている# 一番目の点を含むdp_last = np.full(M + 1, -MAX)dp_non_last = np.full(M + 1, -MAX)dp_last[1] = 0dp_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] = 0dp_last, dp_non_last = dp(dp_last, dp_non_last)ans = max(ans, dp_last[-1], dp_non_last[-1])print(ans)