結果

問題 No.2036 Max Middle
ユーザー NPNP
提出日時 2024-05-13 11:36:47
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 1,107 bytes
コンパイル時間 1,346 ms
コンパイル使用メモリ 97,876 KB
実行使用メモリ 15,912 KB
最終ジャッジ日時 2024-12-20 09:26:01
合計ジャッジ時間 32,495 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 7 TLE * 10
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <set>

using namespace std;

bool is_peak(int i, const vector<int>& A) {
    int N = A.size();
    return 0 < i && i < N - 1 && A[i - 1] < A[i] && A[i] > A[i + 1];
}

int count_operations(int N, vector<int>& A) {
    set<int> peaks;
    for (int i = 1; i < N - 1; ++i) {
        if (is_peak(i, A)) {
            peaks.insert(i);
        }
    }

    int count = 0;
    while (!peaks.empty()) {
        vector<int> to_check(peaks.begin(), peaks.end());
        for (int i : to_check) {
            if (peaks.count(i)) {
                peaks.erase(i);
                count += 1;
                A[i] = min(A[i - 1], A[i + 1]) - 1;
                for (int j : {i - 1, i + 1}) {
                    if (is_peak(j, A)) {
                        peaks.insert(j);
                    }
                }
            }
        }
    }

    return count;
}

int main() {
    int N;
    cin >> N;
    vector<int> A(N);
    for (int i = 0; i < N; ++i) {
        cin >> A[i];
    }

    int result = count_operations(N, A);
    cout << result << endl;

    return 0;
}
0