結果

問題 No.1247 ブロック登り
ユーザー hitonanode
提出日時 2020-10-03 13:49:21
言語 C++17(gcc12)
(gcc 12.3.0 + boost 1.87.0)
結果
AC  
実行時間 282 ms / 3,000 ms
コード長 2,003 bytes
コンパイル時間 3,333 ms
コンパイル使用メモリ 106,948 KB
最終ジャッジ日時 2025-01-15 02:02:45
ジャッジサーバーID
(参考情報)
judge3 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff #

#pragma GCC optimize("O3")
#pragma GCC target("avx")
#include <iostream>
#include <vector>
using namespace std;
#define FOR(i, begin, end) for (int i = (begin), i##_end_ = (end); i < i##_end_; i++)
#define IFOR(i, begin, end) for (int i = (end)-1, i##_begin_ = (begin); i >= i##_begin_; i--)
#define REP(i, n) FOR(i, 0, n)
template <typename T>
bool chmax(T& m, const T q) { return m < q ? (m = q, true) : false; }


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

    constexpr int INF = 2e9;
    // vector dpl(K + 1, vector(N, vector(N, -INF)));
    // vector dpr(K + 1, vector(N, vector(N, -INF)));
    int dpl[301][300][300];
    int dpr[301][300][300];
    // array<array<array<int, 300>, 300>, 301> dpl;
    // array<array<array<int, 300>, 300>, 301> dpr;
    REP(k, K + 1) REP(i, N) REP(j, N) dpl[k][i][j] = dpr[k][i][j] = -INF;
    REP(i, N) dpl[K][i][i] = dpr[K][i][i] = A[i] * K;

    IFOR(k, 1, K + 1)
    {
        REP(l, N) FOR(r, l, N)
        {
            if (l < r and k > 1)
            {
                chmax(dpl[k - 2][l][r], dpl[k][l][r]);
                chmax(dpr[k - 2][l][r], dpr[k][l][r]);
            }
            if (l) chmax(dpl[k - 1][l - 1][r], dpl[k][l][r] + A[l - 1] * (k - 1));
            if (r + 1 < N) chmax(dpr[k - 1][l][r + 1], dpr[k][l][r] + A[r + 1] * (k - 1));

            if (k - (r - l) >= 0) chmax(dpr[k - (r - l)][l][r], dpl[k][l][r]);
            if (k - (r - l) >= 0) chmax(dpl[k - (r - l)][l][r], dpr[k][l][r]);
        }
    }

    IFOR(k, 1, K + 1)
    {
        REP(l, N) FOR(r, l + 1, N)
        {
            chmax(dpl[k - 1][l + 1][r], dpl[k][l][r]);
            chmax(dpr[k - 1][l][r - 1], dpr[k][l][r]);
        }
    }
    REP(i, N)
    {
        int ret = -INF;
        REP(j, N)
        {
            if (j <= i) chmax(ret, dpr[0][j][i]);
            if (i <= j) chmax(ret, dpl[0][i][j]);
        }
        cout << ret << '\n';
    }
}
0