Submission #493768

#	Time	Username	Problem	Language	Result	Execution time	Memory
493768		vulpes	City (BOI06_city)	C++17	90 / 100	1084 ms	16812 KiB

city

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

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

bool cmp(pair<ll,ll> a, 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];
    }
    priority_queue<pair<ll,ll>, vector<pair<ll,ll>>, decltype(&cmp)> q(cmp);
    ll r = 0, d = 1; q.push({0, 0});
    while (n) {
        pair<ll,ll> h = q.top(); q.pop();
        ll z = min(n, 4 * (h.first + 1)); n -= z;
        r += z * (h.first * t + c[h.second]);
        if (h.second + 1 < k) {
            q.push({h.first, h.second + 1});
        }
        if (q.empty() || !cmp({d, 0}, q.top())) {
            q.push({d++, 0});
        }
    }
    cout << r << endl;
}

• Subtask 1: 90 / 100