#include <bits/stdc++.h>
using namespace std;
int main()
{
ios_base::sync_with_stdio(0);
cin.tie(0);
uint64_t n, t, k;
cin >> n >> t >> k;
uint64_t const w = sqrt(n);
vector<uint64_t> c(k), curr_floor(w, 0);
for (uint64_t &x : c)
cin >> x;
auto compare_cost = [&c, &curr_floor, &t](uint64_t x, uint64_t y)
{
return t * x + c[curr_floor[x]] > t * y + c[curr_floor[y]];
};
priority_queue<uint64_t, vector<uint64_t>, decltype(compare_cost)> q(compare_cost);
for (uint64_t i = 0; i < w; i++)
q.push(i);
uint64_t leftover = n, cost = 0;
while (leftover)
{
uint64_t x = q.top();
q.pop();
cost += min(leftover, 4 * (x + 1)) * (t * x + c[curr_floor[x]]);
leftover -= min(leftover, 4 * (x + 1));
curr_floor[x]++;
if (curr_floor[x] < k)
q.push(x);
}
cout << cost << '\n';
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
0 ms |
212 KB |
Output is correct |
2 |
Correct |
216 ms |
16480 KB |
Output is correct |
3 |
Correct |
4 ms |
856 KB |
Output is correct |
4 |
Correct |
176 ms |
1360 KB |
Output is correct |
5 |
Correct |
8 ms |
1368 KB |
Output is correct |
6 |
Correct |
4 ms |
840 KB |
Output is correct |
7 |
Execution timed out |
1084 ms |
16748 KB |
Time limit exceeded |
8 |
Correct |
819 ms |
2256 KB |
Output is correct |
9 |
Correct |
429 ms |
2128 KB |
Output is correct |
10 |
Correct |
0 ms |
212 KB |
Output is correct |