Submission #718524

#	Time	Username	Problem	Language	Result	Execution time	Memory
718524		vinnipuh01	Robot (JOI21_ho_t4)	C++17	0 / 100	538 ms	52068 KiB

Main

#include <iostream>
#include <bits/stdc++.h>
#include <cmath>
#include <algorithm>
#include <vector>
#include <deque>
#include <set>
#include <stack>
#include <string>
#include <map>
#include <queue>
#define int long long
#define sqrt sqrtl

using namespace std;

const long long oo = 1000000000000000000;

long long sum, ans = 0, mx = 0, mn = 1000000000, num, pos;


/*
    ViHHiPuh

   (( `'-""``""-'` ))
     )-__-_.._-__-(
   / --- (o _ o) --- \
   \ .-* ( .0. ) *-. /
   _'-. ,_ '=' _, .-'_
  / `;#'#'# - #'#'#;` \
 \_)) -----'#'----- ((_/
      # --------- #
  '# ------- ------ #'
  /..-'# ------- #'-.\
  _\...-\'# -- #'/-.../_
  ((____)- '#' -(____))


    cout << fixed << setprecision(6) << x;


    freopen ( "sum.in", "r", stdin )
*/

struct Node {
	int dp, u, col;
};

vector <int> vv;

struct cmp {
	bool operator () ( const Node& a, const Node& b ) const {
		if ( a.dp == b.dp ) {
			if ( a.u == b.u ) {
				return a.col < b.col;
			}
			return a.u < b.u;
		}
		return a.dp < b.dp;
	}
};

int n, m, x, y, z[ 200001 ], t[ 200001 ];
map <int, int> mp[ 100001 ], mpp[ 200001 ];
int dp[ 200001 ];
vector <pair<int, int> > v[ 100001 ];
set <Node, cmp > st;


main () {
    ios_base::sync_with_stdio(0);
    cin.tie(0); cout.tie(0);
	cin >> n >> m;
	for ( int i = 1; i <= m; i ++ ) {
		cin >> x >> y >> z[ i ] >> t[ i ];
		mp[ x ][ z[ i ] ] += t[ i ];
		mp[ y ][ z[ i ] ] += t[ i ];
		v[ x ].push_back( { y, i } );
		v[ y ].push_back( { x, i } );
	}
	for ( int i = 1; i <= n; i ++ )
		dp[ i ] = oo;
	dp[ 1 ] = 0;
	st.insert( { 0, 1, 0 } );
	while ( st.size() ) {
		Node u = *st.begin();
		st.erase( st.begin() );
		num = pos = 0;
		if ( u.col ) {
			num = oo;
			for ( auto to : v[ u.u ] ) {
				if ( z[ to.second ] == u.col ) {
					num = min( num, dp[ to.first ] );
				}
			}
			for ( auto to : v[ u.u ] ) {
				if ( z[ to.second ] == u.col && dp[ to.first ] > num + mp[ u.u ][ z[ to.second ] ] - t[ to.second ] ) {
					st.erase( { dp[ to.first ], to.first, 0 } );
					dp[ to.first ] = num + mp[ u.u ][ z[ to.second ] ] - t[ to.second ];
					st.insert( { dp[ to.first ], to.first, 0 } );
				}
			}
		}
		for ( auto to : v[ u.u ] ) {
			if ( dp[ to.first ] > u.dp + t[ to.second ] ) {
				st.erase( { dp[ to.first ], to.first, z[ to.second ] } );
				dp[ to.first ] = u.dp + t[ to.second ];
				st.insert( { dp[ to.first ], to.first, z[ to.second ] } );
			}
			if ( dp[ to.first ] > u.dp + mp[ u.u ][ z[ to.second ] ] - t[ to.second ] ) {
				st.erase( { dp[ to.first ], to.first, 0 } );
				dp[ to.first ] = u.dp + mp[ u.u ][ z[ to.second ] ] - t[ to.second ];
				st.insert( { dp[ to.first ], to.first, 0 } );
			}
		}
	}
	mn = dp[ n ];
	if ( mn == oo )
		cout << "-1\n";
	else
		cout << mn << "\n";
}

Compilation message (stderr)

Main.cpp:70:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   70 | main () {
      | ^~~~

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