結果

問題 No.1117 数列分割
ユーザー keijakkeijak
提出日時 2020-07-17 22:33:03
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
TLE  
実行時間 -
コード長 2,121 bytes
コンパイル時間 2,229 ms
コンパイル使用メモリ 207,520 KB
実行使用メモリ 28,572 KB
最終ジャッジ日時 2024-05-07 11:27:25
合計ジャッジ時間 11,453 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
13,756 KB
testcase_01 AC 2 ms
6,940 KB
testcase_02 AC 2 ms
6,940 KB
testcase_03 AC 61 ms
8,576 KB
testcase_04 AC 513 ms
8,064 KB
testcase_05 AC 2 ms
6,940 KB
testcase_06 AC 2 ms
6,944 KB
testcase_07 AC 3 ms
6,940 KB
testcase_08 AC 1,526 ms
7,296 KB
testcase_09 AC 148 ms
6,944 KB
testcase_10 AC 2,003 ms
7,552 KB
testcase_11 TLE -
testcase_12 -- -
testcase_13 -- -
testcase_14 -- -
testcase_15 -- -
testcase_16 -- -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
testcase_20 -- -
testcase_21 -- -
testcase_22 -- -
testcase_23 -- -
testcase_24 -- -
testcase_25 -- -
testcase_26 -- -
testcase_27 -- -
testcase_28 -- -
権限があれば一括ダウンロードができます

ソースコード

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