Submission #493767

#	Time	Username	Problem	Language	Result	Execution time	Memory
493767		vulpes	City (BOI06_city)	C++17	90 / 100	1094 ms	88768 KiB

city

#include <bits/stdc++.h>
using namespace std;

typedef long long ll;
ll t, c[20007];

bool cmp(const pair<ll,ll> &a, const pair<ll,ll> &b) {
    return a.first * t + c[a.second] <
           b.first * t + c[b.second];
}

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(0); cout.tie(0);
    ll n, k; cin >> n >> t >> k;
    for (int i = 0; i < k; i++) {
        cin >> c[i];
    }
    multiset<pair<ll,ll>, decltype(&cmp)> q(cmp);
    ll r = 0; q.insert({0, 0});
    map<ll,bool> m;
    while (n) {
        pair<ll,ll> h = *q.begin();
        q.erase(q.begin());
        ll z = min(n, 4 * (h.first + 1)); n -= z;
        r += z * (h.first * t + c[h.second]);
        if (h.second + 1 < k) {
            q.insert({h.first, h.second + 1});
        }
        if (!m[h.first + 1]) {
            m[h.first + 1] = true;
            q.insert({h.first + 1, 0});
        }
    }
    cout << r << endl;
}

• Subtask 1: 90 / 100