結果

問題 No.929 よくあるボールを移動するやつ
コンテスト
ユーザー もりを
提出日時 2019-11-22 21:46:30
言語 C++14
(gcc 13.3.0 + boost 1.89.0)
結果
WA  
実行時間 -
コード長 1,307 bytes
記録
コンパイル時間 1,715 ms
コンパイル使用メモリ 177,800 KB
実行使用メモリ 6,824 KB
最終ジャッジ日時 2024-10-11 03:13:31
合計ジャッジ時間 2,688 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 5 WA * 7
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include <bits/stdc++.h>
using namespace std;
using i64 = int_fast64_t;
using ui64 = uint_fast64_t;
#define REP(i, n) for (i64 (i) = 0; (i) < (n); ++(i))
#define FOR(i, a, b) for (i64 (i) = (a); (i) < (b); ++(i))

i64 res = 1LL << 60;
int N;
vector<int> B;

signed main() {

    cin >> N;
    B.resize(N);
    REP(i, N) cin >> B[i];

    {
        set<pair<int, int>> st;
        REP(i, N) {
            if (B[i]) st.emplace(i, B[i]);
        }
        i64 tmp = 0;
        REP(i, N) {
            auto itr = st.lower_bound(make_pair(i, 0));
            if (itr == st.end()) itr = prev(itr);
            pair<int, int> p = *itr;    p.second--;
            st.erase(itr);
            tmp += abs(p.first - i);
            if (p.second > 0) st.emplace(p);
        }
        res = min(res, tmp);
    }

    {
        set<pair<int, int>> st;
        REP(i, N) {
            if (B[N - 1 - i]) st.emplace(i, B[N - 1 - i]);
        }
        i64 tmp = 0;
        REP(i, N) {
            auto itr = st.lower_bound(make_pair(i, 0));
            if (itr == st.end()) itr = prev(itr);
            pair<int, int> p = *itr;    p.second--;
            st.erase(itr);
            tmp += abs(p.first - i);
            if (p.second > 0) st.emplace(p);
        }
        res = min(res, tmp);
    }

    cout << res << endl;

}
0