#605867 (C++17(gcc12)) No.1344 Typical Shortest Path Sum

提出ソース

結果

問題	No.1344 Typical Shortest Path Sum
ユーザー	r1933
提出日時	2021-01-16 15:19:09
言語	C++17(gcc12) (gcc 12.3.0 + boost 1.87.0)
結果	AC
実行時間	10 ms / 2,000 ms
コード長	7,021 bytes
コンパイル時間	2,838 ms
コンパイル使用メモリ	219,892 KB
実行使用メモリ	6,820 KB
最終ジャッジ日時	2024-11-27 17:49:03
合計ジャッジ時間	4,929 ms
ジャッジサーバーID （参考情報）	judge1 / judge2

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
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,816 KB
testcase_05	AC	2 ms 6,820 KB
testcase_06	AC	2 ms 6,816 KB
testcase_07	AC	2 ms 6,820 KB
testcase_08	AC	2 ms 6,820 KB
testcase_09	AC	2 ms 6,820 KB
testcase_10	AC	2 ms 6,820 KB
testcase_11	AC	2 ms 6,816 KB
testcase_12	AC	2 ms 6,820 KB
testcase_13	AC	2 ms 6,816 KB
testcase_14	AC	2 ms 6,820 KB
testcase_15	AC	2 ms 6,816 KB
testcase_16	AC	2 ms 6,820 KB
testcase_17	AC	2 ms 6,816 KB
testcase_18	AC	2 ms 6,820 KB
testcase_19	AC	2 ms 6,816 KB
testcase_20	AC	2 ms 6,820 KB
testcase_21	AC	2 ms 6,816 KB
testcase_22	AC	2 ms 6,816 KB
testcase_23	AC	2 ms 6,816 KB
testcase_24	AC	2 ms 6,820 KB
testcase_25	AC	2 ms 6,816 KB
testcase_26	AC	2 ms 6,816 KB
testcase_27	AC	2 ms 6,816 KB
testcase_28	AC	2 ms 6,816 KB
testcase_29	AC	2 ms 6,816 KB
testcase_30	AC	2 ms 6,816 KB
testcase_31	AC	2 ms 6,816 KB
testcase_32	AC	2 ms 6,816 KB
testcase_33	AC	2 ms 6,820 KB
testcase_34	AC	2 ms 6,816 KB
testcase_35	AC	2 ms 6,816 KB
testcase_36	AC	2 ms 6,816 KB
testcase_37	AC	2 ms 6,820 KB
testcase_38	AC	2 ms 6,820 KB
testcase_39	AC	2 ms 6,816 KB
testcase_40	AC	3 ms 6,816 KB
testcase_41	AC	3 ms 6,816 KB
testcase_42	AC	3 ms 6,816 KB
testcase_43	AC	3 ms 6,820 KB
testcase_44	AC	3 ms 6,816 KB
testcase_45	AC	3 ms 6,816 KB
testcase_46	AC	4 ms 6,820 KB
testcase_47	AC	3 ms 6,816 KB
testcase_48	AC	4 ms 6,816 KB
testcase_49	AC	3 ms 6,816 KB
testcase_50	AC	3 ms 6,816 KB
testcase_51	AC	3 ms 6,816 KB
testcase_52	AC	3 ms 6,816 KB
testcase_53	AC	4 ms 6,820 KB
testcase_54	AC	3 ms 6,820 KB
testcase_55	AC	3 ms 6,816 KB
testcase_56	AC	3 ms 6,816 KB
testcase_57	AC	3 ms 6,816 KB
testcase_58	AC	3 ms 6,816 KB
testcase_59	AC	3 ms 6,820 KB
testcase_60	AC	2 ms 6,816 KB
testcase_61	AC	2 ms 6,820 KB
testcase_62	AC	8 ms 6,816 KB
testcase_63	AC	3 ms 6,816 KB
testcase_64	AC	3 ms 6,820 KB
testcase_65	AC	10 ms 6,816 KB
testcase_66	AC	7 ms 6,820 KB
testcase_67	AC	8 ms 6,820 KB
testcase_68	AC	2 ms 6,816 KB
testcase_69	AC	3 ms 6,816 KB
testcase_70	AC	2 ms 6,820 KB
testcase_71	AC	2 ms 6,816 KB
testcase_72	AC	2 ms 6,816 KB
testcase_73	AC	2 ms 6,816 KB
testcase_74	AC	2 ms 6,820 KB
testcase_75	AC	2 ms 6,816 KB
testcase_76	AC	2 ms 6,816 KB
testcase_77	AC	2 ms 6,820 KB
testcase_78	AC	2 ms 6,820 KB
testcase_79	AC	2 ms 6,816 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#pragma GCC optimize("Ofast")
#include "bits/stdc++.h"

