結果

問題 No.1117 数列分割
コンテスト
ユーザー keijak
提出日時 2020-07-17 22:33:03
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 2,121 bytes
コンパイル時間 3,388 ms
コンパイル使用メモリ 198,028 KB
最終ジャッジ日時 2025-01-11 23:04:37
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 17 WA * 3 TLE * 6
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
using i64 = long long;
using u64 = unsigned long long;
#define REP(i, n) for (int i = 0; (i64)(i) < (i64)(n); ++i)

#ifdef ENABLE_DEBUG
template <typename T>
void debug(T value) {
  cerr << value;
}
template <typename T, typename... Ts>
void debug(T value, Ts... args) {
  cerr << value << ", ";
  debug(args...);
}
#define DEBUG(...)                              \
  do {                                          \
    cerr << " \033[33m (L" << __LINE__ << ") "; \
    cerr << #__VA_ARGS__ << ":\033[0m ";        \
    debug(__VA_ARGS__);                         \
    cerr << endl;                               \
  } while (0)
#else
#define debug(...)
#define DEBUG(...)
#endif

int main() {
  ios::sync_with_stdio(false);
  cin.tie(nullptr);
  i64 N, K, M;
  cin >> N >> K >> M;
  vector<i64> A(N);
  for (auto& x : A) cin >> x;

  // vector<vector<vector<i64>>> dp(n + 1, vector<vector<i64>>(m,
  // vector<i64>(2)));

  vector<vector<i64>> memo(N + 1, vector<i64>(K + 1, -1));
  auto f = [&](auto rec, int len, int block_count) -> i64 {
    if (len == 0 || block_count == 0) return 0;
    if (len < block_count) return 0;
    if (memo[len][block_count] >= 0) return memo[len][block_count];
    if (len == block_count) {
      i64 sum = 0;
      for (int i = 0; i < len; ++i) {
        sum += abs(A[i]);
      }
      DEBUG(len, block_count, sum);
      return (memo[len][block_count] = sum);
    }
    if (block_count == 1) {
      i64 sum = 0;
      for (int i = 0; i < len; ++i) {
        sum += A[i];
      }
      return (memo[len][block_count] = abs(sum));
    }
    int m = min<int>(M, len);
    i64 max_score = 0;
    i64 block_sum = 0;
    for (int x = 1; x <= m; ++x) {
      assert(len - x >= 0);
      assert(len - x < N);
      block_sum += A[len - x];
      i64 r = rec(rec, len - x, block_count - 1);
      max_score = max(max_score, r + abs(block_sum));
      if (len == 4) {
        DEBUG(len, block_count, x, block_sum, r, max_score);
      }
    }
    return (memo[len][block_count] = max_score);
  };
  i64 ans = f(f, N, K);
  cout << ans << endl;
}
0