結果

問題 No.1 道のショートカット
ユーザー はむ吉🐹はむ吉🐹
提出日時 2016-06-05 19:16:45
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 3 ms / 5,000 ms
コード長 4,502 bytes
コンパイル時間 903 ms
コンパイル使用メモリ 87,772 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-07-20 16:23:46
合計ジャッジ時間 1,922 ms
ジャッジサーバーID
(参考情報)
judge2 / judge3
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 1 ms
5,376 KB
testcase_03 AC 2 ms
5,376 KB
testcase_04 AC 2 ms
5,376 KB
testcase_05 AC 1 ms
5,376 KB
testcase_06 AC 2 ms
5,376 KB
testcase_07 AC 2 ms
5,376 KB
testcase_08 AC 2 ms
5,376 KB
testcase_09 AC 2 ms
5,376 KB
testcase_10 AC 3 ms
5,376 KB
testcase_11 AC 2 ms
5,376 KB
testcase_12 AC 2 ms
5,376 KB
testcase_13 AC 2 ms
5,376 KB
testcase_14 AC 1 ms
5,376 KB
testcase_15 AC 2 ms
5,376 KB
testcase_16 AC 2 ms
5,376 KB
testcase_17 AC 2 ms
5,376 KB
testcase_18 AC 2 ms
5,376 KB
testcase_19 AC 2 ms
5,376 KB
testcase_20 AC 2 ms
5,376 KB
testcase_21 AC 2 ms
5,376 KB
testcase_22 AC 2 ms
5,376 KB
testcase_23 AC 2 ms
5,376 KB
testcase_24 AC 2 ms
5,376 KB
testcase_25 AC 2 ms
5,376 KB
testcase_26 AC 2 ms
5,376 KB
testcase_27 AC 2 ms
5,376 KB
testcase_28 AC 2 ms
5,376 KB
testcase_29 AC 2 ms
5,376 KB
testcase_30 AC 2 ms
5,376 KB
testcase_31 AC 2 ms
5,376 KB
testcase_32 AC 2 ms
5,376 KB
testcase_33 AC 1 ms
5,376 KB
testcase_34 AC 1 ms
5,376 KB
testcase_35 AC 2 ms
5,376 KB
testcase_36 AC 1 ms
5,376 KB
testcase_37 AC 1 ms
5,376 KB
testcase_38 AC 1 ms
5,376 KB
testcase_39 AC 2 ms
5,376 KB
testcase_40 AC 2 ms
5,376 KB
testcase_41 AC 2 ms
5,376 KB
testcase_42 AC 1 ms
5,376 KB
testcase_43 AC 1 ms
5,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <ciso646>  // 演算子の代替表記を用いる
#include <cstdlib>  // EXIT_SUCCESSマクロを用いる
#include <functional>  // 優先度付きキューの比較関数としてgreaterを用いる
#include <iostream>  // std::cinおよびstd::coutを用いる
#include <queue>  // 優先度付きキューを用いる
#include <tuple>  // 演算子オーバーロードに用いる
#include <vector>  // いつもの


constexpr int IMPOSSIBLE = -1;  // 制約下で目的地にたどり着けない


typedef int Money;
typedef int Time;
typedef int Vertex;


// 隣接する頂点の情報
class AdjVertex {
public:
    AdjVertex(Vertex dest, Money money, Time time) :
        dest_(dest), money_(money), time_(time) {}
    Vertex dest() const {
        return dest_;
    }
    Money money() const {
        return money_;
    }
    Time time() const {
        return time_;
    }
private:
    Vertex dest_;
    Money money_;
    Time time_;
};


typedef std::vector<std::vector<AdjVertex>> AdjVertices;


// 優先度付きキューに入れる状態
class State {
public:
    State(Time time, Money money, Vertex vertex) :
        time_(time), money_(money), vertex_(vertex) {}
    // Getter
    Time time() const {
        return time_;
    }
    Money money() const {
        return money_;
    }
    Vertex vertex() const {
        return vertex_;
    }
    // 演算子オーバーロード
    // 優先度付きキューに入れるための順序付け
    bool operator==(const State& other) const {
        auto me = std::tie(time_, money_, vertex_);
        auto you = std::tie(other.time_, other.money_, other.vertex_);
        return me == you;
    }
    inline bool operator!=(const State& other) const {
        return not(*this == other);
    }
    bool operator<(const State& other) const {
        auto me = std::tie(time_, money_, vertex_);
        auto you = std::tie(other.time_, other.money_, other.vertex_);
        return me < you;
    }
    inline bool operator>=(const State& other) const {
        return not(*this < other);
    }
    inline bool operator>(const State& other) const {
        return other < *this;
    }
    inline bool operator<=(const State& other) const {
        return not(other < *this);
    }
private:
    Time time_;
    Money money_;
    Vertex vertex_;
};


Time compute_min_time(Vertex num_vertices, Money money_limit, const AdjVertices& adj_vertices) {
    std::priority_queue<State, std::vector<State>, std::greater<State>> pq;
    pq.emplace(0, 0, 0);
    while (not pq.empty()) {
        // キューから状態を取り出す
        auto state0 = pq.top();
        pq.pop();
        // 取り出した状態がゴールならばそれまでの時間を返す
        if (state0.vertex() == num_vertices - 1)
        {
            return state0.time();
        }
        for (const auto& v : adj_vertices[state0.vertex()])
        {
            auto new_money = state0.money() + v.money();
            // お金の制限を満たしていればキューに入れる
            if (new_money <= money_limit)
            {
                auto new_time = state0.time() + v.time();
                pq.emplace(new_time, new_money, v.dest());
            }
        }
    }
    // ゴールにたどり着けないなら
    return IMPOSSIBLE;
}


int main() {
    std::cin.tie(nullptr);
    std::ios::sync_with_stdio(false);
    Vertex num_vertices;
    Money money_limit;
    int num_edges;
    std::cin >> num_vertices >> money_limit >> num_edges;
    std::vector<Vertex> sources(num_edges);
    std::vector<Vertex> dests(num_edges);
    std::vector<Money> moneys(num_edges);
    std::vector<Time> times(num_edges);
    for (std::size_t i = 0; i < num_edges; i++)
    {
        std::cin >> sources[i];
        sources[i]--;
    }
    for (std::size_t i = 0; i < num_edges; i++)
    {
        std::cin >> dests[i];
        dests[i]--;
    }
    for (std::size_t i = 0; i < num_edges; i++)
    {
        std::cin >> moneys[i];
    }
    for (std::size_t i = 0; i < num_edges; i++)
    {
        std::cin >> times[i];
    }
    // 隣接リストの形でグラフを管理する
    AdjVertices adj_vertices(num_vertices, std::vector<AdjVertex>());
    for (std::size_t i = 0; i < num_edges; i++)
    {
        adj_vertices[sources[i]].emplace_back(dests[i], moneys[i], times[i]);
    }
    auto answer = compute_min_time(num_vertices, money_limit, adj_vertices);
    std::cout << answer << std::endl;
    return EXIT_SUCCESS;
}
0