結果

問題 No.788 トラックの移動
コンテスト
ユーザー Shuz*
提出日時 2018-10-31 07:06:11
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 3,172 bytes
コンパイル時間 1,972 ms
コンパイル使用メモリ 185,896 KB
実行使用メモリ 34,816 KB
最終ジャッジ日時 2024-07-01 05:02:23
合計ジャッジ時間 7,316 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 12 WA * 2
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

//#include <boost/multiprecision/cpp_int.hpp>
// using namespace boost::multiprecision;
// using cint = cpp_int;
//#define input(a) scanf("%lld", &(a))
//#define output(a) printf("%lld\n", (a))

// Define
using ll = long long;
using ull = unsigned long long;
using ld = long double;
const ll MOD = 1e9 + 7;
const ll INF = LONG_MAX;
const ull MAX = ULONG_MAX;
#define endl '\n'
#define space ' '
#define def inline auto
#define func inline constexpr ll
#define run __attribute__((constructor)) def _
#define all(v) begin(v), end(v)

// Debug
#define debug(...)                                                             \
    {                                                                          \
        cerr << __LINE__ << ": " << #__VA_ARGS__ << " = ";                     \
        for (auto &&X : {__VA_ARGS__}) cerr << "[" << X << "] ";               \
        cerr << endl;                                                          \
    }

// Loop
#define inc(i, a, n) for (ll i = (a), _##i = (n); i <= _##i; ++i)
#define dec(i, a, n) for (ll i = (a), _##i = (n); i >= _##i; --i)
#define each(i, a) for (auto &&i : a)
#define rep(i, n) inc(i, 0, n - 1)

// Stream
#define fout(n) cout << fixed << setprecision(n)
#define fasten cin.tie(0), ios::sync_with_stdio(0)

// Speed
run() { fasten, fout(10); }
#pragma GCC optimize("O3")
#pragma GCC optimization_level 3
#pragma GCC target("avx")

// Math
func gcd(ll a, ll b) { return b ? gcd(b, a % b) : a; }
func lcm(ll a, ll b) { return a * b / gcd(a, b); }

def input() {
    ll A;
    cin >> A;
    return A;
}

template <typename T> struct DJK {
    ll V;
    using P = pair<ll, ll>;
    vector<vector<P>> G;
    vector<T> dist;
    vector<bool> used;
    DJK(ll v) : V(v), G(v), dist(v), used(v) {
        fill_n(dist.begin(), V, INF);
        fill_n(used.begin(), V, false);
    }
    void setDist(ll a, ll b, ll d) { G[a].push_back(P(d, b)); }
    void culc(ll s = 0) {
        priority_queue<P, vector<P>, greater<P>> Q;
        fill_n(dist.begin(), V, INF);
        fill_n(used.begin(), V, false);
        Q.push(P(0, s));
        while (!Q.empty()) {
            T d;
            ll t;
            tie(d, t) = Q.top(), Q.pop();
            if (used[t]) continue;
            used[t] = true, dist[t] = d;
            for (P e : G[t]) {
                if (dist[e.second] <= d + e.first) continue;
                Q.push(P(d + e.first, e.second));
            }
        }
    }
};

ll A, B, C, N, M, L, res = INF;
vector<ll> T;
vector<vector<ll>> D;

signed main() {
    cin >> N >> M >> L;
    DJK<ll> dijk(N);
    rep(i, N) T.push_back(input());
    rep(i, M) {
        cin >> A >> B >> C;
        dijk.setDist(A - 1, B - 1, C);
        dijk.setDist(B - 1, A - 1, C);
    }

    rep(i, N) dijk.culc(i), D.push_back(dijk.dist);

    rep(i, N) {
        ll S = 0, P = D[L - 1][i];
        dijk.culc(i);
        rep(j, N) {
            S += 2 * D[i][j] * T[j];
            if (T[j]) P = min(P, D[L - 1][j] - D[i][j]);
        }
        res = min(res, S + P);
    }

    cout << res << endl;
}

// for compilation: g++ -Ofast -march=native -o _ _.cpp -std=c++17
0