oj.uz

답안 #698073

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
698073 2023-02-12T08:20:56 Z finn__ City (BOI06_city) C++17
90 / 100
 1000 ms 16452 KB 
#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;
    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);
    q.push(0);
    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);
        if (curr_floor[x] == 1)
            q.push(x + 1);
    }

    cout << cost << '\n';
}

Subtask #1
90.0 / 100.0

# 결과 실행 시간 메모리 Grader output
1 Correct 0 ms 212 KB Output is correct
2 Correct 172 ms 16452 KB Output is correct
3 Correct 1 ms 468 KB Output is correct
4 Correct 94 ms 728 KB Output is correct
5 Correct 1 ms 724 KB Output is correct
6 Correct 1 ms 468 KB Output is correct
7 Execution timed out 1098 ms 8452 KB Time limit exceeded
8 Correct 359 ms 1232 KB Output is correct
9 Correct 240 ms 1020 KB Output is correct
10 Correct 0 ms 212 KB Output is correct