結果

問題 No.1247 ブロック登り
ユーザー beetbeet
提出日時 2020-10-02 23:30:45
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 2,626 ms / 3,000 ms
コード長 3,046 bytes
コンパイル時間 2,188 ms
コンパイル使用メモリ 211,180 KB
実行使用メモリ 434,476 KB
最終ジャッジ日時 2024-07-18 00:05:24
合計ジャッジ時間 13,772 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
9,700 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 2 ms
5,376 KB
testcase_04 AC 2 ms
5,376 KB
testcase_05 AC 2 ms
5,376 KB
testcase_06 AC 2 ms
5,376 KB
testcase_07 AC 21 ms
17,280 KB
testcase_08 AC 24 ms
19,072 KB
testcase_09 AC 11 ms
10,880 KB
testcase_10 AC 11 ms
11,264 KB
testcase_11 AC 2,626 ms
434,144 KB
testcase_12 AC 2,308 ms
414,264 KB
testcase_13 AC 2,480 ms
434,256 KB
testcase_14 AC 2,536 ms
434,264 KB
testcase_15 AC 2,524 ms
434,400 KB
testcase_16 AC 2,451 ms
434,320 KB
testcase_17 AC 2,446 ms
434,336 KB
testcase_18 AC 2,197 ms
409,344 KB
testcase_19 AC 117 ms
71,372 KB
testcase_20 AC 26 ms
30,320 KB
testcase_21 AC 5 ms
7,160 KB
testcase_22 AC 100 ms
234,684 KB
testcase_23 AC 87 ms
230,660 KB
testcase_24 AC 42 ms
206,188 KB
testcase_25 AC 2,507 ms
434,476 KB
testcase_26 AC 2,503 ms
434,208 KB
testcase_27 AC 2,559 ms
434,260 KB
testcase_28 AC 2,442 ms
434,428 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

using Int = long long;
const char newl = '\n';

template<typename T1,typename T2> inline void chmin(T1 &a,T2 b){if(a>b) a=b;}
template<typename T1,typename T2> inline void chmax(T1 &a,T2 b){if(a<b) a=b;}
template<typename T> void drop(const T &x){cout<<x<<endl;exit(0);}
template<typename T=Int>
vector<T> read(size_t n){
  vector<T> ts(n);
  for(size_t i=0;i<n;i++) cin>>ts[i];
  return ts;
}

template <typename E>
struct SegmentTree{
  using H = function<E(E,E)>;
  Int n,height;
  H h;
  E ei;
  vector<E> laz;

  SegmentTree(H h,E ei):h(h),ei(ei){}

  void init(Int n_){
    n=1;height=0;
    while(n<n_) n<<=1,height++;
    laz.assign(2*n,ei);
  }

  inline void propagate(Int k){
    if(laz[k]==ei) return;
    laz[(k<<1)|0]=h(laz[(k<<1)|0],laz[k]);
    laz[(k<<1)|1]=h(laz[(k<<1)|1],laz[k]);
    laz[k]=ei;
  }

  inline void thrust(Int k){
    for(Int i=height;i;i--) propagate(k>>i);
  }

  void update(Int a,Int b,E x){
    if(a>=b) return;
    thrust(a+=n);
    thrust(b+=n-1);
    for(Int l=a,r=b+1;l<r;l>>=1,r>>=1){
      if(l&1) laz[l]=h(laz[l],x),l++;
      if(r&1) --r,laz[r]=h(laz[r],x);
    }
  }

  E get_val(Int a){
    thrust(a+=n);
    return laz[a];
  }

  void set_val(Int a,E x){
    thrust(a+=n);
    laz[a]=x;
  }
};

//INSERT ABOVE HERE
const Int MAX = 303;
const Int INF = 1e18;
Int dp[MAX][2][MAX][MAX];

signed main(){
  cin.tie(0);
  ios::sync_with_stdio(0);

  Int n,k;
  cin>>n>>k;
  auto as=read(n);

  vector<Int> ans(n);
  // start from parity
  for(Int p=0;p<2;p++){

    // init
    for(Int i=0;i<=k;i++)
      for(Int l=0;l<n;l++)
        for(Int r=0;r<n;r++)
          dp[i][0][l][r]=dp[i][1][l][r]=-INF;

    // [l, r]
    for(Int l=0;l+1<n;l++){
      Int r=l+1;
      if((abs(r-k)&1)==p) chmax(dp[k-1][0][l][r],as[l]*(k-1)+as[r]*(k-0));
      if((abs(l-k)&1)==p) chmax(dp[k-1][1][l][r],as[l]*(k-0)+as[r]*(k-1));
    }

    auto h=[&](Int a,Int b){return max(a,b);};
    SegmentTree<Int> seg(h,-INF);
    seg.init(n);

    for(Int i=k-1;i>=0;i--){
      for(Int l=0;l<n;l++){
        for(Int r=l+1;r<n;r++){
          // [l, r]
          if(dp[i][0][l][r]!=-INF)
            seg.update(l,min(l+i,r)+1,dp[i][0][l][r]);

          if(dp[i][1][l][r]!=-INF)
            seg.update(max(l,r-i),r+1,dp[i][1][l][r]);

          if(i==0) continue;

          if(dp[i][0][l][r]!=-INF){
            Int res=dp[i][0][l][r];
            // cout<<res<<endl;
            if(l>=1) chmax(dp[i-1][0][l-1][r],res+as[l-1]*(i-1));
            if(i>=2) chmax(dp[i-2][0][l][r],res);
            if((r-l)<=i) chmax(dp[i-(r-l)][1][l][r],res);
          }
          if(dp[i][1][l][r]!=-INF){
            Int res=dp[i][1][l][r];
            // cout<<res<<endl;
            if(r+1<n) chmax(dp[i-1][1][l][r+1],res+as[r+1]*(i-1));
            if(i>=2) chmax(dp[i-2][1][l][r],res);
            if((r-l)<=i) chmax(dp[i-(r-l)][0][l][r],res);
          }
        }
      }
    }
    for(Int i=p;i<n;i+=2) ans[i]=seg.get_val(i);
  }
  for(Int i=0;i<n;i++) cout<<ans[i]<<newl;
  return 0;
}
0