結果

問題 No.1812 Uribo Road
コンテスト
ユーザー trineutron
提出日時 2022-01-14 23:16:18
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 1,431 bytes
コンパイル時間 3,640 ms
コンパイル使用メモリ 265,772 KB
実行使用メモリ 997,472 KB
最終ジャッジ日時 2024-11-20 14:17:22
合計ジャッジ時間 77,210 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 18 TLE * 11 MLE * 1
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

using edge = tuple<int64_t, int64_t, int64_t>;

int main()
{
    int n, m, k;
    cin >> n >> m >> k;
    vector<bool> must(m);
    for (int i = 0; i < k; i++)
    {
        int r;
        cin >> r;
        r--;
        must.at(r) = true;
    }
    const int s = 1 << k;
    vector<edge> to[n];
    int idx = 0;
    for (int i = 0; i < m; i++)
    {
        int a, b, c;
        cin >> a >> b >> c;
        a--;
        b--;
        if (must.at(i))
        {
            idx++;
            to[a].emplace_back(b, c, idx);
            to[b].emplace_back(a, c, idx);
        }
        to[a].emplace_back(b, c, 0);
        to[b].emplace_back(a, c, 0);
    }
    vector dist(n, vector<int64_t>(s, 1e18));
    priority_queue<edge, vector<edge>, greater<edge>> q;
    q.emplace(0, 0, 0);
    while (not q.empty())
    {
        auto [d0, v, seen] = q.top();
        q.pop();
        if (d0 >= dist[v][seen])
        {
            continue;
        }
        dist[v][seen] = d0;
        for (auto &&[c, d_edge, r_idx] : to[v])
        {
            int seen_new = seen;
            if (r_idx)
            {
                seen_new |= 1 << (r_idx - 1);
            }
            if (d0 + d_edge >= dist[c][seen_new])
            {
                continue;
            }
            q.emplace(d0 + d_edge, c, seen_new);
        }
    }
    cout << dist.back().back() << endl;
    return 0;
}
0