結果

問題 No.2423 Merge Stones
コンテスト
ユーザー Nzt3
提出日時 2023-08-15 18:57:56
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 884 bytes
コンパイル時間 2,526 ms
コンパイル使用メモリ 200,940 KB
最終ジャッジ日時 2025-02-16 08:25:05
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 10 TLE * 1 -- * 61
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
using ll=long long;
// TLE
int main(){
  ios::sync_with_stdio(false);
  cin.tie(nullptr);
  int N,K;
  cin>>N>>K;
  vector A(N,0),C(N,0);
  for(int &i:A)cin>>i;
  for(int &i:C)cin>>i,--i;
  vector dp(N,vector(N,vector(50,-(int)1e9)));
  for(int i=0;i<N;i++){
    dp[i][1][C[i]]=A[i];
  }
  for(int w=2;w<N;w++){
    for(int i=0;i<N;i++){
      for(int j=1;j<w;j++){
        for(int c=0;c<50;c++){
          for(int k=max(0,c-K);k<=min(49,c+K);k++){
            if(dp[i][j][c]>0&&dp[(i+j)%N][w-j][k]>0){
              dp[i][w][c]=dp[i][j][c]+dp[(i+j)%N][w-j][k];
              dp[i][w][k]=dp[i][j][c]+dp[(i+j)%N][w-j][k];
            }
          }
        }
      }
    }
  }
  int ans=0;
  for(int i=0;i<N;i++){
    for(int j=0;j<N;j++){
      for(int k=0;k<50;k++){
        ans=max(ans,dp[i][j][k]);
      }
    }
  }
  cout<<ans<<'\n';
}
0