// Begin Header {{{
using namespace std;

#ifndef DEBUG
#define dump(...)
#endif

#define all(x) begin(x), end(x)
#define rall(x) rbegin(x), rend(x)
#define rep(i, b, e) for (intmax_t i = (b), i##_limit = (e); i < i##_limit; ++i)
#define repc(i, b, e) for (intmax_t i = (b), i##_limit = (e); i <= i##_limit; ++i)
#define repr(i, b, e) for (intmax_t i = (b), i##_limit = (e); i >= i##_limit; --i)
#define var(Type, ...) Type __VA_ARGS__; input(__VA_ARGS__)

constexpr size_t    operator""_zu(unsigned long long value) { return value; };
constexpr intmax_t  operator""_jd(unsigned long long value) { return value; };
constexpr uintmax_t operator""_ju(unsigned long long value) { return value; };

constexpr int INF = 0x3f3f3f3f;
constexpr intmax_t LINF = 0x3f3f3f3f3f3f3f3f_jd;

template <class T, class Compare = less<>>
using MaxHeap = priority_queue<T, vector<T>, Compare>;
template <class T, class Compare = greater<>>
using MinHeap = priority_queue<T, vector<T>, Compare>;

inline void input() {}
template <class Head, class... Tail>
inline void input(Head&& head, Tail&&... tail) {
    cin >> head;
    input(forward<Tail>(tail)...);
}

template <class Container, class Value = typename Container::value_type,
enable_if_t<!is_same<Container, string>::value, nullptr_t> = nullptr>
inline istream& operator>>(istream &is, Container &vs) {
    for (auto &v: vs) is >> v;
    return is;
}

inline void output() { cout << "\n"; }
template <class Head, class... Tail>
inline void output(Head&& head, Tail&&... tail) {
    cout << head;
    if (sizeof...(tail)) cout << " ";
    output(forward<Tail>(tail)...);
}

template <class Container, class Value = typename Container::value_type,
enable_if_t<!is_same<Container, string>::value, nullptr_t> = nullptr>
inline ostream& operator<<(ostream &os, const Container &vs) {
    static constexpr const char *delim[] = {" ", ""};
    for (auto it = begin(vs); it != end(vs); ++it) {
        os << delim[it == begin(vs)] << *it;
    }
    return os;
}

template <class Iterator>
inline void join(const Iterator &Begin, const Iterator &End, const string &delim = "\n", const string &last = "\n") {
    for (auto it = Begin; it != End; ++it) {
        cout << ((it == Begin) ? "" : delim) << *it;
    }
    cout << last;
}

template <class T>
inline vector<T> makeVector(const T &init_value, size_t sz) {
    return vector<T>(sz, init_value);
}

template <class T, class... Args>
inline auto makeVector(const T &init_value, size_t sz, Args... args) {
    return vector<decltype(makeVector<T>(init_value, args...))>(sz, makeVector<T>(init_value, args...));
}

template <class Func>
class FixPoint : Func {
public:
    explicit constexpr FixPoint(Func&& f) noexcept : Func(forward<Func>(f)) {}

    template <class... Args>
    constexpr decltype(auto) operator()(Args&&... args) const {
        return Func::operator()(*this, std::forward<Args>(args)...);
    }
};

template <class Func>
static inline constexpr decltype(auto) makeFixPoint(Func&& f) noexcept {
    return FixPoint<Func>{forward<Func>(f)};
}

template <class Container>
struct reverse_t {
    Container &c;
    reverse_t(Container &c) : c(c) {}
    auto begin() { return c.rbegin(); }
    auto end() { return c.rend(); }
};

