結果

問題 No.3111 Toll Optimization
ユーザー cho435
提出日時 2025-04-19 02:12:31
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,310 bytes
コンパイル時間 2,628 ms
コンパイル使用メモリ 209,604 KB
実行使用メモリ 15,044 KB
最終ジャッジ日時 2025-04-19 02:12:40
合計ジャッジ時間 8,067 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 47 WA * 23
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
#define rep(i, s, t) for (ll i = s; i < (ll)(t); i++)
#define all(x) begin(x), end(x)

template <typename T> bool chmin(T& x, T y) {
	return x > y ? (x = y, true) : false;
}
template <typename T> bool chmax(T& x, T y) {
	return x < y ? (x = y, true) : false;
}

struct IOST {
	IOST() {
		ios::sync_with_stdio(false);
		cin.tie(nullptr);
		cout << fixed << setprecision(20);
	}
} IOST;

void solve() {
	int n, m, k;
	cin >> n >> m >> k;
	vector<int> c(m);
	rep(i, 0, m) cin >> c[i];
	vector<vector<pair<int, int>>> g(n);
	rep(i, 0, m) {
		int u, v;
		cin >> u >> v;
		u--;
		v--;
		g[u].push_back({v, c[i]});
		g[v].push_back({u, c[i]});
	}
	priority_queue<array<ll, 3>, vector<array<ll, 3>>, greater<array<ll, 3>>>
		pq;
	pq.push({0, 0, 0});
	vector dp(k + 1, vector<ll>(n, 1e18));
	while (!pq.empty()) {
		auto [cs, nk, nw] = pq.top();
		pq.pop();
		if (cs > dp[nk][nw]) break;
		for (auto [nx, ad] : g[nw]) {
			if (nk < k) {
				if (chmin(dp[nk + 1][nx], cs)) pq.push({cs, nk + 1, nx});
			}
			if (chmin(dp[nk][nx], cs + ad)) pq.push({cs + ad, nk, nx});
		}
	}
	ll ans = 1e18;
	rep(i, 0, k + 1) chmin(ans, dp[i][n - 1]);
	if (ans == 1e18) ans = -1;
	cout << ans << "\n";
}

int main() {
	int t = 1;
	// cin >> t;
	rep(i, 0, t) solve();
}
0