結果
問題 | No.1247 ブロック登り |
ユーザー |
![]() |
提出日時 | 2020-10-03 00:07:51 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 598 ms / 3,000 ms |
コード長 | 2,066 bytes |
コンパイル時間 | 855 ms |
コンパイル使用メモリ | 98,460 KB |
実行使用メモリ | 433,628 KB |
最終ジャッジ日時 | 2024-07-18 00:42:04 |
合計ジャッジ時間 | 7,682 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 26 |
ソースコード
#include <iostream>#include <algorithm>#include <string>#include <vector>#include <cmath>#include <map>#include <queue>#include <iomanip>#include <set>#include <tuple>#define mkp make_pair#define mkt make_tuple#define rep(i,n) for(int i = 0; i < (n); ++i)#define all(v) v.begin(),v.end()using namespace std;typedef long long ll;const ll MOD=1e9+7;template<class T> void chmin(T &a,const T &b){if(a>b) a=b;}template<class T> void chmax(T &a,const T &b){if(a<b) a=b;}const ll INF=1e18;ll dp[303][303][303][2];int main(){cin.tie(0);ios::sync_with_stdio(false);int N,K;cin>>N>>K;vector<ll> A(N);rep(i,N) cin>>A[i];//vector<vector<vector<ll>>> dp(N,vector<vector<ll>> (N,vector<ll> (2,-INF)));rep(i,K) rep(j,N) rep(k,N) rep(m,2) dp[i][j][k][m]=-INF;rep(i,N){dp[K-1][i][i][0]=A[i]*K;dp[K-1][i][i][1]=A[i]*K;}if(K==1){rep(i,N){ll ans=-INF;if(i-1>=0) chmax(ans,A[i-1]);if(i+1<N) chmax(ans,A[i+1]);cout<<ans<<"\n";}return 0;}for(int i=K-1;i>=1;i--){for(int j=0;j<N;j++) for(int k=j;k<N;k++) for(int m=0;m<2;m++){if(dp[i][j][k][m]==-INF) continue;ll val=dp[i][j][k][m];if(m==0){if(j-1>=0) chmax(dp[i-1][j-1][k][0],val+i*A[j-1]);if(i-(k-j)>=0)chmax(dp[i-(k-j)][j][k][1],val);if(i-2>=0&&j<k) chmax(dp[i-2][j][k][m],val);}else{if(k+1<N) chmax(dp[i-1][j][k+1][1],val+i*A[k+1]);if(i-(k-j)>=0)chmax(dp[i-(k-j)][j][k][0],val);if(i-2>=0&&j<k) chmax(dp[i-2][j][k][m],val);}}}vector<ll> ans(N,-INF);for(int i=0;i<=K-2;i++) for(int j=0;j<N;j++) for(int k=j;k<N;k++) for(int m=0;m<2;m++){ll val=dp[i][j][k][m];if(val==-INF) continue;if(m==1){if(k-i>=j){if(k-i+1<N) chmax(ans[k-i+1],val);if(k-i-1>=0) chmax(ans[k-i-1],val);}}else{if(j+i<=k){if(j+i+1<N) chmax(ans[j+i+1],val);if(j+i-1>=0) chmax(ans[j+i-1],val);}}}rep(i,N) cout<<ans[i]<<"\n";return 0;}