제출 #1265400

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1265400		kustov_vadim_533	Robot (JOI21_ho_t4)	C++17	0 / 100	3093 ms	48908 KiB

Main

#include <bits/stdc++.h>

using namespace std;

typedef long long ll;
typedef long double ld;
typedef unsigned long long ull;

#define len(v) (int)((v).size())

const ll inf = 1e18;

#define int ll

struct edge{
	int to, cl, w;
	edge(int to, int cl, int w) : to(to), cl(cl), w(w) {};
};

bool operator<(edge a, edge b){
	if (a.w != b.w) return a.w < b.w;
	if (a.to != b.to) return a.to < b.to;
	return a.cl < b.cl;
}

inline void solve(){
	int n, m;
	cin >> n >> m;

	vector<vector<edge>> g(n);
	vector<map<int, ll>> cnt2(n);

	for (int i = 0; i < m; ++i){
		int a, b, c, p;
		cin >> a >> b >> c >> p;
		--a, --b, --c;

		cnt2[a][c] += p;
		cnt2[b][c] += p;

		g[a].emplace_back(b, c, p);
		g[b].emplace_back(a, c, p);
	}

	vector<map<pair<int, int>, ll>> d(n);
	for (int i = 0; i < n; ++i){
		for (edge u : g[i]){
			d[i][{u.cl, u.w}] = inf;
			d[i][{u.cl, 0}] = inf;
		}
	}

	set<pair<ll, edge>> q;
	for (int i = 0; i < m; ++i){
		d[0][{i, 0}] = 0;
		q.insert({0, edge(0, i, 0)});
	}

	while (!q.empty()){
		int v = q.begin()->second.to;
		int c = q.begin()->second.cl;
		int qw = q.begin()->second.w;

		q.erase(q.begin());

		for (edge u : g[v]){
			ll w = cnt2[v][u.cl] - u.w + d[v][{c, qw}];
			if (u.cl == c) w -= qw;
			if ((u.cl != c || qw > 0) && d[u.to][{u.cl, 0}] > w && w >= d[v][{c, qw}]){
				pair<ll, edge> obj = {d[u.to][{u.cl, 0}], edge(u.to, u.cl, 0)};
				if (q.find(obj) != q.end()){
					q.erase(obj);
				}
				d[u.to][{u.cl, 0}] = w;
				q.insert(obj);
			}


			w = d[v][{c, qw}] + u.w;
			if (d[u.to][{u.cl, u.w}] > w){
				pair<ll, edge> obj = {d[u.to][{u.cl, u.w}], u};
				if (q.find(obj) != q.end()){
					q.erase(obj);
				}
				d[u.to][{u.cl, u.w}] = w;
				q.insert(obj);
			}
		}
	}


	ll ans = inf;
	for (pair<pair<int, int>, ll> p : d[n - 1]){
		ans = min(ans, p.second);
	}

	if (ans >= inf){
		cout << "-1\n";
	}
	else{
		cout << ans << '\n';
	}

}

signed main() {
	ios_base::sync_with_stdio(false);
	cin.tie(nullptr);
	cout.tie(nullptr);
	cout.precision(60);

	int t = 1;
//	cin >> t;

	while (t--) {
		solve();
	}
}

• Subtask 1: 0 / 34
• Subtask 2: 0 / 24
• Subtask 3: 0 / 42