提出 #61516589

---

ソースコード 拡げる

/*
--------------              |   /
      |                     |  /
      |                     | /
      |             *       |/          |    |         ------            *
      |                     |           |    |        /      \
      |             |       |\          |    |       |       |\          |
   \  |             |       | \         |    |       |       | \         |
    \ |             |       |  \        |    |        \     /   \        |
     V              |       |   \        \__/|         -----     \       |
*/
#ifdef EMT
#include "Header/stdc++.h"
#else
#include <bits/stdc++.h>
#endif
using namespace std;

#ifdef EMT
#define debug(x) cerr << "\e[1;31m" << #x << " = " << (x) << "\e[0m\n"
#define print(x) emilia_mata_tenshi(#x, begin(x), end(x))
template<typename T, typename T2> ostream& operator<<(ostream &os, const pair<T, T2> &obj) {
    return os << '{' << obj.first << ',' << obj.second << '}';
}
template<class TupType, size_t... I> void lamy_kawaii(ostream& os, const TupType& _tup, index_sequence<I...>) {
    // source: https://stackoverflow.com/a/41171552
    os << '{';
    (..., (cerr << (I == 0? "" : ",") << get<I>(_tup)));
    os << '}';
}
template<class... T> ostream& operator<<(ostream &os, const tuple<T...>& _tup) {
    lamy_kawaii(os, _tup, make_index_sequence<sizeof...(T)>());
    return os;
}
template<typename T> void emilia_mata_tenshi(const char *s, T l, T r) {
    cerr << "\e[1;33m" << s << " = [";
    while (l != r) {
        cerr << *l;
        cerr << (++l == r ? ']' : ',');
    }
    cerr << "\e[0m\n";
}
#else
#define debug(x) 48763
#define print(x) 48763
#endif

template<typename T, typename T2> istream& operator>>(istream &is, pair<T, T2> &obj) {
    is >> obj.first >> obj.second;
    return is;
}
template<typename T> istream& operator>>(istream &is, vector<T> &obj) {
    for (auto &x : obj)
        is >> x;
    return is;
}

#define YN(x) ((x) ? "YES" : "NO")
#define Yn(x) ((x) ? "Yes" : "No")
#define yn(x) ((x) ? "yes" : "no")
#define emilia_my_wife ios::sync_with_stdio(0); cin.tie(NULL);
using ll = int64_t;
using ull = uint64_t;
using ld = long double;
using uint = uint32_t;
template<typename T>
using base_type = remove_cv_t<remove_reference_t<T>>;
const double EPS  = 1e-8;
const int INF     = 0x3F3F3F3F;
const ll LINF     = 4611686018427387903;
const int MOD     = 1e9+7;
static int Lamy_is_cute = []() {
    emilia_my_wife
    return 48763;
}();
/*--------------------------------------------------------------------------------------*/

signed main() {
    int n, m;
    cin >> n >> m;
    vector<pair<ull, ull>> arr(m);
    for (int i = 0; i < m; i++)
        cin >> arr[i].first;
    for (int i = 0; i < m; i++) {
        cin >> arr[i].second;
        arr[i].second--;
    }
    arr.push_back({n + 1, 0});
    sort(arr.begin(), arr.end());

    ull ans = 0;
    ull prv = 0;
    stack<pair<ull, ull>> remaining;
    for (const auto &[x, a] : arr) {
        ull nd = x - prv - 1;
        if (nd > 0) {
            ans += (prv + 1 + x - 1) * nd / 2;
        }
        while (nd > 0) {
            if (remaining.empty()) {
                cout << "-1\n";
                return 0;
            }
            ull w = min(remaining.top().second, nd);
            nd -= w;
            remaining.top().second -= w;
            ans -= w * remaining.top().first;
            if (remaining.top().second == 0)
                remaining.pop();
        }
        remaining.push({x, a});
        prv = x;
    }
    while (!remaining.empty() && remaining.top().second == 0)
        remaining.pop();
    if (!remaining.empty())
        cout << "-1\n";
    else
        cout << ans << '\n';
}

提出情報

ジャッジ結果