結果

問題 No.1812 Uribo Road
ユーザー trineutron
提出日時 2022-01-14 22:59:48
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 1,461 bytes
コンパイル時間 3,356 ms
コンパイル使用メモリ 272,252 KB
実行使用メモリ 484,272 KB
最終ジャッジ日時 2024-11-20 13:37:10
合計ジャッジ時間 71,293 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 20 TLE * 10
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

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

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;
    map<pair<int, int>, int> r_idx;
    vector<pair<int, int>> to[n];
    int idx = 0;
    for (int i = 0; i < m; i++)
    {
        int a, b, c;
        cin >> a >> b >> c;
        a--;
        b--;
        to[a].emplace_back(b, c);
        to[b].emplace_back(a, c);
        if (must.at(i))
        {
            idx++;
            r_idx[{a, b}] = idx;
            r_idx[{b, a}] = idx;
        }
    }
    vector dist(n, vector<int>(s, 2e9));
    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.at(v).at(seen))
        {
            continue;
        }
        dist.at(v).at(seen) = d0;
        for (auto &&[c, d_edge] : to[v])
        {
            int seen_new = seen;
            if (r_idx[{v, c}])
            {
                seen_new |= 1 << (r_idx[{v, c}] - 1);
            }
            if (d0 + d_edge >= dist.at(c).at(seen_new))
            {
                continue;
            }
            q.emplace(d0 + d_edge, c, seen_new);
        }
    }
    cout << dist.back().back() << endl;
    return 0;
}
0