結果

問題 No.705 ゴミ拾い Hard
ユーザー 👑 rin204rin204
提出日時 2024-10-27 14:34:38
言語 C++23
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 115 ms / 1,500 ms
コード長 2,484 bytes
コンパイル時間 3,162 ms
コンパイル使用メモリ 249,824 KB
実行使用メモリ 13,824 KB
最終ジャッジ日時 2024-10-27 14:34:46
合計ジャッジ時間 7,468 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,816 KB
testcase_01 AC 2 ms
6,820 KB
testcase_02 AC 2 ms
6,820 KB
testcase_03 AC 2 ms
6,816 KB
testcase_04 AC 2 ms
6,820 KB
testcase_05 AC 2 ms
6,816 KB
testcase_06 AC 2 ms
6,816 KB
testcase_07 AC 2 ms
6,816 KB
testcase_08 AC 2 ms
6,816 KB
testcase_09 AC 2 ms
6,816 KB
testcase_10 AC 2 ms
6,820 KB
testcase_11 AC 2 ms
6,816 KB
testcase_12 AC 2 ms
6,816 KB
testcase_13 AC 2 ms
6,816 KB
testcase_14 AC 2 ms
6,816 KB
testcase_15 AC 2 ms
6,820 KB
testcase_16 AC 2 ms
6,816 KB
testcase_17 AC 2 ms
6,816 KB
testcase_18 AC 2 ms
6,816 KB
testcase_19 AC 2 ms
6,816 KB
testcase_20 AC 2 ms
6,816 KB
testcase_21 AC 2 ms
6,816 KB
testcase_22 AC 2 ms
6,816 KB
testcase_23 AC 2 ms
6,820 KB
testcase_24 AC 102 ms
13,608 KB
testcase_25 AC 103 ms
13,696 KB
testcase_26 AC 102 ms
13,640 KB
testcase_27 AC 115 ms
13,616 KB
testcase_28 AC 101 ms
13,696 KB
testcase_29 AC 102 ms
13,688 KB
testcase_30 AC 113 ms
13,696 KB
testcase_31 AC 96 ms
13,744 KB
testcase_32 AC 104 ms
13,824 KB
testcase_33 AC 69 ms
13,768 KB
testcase_34 AC 69 ms
13,760 KB
testcase_35 AC 76 ms
13,724 KB
testcase_36 AC 84 ms
13,696 KB
testcase_37 AC 77 ms
13,708 KB
testcase_38 AC 84 ms
13,628 KB
testcase_39 AC 104 ms
13,776 KB
testcase_40 AC 2 ms
6,816 KB
testcase_41 AC 2 ms
6,816 KB
testcase_42 AC 2 ms
6,820 KB
testcase_43 AC 98 ms
13,696 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#define PROBLEM "https://yukicoder.me/problems/no/705"

#include <bits/stdc++.h>

template <typename T = long long, typename Func>
T monge_shortest_path(int n, Func cost_function) {
    assert(n > 0);
    std::vector<T> dist(n);
    std::vector<int> amin(n, 0);

    auto update = [&](int i, int k) {
        if (i <= k) return;
        T c = dist[k] + cost_function(k, i);
        if (c < dist[i]) {
            dist[i] = c;
            amin[i] = k;
        }
    };

    auto rec_solve = [&](auto &&self, int l, int r) -> void {
        if (r - l == 1) return;

        int m = (l + r) / 2;
        for (int k = amin[l]; k <= amin[r]; k++) {
            update(m, k);
        }

        self(self, l, m);
        for (int k = l + 1; k <= m; k++) {
            update(r, k);
        }
        self(self, m, r);
    };

    dist[0] = T();
    for (int i = 1; i < n; i++) {
        dist[i] = cost_function(0, i);
    }

    rec_solve(rec_solve, 0, n - 1);
    return dist.back();
}

void solve() {
    int n;
    std::cin >> n;
    std::vector<long long> A(n);
    std::vector<long long> X(n);
    std::vector<long long> Y(n);
    for (auto &a : A) std::cin >> a;
    for (auto &x : X) std::cin >> x;
    for (auto &y : Y) std::cin >> y;

    auto f = [&](int l, int r) {
        assert(l < r);
        long long dx = abs(A[r - 1] - X[l]);
        long long dy = Y[l];
        return dx * dx * dx + dy * dy * dy;
    };

    long long ans = monge_shortest_path(n + 1, f);
    std::cout << ans << std::endl;
}

int main() {
    std::cin.tie(0)->sync_with_stdio(0);

    int t = 1;
    // cin >> t;
    while (t--) solve();
}

// #define PROBLEM "https://yukicoder.me/problems/no/705"
//
// #include <bits/stdc++.h>
//
// #include "graph/monge_shortest_path.hpp"
//
// void solve() {
//     int n;
//     std::cin >> n;
//     std::vector<long long> A(n);
//     std::vector<long long> X(n);
//     std::vector<long long> Y(n);
//     for (auto &a : A) std::cin >> a;
//     for (auto &x : X) std::cin >> x;
//     for (auto &y : Y) std::cin >> y;
//
//     auto f = [&](int l, int r) {
//         assert(l < r);
//         long long dx = abs(A[r - 1] - X[l]);
//         long long dy = Y[l];
//         return dx * dx * dx + dy * dy * dy;
//     };
//
//     long long ans = monge_shortest_path(n + 1, f);
//     std::cout << ans << std::endl;
// }
//
// int main() {
//     std::cin.tie(0)->sync_with_stdio(0);
//
//     int t = 1;
//     // cin >> t;
//     while (t--) solve();
// }
0