結果
問題 | No.324 落ちてた閉路グラフ |
ユーザー |
|
提出日時 | 2022-01-28 14:21:39 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 828 ms / 5,000 ms |
コード長 | 1,330 bytes |
コンパイル時間 | 442 ms |
コンパイル使用メモリ | 82,140 KB |
実行使用メモリ | 136,108 KB |
最終ジャッジ日時 | 2024-12-29 09:52:48 |
合計ジャッジ時間 | 8,310 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 34 |
ソースコード
n,m = map(int,input().split())w = list(map(int,input().split()))inf = -10 ** 8import sysif m <= 1:print(0)exit()"""dp1 = [[[inf] * 2 for _ in range(m+1)] for _ in range(n)]dp1[0][1][1] = 0dp2 = [[[inf] * 2 for _ in range(m+1)] for _ in range(n)]dp2[0][0][0] = 0for i in range(n-1):for j in range(m+1):dp1[i+1][j][0] = max(dp1[i][j][0],dp1[i][j][1])dp2[i+1][j][0] = max(dp2[i][j][0],dp2[i][j][1])if j > 0:dp1[i+1][j][1] = max(dp1[i][j-1][1] + w[i],dp1[i][j-1][0])dp2[i+1][j][1] = max(dp2[i][j-1][1] + w[i],dp2[i][j-1][0])dp1[-1][m][1] += w[-1]ans = max(dp1[-1][m][1],dp1[-1][m][0],dp2[-1][m][0],dp2[-1][m][1])print(ans)"""dp1 = [[inf] * 2 for _ in range(m+1)]dp2 = [[inf] * 2 for _ in range(m+1)]dp1[1][1] = 0dp2[0][0] = 0for i in range(n-1):n1 = [[inf] * 2 for _ in range(m+1)]n2 = [[inf] * 2 for _ in range(m+1)]for j in range(m+1):n1[j][0] = max(dp1[j][0],dp1[j][1])n2[j][0] = max(dp2[j][0],dp2[j][1])if j > 0:n1[j][1] = max(dp1[j-1][1] + w[i],dp1[j-1][0])n2[j][1] = max(dp2[j-1][1] + w[i],dp2[j-1][0])dp1 = n1dp2 = n2dp1[m][1] += w[-1]ans = max(dp1[m][0],dp1[m][1],dp2[m][0],dp2[m][1])print(ans)