제출 #1005575

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1005575		c2zi6	Commuter Pass (JOI18_commuter_pass)	C++14	55 / 100	192 ms	20172 KiB

commuter_pass

#define _USE_MATH_DEFINES
#include <bits/stdc++.h>
#define ff first
#define ss second
#define pb push_back
#define all(a) (a).begin(), (a).end()
#define replr(i, a, b) for (int i = int(a); i <= int(b); ++i)
#define reprl(i, a, b) for (int i = int(a); i >= int(b); --i)
#define rep(i, n) for (int i = 0; i < int(n); ++i)
#define mkp(a, b) make_pair(a, b)
using namespace std;
typedef long long ll;
typedef long double ld;
typedef pair<int, int> PII;
typedef vector<int> VI;
typedef vector<PII> VPI;
typedef vector<VI> VVI;
typedef vector<VVI> VVVI;
typedef vector<VPI> VVPI;
typedef pair<ll, ll> PLL;
typedef vector<ll> VL;
typedef vector<PLL> VPL;
typedef vector<VL> VVL;
typedef vector<VVL> VVVL;
typedef vector<VPL> VVPL;
template<class T> T setmax(T& a, T b) {if (a < b) return a = b; return a;}
template<class T> T setmin(T& a, T b) {if (a < b) return a; return a = b;}
#include <ext/pb_ds/assoc_container.hpp>
using namespace __gnu_pbds;
template<class T>
using indset = tree<T, null_type, less<T>, rb_tree_tag, tree_order_statistics_node_update>;

int S, T, U, V;
int n, m;
VVPI gp;

VI dijkstra(int s, VL& dist, int to = -1) {
    dist = VL(n, 1e18);
    VI prev(n, -1);
    priority_queue<PLL> pq;
    VI vis(n);
    dist[s] = 0;
    pq.push({0, s});
    while (pq.size()) {
        int u = pq.top().ss;
        pq.pop();
        if (vis[u]) continue;
        vis[u] = true;
        for (auto[v, w] : gp[u]) {
            if (dist[u] + w < dist[v]) {
                prev[v] = u;
                dist[v] = dist[u] + w;
                pq.push({-dist[v], v});
            }
        }
    }
    if (to == -1) return VI();
    VI route;
    while (to != s) {
        route.pb(to);
        to = prev[to];
    }
    route.pb(to);
    reverse(all(route));
    return route;
}

void solve() {
    cin >> n >> m;
    cin >> S >> T >> U >> V;
    S--, T--, U--, V--;
    gp = VVPI(n);
    rep(i, m) {
        int u, v, c;
        cin >> u >> v >> c;
        u--, v--;
        gp[u].pb({v, c});
        gp[v].pb({u, c});
    }
    VL dists, distt, distu, distv;
    VI route = dijkstra(S, dists, T);
    dijkstra(T, distt);
    dijkstra(U, distu);
    dijkstra(V, distv);
    if (U == S) {
        ll mindist = dists[T];
        ll ans = 1e18;
        rep(x, n) {
            /*if laying on mindist route*/
            if (dists[x] + distt[x] == mindist) {
                setmin(ans, distv[x]);
            }
        }
        cout << ans << endl;
    } else if (n <= 300) {
        ll mindist = dists[T];
        ll ans = distu[V];
        VL buf;
        rep(x, n) if (dists[x] + distt[x] == mindist) {
            /*cout << "SOLVING FOR " << x+1 << " " << T+1 << endl;*/
            VL distx;
            dijkstra(x, distx);
            ll mindist = distx[T];
            rep(y, n) {
                if (distx[y] + distt[y] == mindist) {
                    setmin(ans, distu[x] + distv[y]);
                    setmin(ans, distu[y] + distv[x]);
                }
            }
        }
        cout << ans << endl;
    } else {
        ll ans = distu[V];
        ll mnin = 1e18;
        ll mnout = 1e18;
        for (int x : route) {
            setmin(mnin, distu[x]);
            setmin(mnout, distv[x]);
        }
        cout << min(ans, mnin + mnout) << endl;
    }
}

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL); cout.tie(NULL);
    int t = 1;
    /*cin >> t;*/
    while (t--) solve();
}

컴파일 시 표준 에러 (stderr) 메시지

commuter_pass.cpp: In function 'VI dijkstra(int, VL&, int)':
commuter_pass.cpp:49:18: warning: structured bindings only available with '-std=c++17' or '-std=gnu++17'
   49 |         for (auto[v, w] : gp[u]) {
      |                  ^

• Subtask 1: 16 / 16
• Subtask 2: 15 / 15
• Subtask 3: 24 / 24
• Subtask 4: 0 / 45