結果
問題 |
No.1247 ブロック登り
|
ユーザー |
|
提出日時 | 2022-07-06 22:46:00 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 595 ms / 3,000 ms |
コード長 | 2,497 bytes |
コンパイル時間 | 1,997 ms |
コンパイル使用メモリ | 197,188 KB |
最終ジャッジ日時 | 2025-01-30 05:07:09 |
ジャッジサーバーID (参考情報) |
judge4 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 26 |
ソースコード
#include<bits/stdc++.h> //#include<atcoder/maxflow> using namespace std; //using namespace atcoder; const int inf = 1 << 30; void solve(){ int n, K; cin >> n >> K; vector<int> A(n); for(int i = 0; i < n; i++) cin >> A[i]; vector<int> dp(n * n * (K + 1) * 2, -inf); auto f = [&](int i, int j, int k, int l){ return ((i * n + j) * (K + 1) + k) * 2 + l; }; for(int i = 0; i < n; i++){ dp[f(i, i, K, 0)] = A[i] * K; dp[f(i, i, K, 1)] = A[i] * K; } for(int x = K; x > 0; x--){ for(int l = 0; l < n; l++){ for(int r = l; r < n; r++){ int p = f(l, r, x, 0); if(l != 0){ int p2 = f(l - 1, r, x - 1, 0); dp[p2] = max(dp[p2], dp[p] + A[l - 1] * (x - 1)); } if(l != r){ if(x >= 2){ int p2 = f(l, r, x - 2, 0); dp[p2] = max(dp[p2], dp[p]); } if(x - (r - l) >= 0){ int p2 = f(l, r, x - (r - l), 1); dp[p2] = max(dp[p2], dp[p]); } } p = f(l, r, x, 1); if(r != n - 1){ int p2 = f(l, r + 1, x - 1, 1); dp[p2] = max(dp[p2], dp[p] + A[r + 1] * (x - 1)); } if(l != r){ if(x >= 2){ int p2 = f(l, r, x - 2, 1); dp[p2] = max(dp[p2], dp[p]); } if(x - (r - l) >= 0){ int p2 = f(l, r, x - (r - l), 0); dp[p2] = max(dp[p2], dp[p]); } } } } } for(int i = 0; i < n; i++){ int ans = -inf; for(int l = 0; l < min(i + 2, n); l++){ for(int r = max(0, i - 1); r < n; r++){ if(l > r) continue; if(abs(r - i) <= K){ int p = f(l, r, abs(r - i), 1); ans = max(ans, dp[p]); } if(abs(i - l) <= K){ int p = f(l, r, abs(i - l), 0); ans = max(ans, dp[p]); } } } cout << ans << "\n"; } } int main(){ cin.tie(0)->sync_with_stdio(0); int t; t = 1; //cin >> t; while(t--) solve(); }