結果

問題 No.788 トラックの移動
ユーザー はまやんはまやん
提出日時 2019-02-09 09:36:31
言語 C++14
(gcc 8.3.0)
結果
AC  
実行時間 501 ms
コード長 2,838 Byte
コンパイル時間 1,466 ms
使用メモリ 32,984 KB
最終ジャッジ日時 2019-09-12 03:10:57

テストケース

テストケース表示
入力 結果 実行時間
使用メモリ
sample_1 AC 3 ms
1,628 KB
sample_2 AC 3 ms
1,628 KB
sample_3 AC 3 ms
1,628 KB
testcase_1 AC 108 ms
9,480 KB
testcase_2 AC 465 ms
32,976 KB
testcase_3 AC 474 ms
32,984 KB
testcase_4 AC 3 ms
1,632 KB
testcase_5 AC 3 ms
1,636 KB
testcase_6 AC 3 ms
1,632 KB
testcase_7 AC 3 ms
1,636 KB
testcase_8 AC 3 ms
1,636 KB
testcase_9 AC 4 ms
1,632 KB
testcase_10 AC 3 ms
1,624 KB
testcase_11 AC 3 ms
1,624 KB
testcase_12 AC 190 ms
32,952 KB
testcase_13 AC 501 ms
32,980 KB
テストケース一括ダウンロード

ソースコード

diff #
#include<bits/stdc++.h>
#define rep(i,a,b) for(int i=a;i<b;i++)
#define rrep(i,a,b) for(int i=a;i>=b;i--)
#define fore(i,a) for(auto &i:a)
#define all(x) (x).begin(),(x).end()
//#pragma GCC optimize ("-O3")
using namespace std; void _main(); int main() { cin.tie(0); ios::sync_with_stdio(false); _main(); }
typedef long long ll; const int inf = INT_MAX / 2; const ll infl = 1LL << 60;
template<class T>bool chmax(T &a, const T &b) { if (a<b) { a = b; return 1; } return 0; }
template<class T>bool chmin(T &a, const T &b) { if (b<a) { a = b; return 1; } return 0; }
//---------------------------------------------------------------------------------------------------
/*---------------------------------------------------------------------------------------------------
            ∧_∧  
      ∧_∧  (´<_` )  Welcome to My Coding Space!
     ( ´_ゝ`) /  ⌒i     
    /   \     | |     
    /   / ̄ ̄ ̄ ̄/  |  
  __(__ニつ/     _/ .| .|____  
     \/____/ (u ⊃  
---------------------------------------------------------------------------------------------------*/








int N, M, L, T[2010];
vector<pair<int, int>> E[2010];
vector<ll> D[2010];
//---------------------------------------------------------------------------------------------------
int vis[101010];
template<typename T> using min_priority_queue = priority_queue<T, vector<T>, greater<T>>;
void dijk(int s, vector<ll> &D) {
    D.resize(N);
    rep(i, 0, N) D[i] = infl;
    rep(i, 0, N) vis[i] = 0;

    min_priority_queue<pair<ll, int>> que;

    D[s] = 0;
    que.push({ 0, s });

    while (!que.empty()) {
        auto q = que.top(); que.pop();

        ll cst = q.first;
        int cu = q.second;

        if (vis[cu]) continue;
        vis[cu] = 1;

        fore(p, E[cu]) {
            ll cst2 = cst + p.second;
            int to = p.first;

            if (chmin(D[to], cst2)) que.push({ D[to], to });
        }
    }
}
//---------------------------------------------------------------------------------------------------
void _main() {
    cin >> N >> M >> L; L--;
    rep(i, 0, N) cin >> T[i];
    rep(i, 0, M) {
        int a, b, c; cin >> a >> b >> c;
        a--; b--;
        E[a].push_back({ b,c });
        E[b].push_back({ a,c });
    }

    int cnt = 0;
    rep(i, 0, N) if (0 < T[i]) cnt++;
    if (cnt <= 1) {
        printf("0\n");
        return;
    }

    rep(i, 0, N) dijk(i, D[i]);

    ll ans = infl;
    rep(p, 0, N) {
        ll base = 0;
        rep(i, 0, N) base += T[i] * D[p][i] * 2;
        rep(st, 0, N) if(T[st]) {
            ll sm = D[L][st] + D[st][p];
            ll cand = base - D[p][st] * 2 + sm;
            chmin(ans, cand);
        }
    }
    cout << ans << endl;
}
0