結果

問題 No.929 よくあるボールを移動するやつ
コンテスト
ユーザー r1933
提出日時 2019-11-22 22:45:26
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 11 ms / 2,000 ms
コード長 1,477 bytes
コンパイル時間 1,957 ms
コンパイル使用メモリ 179,648 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-10-11 04:22:48
合計ジャッジ時間 2,995 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 12
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include "bits/stdc++.h"
#pragma GCC optimize("Ofast")

// Begin {{{
using namespace std;

#ifndef DEBUG
#define dump(...)
#endif

template <class A, class B> inline bool chmax(A &a, const B &b) { return b > a && (a = b, true); }
template <class A, class B> inline bool chmin(A &a, const B &b) { return b < a && (a = b, true); }

template <class T>
inline vector<T> make_v(const T &initvalue, size_t sz) {
    return vector<T>(sz, initvalue);
}

template <class T, class... Args>
inline auto make_v(const T &initvalue, size_t sz, Args... args) {
    return vector<decltype(make_v<T>(initvalue, args...))>(sz, make_v<T>(initvalue, args...));
}

constexpr int INF = 0x3f3f3f3f;
constexpr int64_t LINF = 0x3f3f3f3f3f3f3f3fLL;
// }}} End

signed main() {
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);

    size_t N; cin >> N;

    vector<pair<intmax_t, intmax_t>> exist;
    vector<intmax_t> empty_idx;
    for (size_t i = 0; i < N; ++i) {
        intmax_t b; cin >> b;
        if (b == 0) {
            empty_idx.emplace_back(i);
        } else if (b >= 2) {
            exist.emplace_back(i, b - 1);
        }
    }

    sort(empty_idx.begin(), empty_idx.end());
    sort(exist.begin(), exist.end());

    size_t pos = 0;
    intmax_t res = 0;
    for (size_t i = 0; i < exist.size(); ++i) {
        while (exist[i].second--) {
            res += llabs(exist[i].first - empty_idx[pos]);
            pos++;
        }
    }

    cout << res << "\n";

    return 0;
}
0