제출 #276524

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
276524		islingr	Olympic Bus (JOI20_ho_t4)	C++17	16 / 100	1087 ms	5184 KiB

ho_t4

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

#define rep(i, a, b) for (auto i = (a); i < (b); ++i)
#define all(x...) begin(x), end(x)

using ll = long long;

template<class T> bool ckmin(T &a, const T &b) { return a > b ? a = b, 1 : 0; }
template<class T> bool ckmax(T &a, const T &b) { return a < b ? a = b, 1 : 0; }

const int N = 1 << 9, M = 1 << 16;
const ll inf = 1e18;

int n;
ll d[N][N];
int a[M], b[M], w[M], c[M];
vector<int> g[N];
bool onPath[M], vis[N];

bool dfs(int u, int t) {
	if (u == t) return true;
	vis[u] = true;
	for (int e : g[u]) {
		assert(a[e] == u);
		int v = b[e];
		assert(d[u][t] <= w[e] + d[v][t]);
		if (!vis[v] && w[e] + d[v][t] == d[u][t] && dfs(v, t))
			return onPath[e] = true;
	}
	return false;
}

ll dist[N];
ll dij(int s, int t) {
	fill(dist, dist + n, inf);
	priority_queue<pair<ll, int>> pq;
	pq.push({0, s});
	while (!pq.empty()) {
		int u = pq.top().second;
		if (dist[u] == inf) {
			dist[u] = -pq.top().first;
			for (int e : g[u]) {
				int v = b[e];
				pq.push({-(dist[u] + w[e]), v});
			}
		} else assert(!ckmin(dist[u], -pq.top().first));
		pq.pop();
		if (u == t) break;
	}
	return dist[t];
}

signed main() {
	cin.tie(nullptr)->sync_with_stdio(false);

	int m; cin >> n >> m;
	rep(u, 0, n) rep(v, 0, n) d[u][v] = inf;
	rep(u, 0, n) d[u][u] = 0;
	rep(e, 0, m) {
		cin >> a[e] >> b[e] >> w[e] >> c[e], --a[e], --b[e];
		ckmin(d[a[e]][b[e]], ll(w[e]));
		g[a[e]].push_back(e);
	}

	rep(w, 0, n)
		rep(u, 0, n)
			rep(v, 0, n)
				ckmin(d[u][v], d[u][w] + d[w][v]);

	int s = 0, t = n - 1;
	fill(vis, vis + n, 0); dfs(s, t);
	fill(vis, vis + n, 0); dfs(t, s);

	ll ans = d[s][t] + d[t][s];
	rep(e, 0, m) {
		if (!onPath[e]) {
			ll to = min(d[s][t], d[s][b[e]] + w[e] + d[a[e]][t]);
			ll fr = min(d[t][s], d[t][b[e]] + w[e] + d[a[e]][s]);
			ckmin(ans, c[e] + to + fr);
		} else {
			g[a[e]].erase(find(all(g[a[e]]), e));
			swap(a[e], b[e]);
			g[a[e]].push_back(e);
			ckmin(ans, c[e] + dij(0, n - 1) + dij(n - 1, 0));
			g[a[e]].pop_back();
			swap(a[e], b[e]);
			g[a[e]].push_back(e);
		}
	}

	cout << (ans < inf ? ans : -1);
}

• Subtask 1: 5 / 5
• Subtask 2: 11 / 11
• Subtask 3: 0 / 21
• Subtask 4: 0 / 63