template <class Container>
auto reversed(Container &c) {
    return reverse_t<Container>(c);
}

template <class T>
inline bool chmax(T &a, const T &b) noexcept {
    return b > a && (a = b, true);
}

template <class T>
inline bool chmin(T &a, const T &b) noexcept {
    return b < a && (a = b, true);
}

template <class T>
inline T diff(const T &a, const T &b) noexcept {
    return a < b ? b - a : a - b;
}

void operator|=(vector<bool>::reference lhs, const bool rhs) {
    lhs = lhs | rhs;
}

void ioinit() {
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);
    cout << fixed << setprecision(10);
    cerr << fixed << setprecision(10);
    clog << fixed << setprecision(10);
}
// }}} End Header

// Edge {{{
template <class Weight>
struct Edge {
    size_t from, to;
    Weight weight;

    Edge() {}
    Edge(size_t from, size_t to, Weight weight = 1) :
        from(from), to(to), weight(weight)
    {}

    bool operator<(const Edge &rhs) const {
        return weight < rhs.weight;
    }

    bool operator>(const Edge &rhs) const {
        return weight > rhs.weight;
    }

    operator size_t() const {
        return to;
    }
};
// }}}

// Graph {{{
template <class Weight>
class Graph : public vector<vector<Edge<Weight>>> {
    using graph = vector<vector<Edge<Weight>>>;

public:
    Graph() {}
    Graph(const size_t V) : graph(V) {}

    void connect(size_t from, size_t to, Weight weight = 1) {
        (*this)[from].emplace_back(from, to, weight);
    }

    friend ostream& operator<<(ostream &strm, const Graph &G) {
        for (size_t v = 0; v < G.size(); ++v) {
            strm << '[' << setw(2) << v << ']';
            for (const auto &e: G[v]) {
                strm << ' ' << setw(2) << e.to;
            }
            strm << '\n';
        }
        return strm;
    }
};
// }}}

/**
 * 始点からの経路が存在しない頂点 v: dp[v] = INF
 * 重みが無限大 (又はその逆) となる頂点 u: dp[u] = RINF
 **/

// bellmanFord {{{
template <class Weight, class Compare = less<>>
vector<Weight> bellmanFord(const Graph<Weight> &G, const size_t src) {
    const auto isBetter = Compare();
    const Weight MAX = numeric_limits<Weight>::max();
    const Weight MIN = numeric_limits<Weight>::min();
    const Weight INF = (Compare()(MIN, MAX) ? MAX : MIN);
    const Weight RINF = (Compare()(MAX, MIN) ? MAX : MIN);

    const auto grow = [&isBetter](Weight &a, const Weight &b) -> bool {
        return isBetter(b, a) && (a = b, true);
    };

    const size_t V = G.size();

    vector<Weight> dp(V, INF);
    dp[src] = 0;

    for (size_t i = 0; i < V - 1; ++i) {
        for (size_t s = 0; s < V; ++s) {
            if (dp[s] != INF) {
                for (const auto &e: G[s]) {
                    grow(dp[e.to], dp[s] + e.weight);
                }
            }
        }
    }

    vector<bool> updated(V);
    for (size_t i = 0; i < V; ++i) {
        for (size_t s = 0; s < V; ++s) {
            if (dp[s] != INF) {
                for (const auto &e: G[s]) {
                    if (updated[s] || grow(dp[e.to], dp[s] + e.weight)) {
                        updated[e.to] = true;
                    }
                }
            }
        }
    }

    for (size_t v = 0; v < V; ++v) {
        if (updated[v]) {
            dp[v] = RINF;
        }
    }

    return dp;
}
// }}}

int main() {
    ioinit();

    var(size_t, N, M);

    Graph<intmax_t> G(N);
    rep(_, 0, M) {
        var(size_t, s, t);
        s--, t--;
        var(intmax_t, d);
        G.connect(s, t, d);
    }

    rep(i, 0, N) {
        auto dist = bellmanFord(G, i);
        intmax_t res = 0;
        rep(j, 0, N) if (dist[j] != INTMAX_MAX) res += dist[j];
        output(res);
    }
    return 0;
}

yukicoder

結果

テストケース

ソースコード