結果

問題 No.1867 Partitions and Inversions
ユーザー Cyanmond
提出日時 2022-03-06 10:10:39
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,787 bytes
コンパイル時間 2,199 ms
コンパイル使用メモリ 206,692 KB
最終ジャッジ日時 2025-01-28 07:11:12
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 7 WA * 58
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#pragma GCC target("avx2")
#pragma GCC optimize("Ofast")
#pragma GCC optimize("unroll-loops")
using namespace std;

int cnt[3001][3001], dp[3001][3001];

int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    int N;
    cin >> N;
    vector<int> P(N);
    for (int i = 0; i < N; i++) {
        cin >> P[i];
    }
    for (int i = 0; i < N; i++) {
        for (int j = i; j < N; j++) {
            if (P[i] > P[j]) {
                cnt[i][j]++;
                cnt[j][j]--;
            }
        }
    }
    for (int j = 0; j < N; j++) {
        for (int i = 0; i < j; i++) {
            cnt[i + 1][j] += cnt[i][j];
        }
    }
    for (int i = 0; i < N; i++) {
        for (int j = i; j < N; j++) {
            cnt[i][j + 1] += cnt[i][j];
        }
    }

    for (int i = 0; i < 3001; ++i) dp[i][1] = 0;
    for (int i = 2; i <= N; i++) {
        stack<int> st;
        st.push(i - 2);

        auto is_better = [&](int a, int b, int c) {
            return dp[a][i - 1] + cnt[a][c] < dp[b][i - 1] + cnt[b][c];
        };

        for (int j = i - 1; j < N; j++) {
            while (st.size() > 1) {
                int t = st.top();
                st.pop();
                int b = st.top();
                if (is_better(t, b, j)) {
                    st.push(t);
                    break;
                }
            }
            dp[j][i] = dp[st.top()][i - 1] + cnt[st.top()][j];

            while (not st.empty()) {
                int t = st.top();
                if (is_better(j, t, N - 1))
                    st.pop();
                else
                    break;
            }

            st.push(j);
        }
    }
    for (int i = 1; i <= N; i++) {
        cout << dp[N - 1][i] << '\n';
    }
}
// 正当性なし
0