Submission #638841

#	Time	Username	Problem	Language	Result	Execution time	Memory
638841		sysia	Robot (JOI21_ho_t4)	C++17	0 / 100	409 ms	39360 KiB

Main

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

void __print(int x) {cerr << x;}
void __print(long long x) {cerr << x;}
void __print(long double x) {cerr << x;}
void __print(char x) {cerr << '\'' << x << '\'';}
void __print(const char *x) {cerr << '\"' << x << '\"';}
void __print(const string &x) {cerr << '\"' << x << '\"';}
void __print(bool x) {cerr << (x ? "true" : "false");}

template<typename T, typename V>
void __print(const pair<T, V> &x) {cerr << '{'; __print(x.first); cerr << ", "; __print(x.second); cerr << '}';}
template<typename T>
void __print(const T &x) {int f = 0; cerr << '{'; for (auto &i: x) cerr << (f++ ? ", " : ""), __print(i); cerr << "}";}
void _print() {cerr << "]\n";}
template <typename T, typename... V>
void _print(T t, V... v) {__print(t); if (sizeof...(v)) cerr << ", "; _print(v...);}
#ifndef ONLINE_JUDGE
#define debug(x...) cerr << "[" << #x << "] = ["; _print(x)
#else
#define debug(x...)
#endif

#define int long long
const int oo = 1e18, oo2 = 1e9+7, K = 30;
const int mod = 998244353;

struct T{
	int b, c, p;
	T(int _b = 0, int _c = 0, int _p = 0){
		b = _b, c = _c, p = _p;
	}
};

void solve(){
	int n, m; cin >> n >> m;
	vector<vector<T>>g(n+1);
	vector<map<int, pair<int, int>>>curr(n+1);
	for (int i = 0; i<m; i++){
		int a, b, c, p; cin >> a >> b >> c >> p;
		g[a].emplace_back(b, c, p);
		g[b].emplace_back(a, c, p);
		curr[a][c].first++;
		curr[a][c].second+=p;
		swap(a, b);
		curr[a][c].first++;
		curr[a][c].second+=p;
	}
	// for (int i = 1; i<=n; i++){
	// 	debug(i, curr[i]);
	// }
	set<pair<int, int>>s;
	vector<int>dist(n+1, oo);
	s.insert({0LL, 1});
	dist[1] = 0;
	while (s.size()){
		int d = s.begin()->first;
		int v = s.begin()->second;
		s.erase(s.begin());
		if (dist[v] < d) continue;
	
		for (auto [x, c, p]:g[v]){
			int cost = oo;
			if (curr[v][c].first == 1){
				cost = 0;
			} else if (curr[v].size() < m){
				cost = p;
			} else {
				for (auto u: curr[v]){
					int b = u.second.second;
					if (u.first == c) continue;
					cost = min(cost, b);
				}
				cost += p;
			}
			// debug(v, x, cost, dist[v]);
			if (dist[x] > dist[v] + cost){
				dist[x] = dist[v] + cost;
				s.insert({dist[x], x});
			}
		}
	}
	cout << (dist[n] == oo? -1:dist[n]) << "\n";
}

int32_t main(){
	ios_base::sync_with_stdio(0);
	cin.tie(0);
	cout.tie(0);

	solve();
	
	return 0;
}

Compilation message (stderr)

Main.cpp: In function 'void solve()':
Main.cpp:68:30: warning: comparison of integer expressions of different signedness: 'std::map<long long int, std::pair<long long int, long long int> >::size_type' {aka 'long unsigned int'} and 'long long int' [-Wsign-compare]
   68 |    } else if (curr[v].size() < m){
      |               ~~~~~~~~~~~~~~~^~~

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