#include <bits/stdc++.h>
using namespace std;
#pragma GCC optimize("O3")
#pragma GCC target("avx2")
int main()
{
ios_base::sync_with_stdio(0);
cin.tie(0);
uint64_t n, t, k;
cin >> n >> t >> k;
vector<uint64_t> c(k);
for (uint64_t &x : c)
cin >> x;
auto compare_cost = [&c, &t](pair<uint64_t, uint64_t> const &x, pair<uint64_t, uint64_t> const &y)
{
return t * x.first + c[x.second] > t * y.first + c[y.second];
};
priority_queue<pair<uint64_t, uint64_t>, vector<pair<uint64_t, uint64_t>>, decltype(compare_cost)> q(compare_cost);
q.emplace(0, 0);
uint64_t leftover = n, cost = 0;
while (leftover)
{
auto p = q.top();
q.pop();
while (leftover && (q.empty() || compare_cost(q.top(), p)) && p.second < k)
{
cost += min(leftover, 4 * (p.first + 1)) * (t * p.first + c[p.second]);
leftover -= min(leftover, 4 * (p.first + 1));
p.second++;
if (p.second == 1)
q.emplace(p.first + 1, 0);
}
if (p.second < k)
q.push(p);
}
cout << cost << '\n';
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
0 ms |
212 KB |
Output is correct |
2 |
Correct |
161 ms |
16856 KB |
Output is correct |
3 |
Correct |
1 ms |
212 KB |
Output is correct |
4 |
Execution timed out |
1086 ms |
340 KB |
Time limit exceeded |
5 |
Correct |
1 ms |
212 KB |
Output is correct |
6 |
Correct |
1 ms |
212 KB |
Output is correct |
7 |
Execution timed out |
1098 ms |
468 KB |
Time limit exceeded |
8 |
Execution timed out |
1092 ms |
364 KB |
Time limit exceeded |
9 |
Execution timed out |
1090 ms |
340 KB |
Time limit exceeded |
10 |
Correct |
0 ms |
212 KB |
Output is correct |