#include <bits/stdc++.h>
using namespace std;
#define num long long
#define vn vector<num>
#define ph pair<num, pair<int, int>>
#define sp set<ph>
#define stage second.second
#define dist second.first
#define cost first
vn costs;
num N, T;
int K;
sp current;
ph next (ph &x) {
int s = x.stage + 1;
int d = x.dist;
num c = d * T + costs[s];
return { c, { d, s } };
}
ph create (int d) {
return { d * T + costs[0], { d, 0 } };
}
num count (int d) {
return 4 * d + 4;
}
int main () {
cin >> N >> T >> K;
costs.resize(K);
for (int i = 0; i < K; i ++) cin >> costs[i];
current.insert(create(0));
num ans = 0;
int i = 0;
while (N > 0) {
ph res = *(current.begin());
current.erase(res);
if (res.stage + 1 != K)
current.insert(next(res));
if (res.stage == 0)
current.insert(create(res.dist + 1));
num size = min(count(res.dist), N);
N -= size;
ans += size * res.cost;
}
cout << ans << endl;
}
Compilation message
city.cpp: In function 'int main()':
city.cpp:42:9: warning: unused variable 'i' [-Wunused-variable]
42 | int i = 0;
| ^
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
212 KB |
Output is correct |
2 |
Correct |
314 ms |
44476 KB |
Output is correct |
3 |
Correct |
1 ms |
304 KB |
Output is correct |
4 |
Correct |
159 ms |
392 KB |
Output is correct |
5 |
Correct |
2 ms |
212 KB |
Output is correct |
6 |
Correct |
2 ms |
296 KB |
Output is correct |
7 |
Execution timed out |
1089 ms |
924 KB |
Time limit exceeded |
8 |
Correct |
523 ms |
468 KB |
Output is correct |
9 |
Correct |
317 ms |
484 KB |
Output is correct |
10 |
Correct |
1 ms |
212 KB |
Output is correct |