結果

問題 No.17 2つの地点に泊まりたい
ユーザー torus711torus711
提出日時 2015-06-25 14:18:11
言語 C++11
(gcc 11.4.0)
結果
AC  
実行時間 123 ms / 5,000 ms
コード長 3,436 bytes
コンパイル時間 1,010 ms
コンパイル使用メモリ 109,096 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-04-23 02:05:09
合計ジャッジ時間 2,668 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
5,248 KB
testcase_01 AC 1 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 24 ms
5,376 KB
testcase_04 AC 7 ms
5,376 KB
testcase_05 AC 66 ms
5,376 KB
testcase_06 AC 34 ms
5,376 KB
testcase_07 AC 20 ms
5,376 KB
testcase_08 AC 123 ms
5,376 KB
testcase_09 AC 121 ms
5,376 KB
testcase_10 AC 54 ms
5,376 KB
testcase_11 AC 83 ms
5,376 KB
testcase_12 AC 2 ms
5,376 KB
testcase_13 AC 2 ms
5,376 KB
testcase_14 AC 1 ms
5,376 KB
testcase_15 AC 1 ms
5,376 KB
testcase_16 AC 2 ms
5,376 KB
testcase_17 AC 2 ms
5,376 KB
testcase_18 AC 13 ms
5,376 KB
testcase_19 AC 6 ms
5,376 KB
testcase_20 AC 2 ms
5,376 KB
testcase_21 AC 1 ms
5,376 KB
testcase_22 AC 19 ms
5,376 KB
testcase_23 AC 54 ms
5,376 KB
testcase_24 AC 64 ms
5,376 KB
testcase_25 AC 2 ms
5,376 KB
testcase_26 AC 99 ms
5,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <iomanip>
#include <sstream>
#include <vector>
#include <string>
#include <set>
#include <unordered_set>
#include <map>
#include <unordered_map>
#include <stack>
#include <queue>
#include <deque>
#include <algorithm>
#include <functional>
#include <iterator>
#include <limits>
#include <numeric>
#include <utility>
#include <cmath>
#include <cassert>
#include <cstdio>

using namespace std; using namespace placeholders;

using LL = long long;
using ULL = unsigned long long;
using VI = vector< int >;
using VVI = vector< vector< int > >;
using VS = vector< string >;
using SS = stringstream;
using PII = pair< int, int >;
using VPII = vector< pair< int, int > >;
template < typename T = int > using VT = vector< T >;
template < typename T = int > using VVT = vector< vector< T > >;
template < typename T = int > using LIM = numeric_limits< T >;

template < typename T > inline istream& operator>>( istream &s, vector< T > &v ){ for ( T &t : v ) { s >> t; } return s; }
template < typename T > inline ostream& operator<<( ostream &s, const vector< T > &v ){ for ( int i = 0; i < int( v.size() ); ++i ){ s << ( " " + !i ) << v[i]; } return s; }
template < typename T > inline T fromString( const string &s ) { T res; istringstream iss( s ); iss >> res; return res; };
template < typename T > inline string toString( const T &a ) { ostringstream oss; oss << a; return oss.str(); };

#define REP2( i, n ) REP3( i, 0, n )
#define REP3( i, m, n ) for ( int i = ( int )( m ); i < ( int )( n ); ++i )
#define GET_REP( a, b, c, F, ... ) F
#define REP( ... ) GET_REP( __VA_ARGS__, REP3, REP2 )( __VA_ARGS__ )
#define FOR( e, c ) for ( auto &e : c )
#define ALL( c ) ( c ).begin(), ( c ).end()
#define AALL( a, t ) ( t* )a, ( t* )a + sizeof( a ) / sizeof( t )
#define DRANGE( c, p ) ( c ).begin(), ( c ).begin() + ( p ), ( c ).end()

#define SZ( v ) ( (int)( v ).size() )
#define PB push_back
#define EM emplace
#define EB emplace_back
#define BI back_inserter

#define EXIST( c, e ) ( ( c ).find( e ) != ( c ).end() )

#define MP make_pair
#define fst first
#define snd second

#define DUMP( x ) cerr << #x << " = " << ( x ) << endl

constexpr int INF = LIM<>::max() / 2;

int distances[ 4 ][ 64 ];

int main()
{
	cin.tie( 0 );
	ios::sync_with_stdio( false );

	int N;
	cin >> N;

	VI S( N );
	cin >> S;

	int M;
	cin >> M;

	VT< VPII > G( N );
	REP( i, M )
	{
		int a, b, c;
		cin >> a >> b >> c;
		G[a].EB( b, c );
		G[b].EB( a, c );
	}


	int res = INF;
	REP( s1, 1, N - 1 )
	{
		REP( s2, 1, N  - 1 )
		{
			if ( s1 == s2 )
			{
				continue;
			}

			VI stay( { s1, s2 } );

			fill( AALL( distances, int ), INF );
			distances[0][0] = 0;

			using State = tuple< int, int, int >;
			priority_queue< State, VT< State >, greater< State > > que;
			que.EM( 0, 0, 0 );

			while ( !que.empty() )
			{
				const int dist = get< 0 >( que.top() );
				const int k = get< 1 >( que.top() );
				const int u = get< 2 >( que.top() );
				que.pop();

				if ( distances[k][u] < dist )
				{
					continue;
				}

				if ( k < 2 && u == stay[k] && dist + S[u] < distances[ k + 1 ][u] )
				{
					que.EM( distances[ k + 1 ][u] = dist + S[u], k + 1, u );
				}
				FOR( e, G[u] )
				{
					const int v = e.fst, d = e.snd;
					if ( dist + d < distances[k][v] )
					{
						que.EM( distances[k][v] = dist + d, k, v );
					}
				}
			}
			res = min( res, distances[2][ N - 1 ] );
		}
	}

	cout << res << endl;

	return 0;
}
0