#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;
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
0 ms |
204 KB |
Output is correct |
2 |
Correct |
202 ms |
16812 KB |
Output is correct |
3 |
Correct |
1 ms |
204 KB |
Output is correct |
4 |
Correct |
84 ms |
364 KB |
Output is correct |
5 |
Correct |
2 ms |
204 KB |
Output is correct |
6 |
Correct |
1 ms |
204 KB |
Output is correct |
7 |
Execution timed out |
1084 ms |
508 KB |
Time limit exceeded |
8 |
Correct |
321 ms |
356 KB |
Output is correct |
9 |
Correct |
178 ms |
332 KB |
Output is correct |
10 |
Correct |
0 ms |
204 KB |
Output is correct |