結果

問題 No.92 逃走経路
コンテスト
ユーザー ty70
提出日時 2015-06-02 13:01:58
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
AC  
実行時間 194 ms / 5,000 ms
コード長 2,660 bytes
コンパイル時間 1,384 ms
コンパイル使用メモリ 104,568 KB
実行使用メモリ 19,584 KB
最終ジャッジ日時 2024-07-06 13:35:26
合計ジャッジ時間 2,587 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <string>
#include <stack>
#include <queue>
#include <deque>
#include <set>
#include <map>
#include <algorithm>	// require sort next_permutation count __gcd reverse etc.
#include <cstdlib>	// require abs exit atof atoi 
#include <cstdio>		// require scanf printf
#include <functional>
#include <numeric>	// require accumulate
#include <cmath>		// require fabs
#include <climits>
#include <limits>
#include <cfloat>
#include <iomanip>	// require setw
#include <sstream>	// require stringstream 
#include <cstring>	// require memset
#include <cctype>		// require tolower, toupper
#include <fstream>	// require freopen
#include <ctime>		// require srand
#define rep(i,n) for(int i=0;i<(n);i++)
#define ALL(A) A.begin(), A.end()

using namespace std;

typedef long long ll;
typedef pair<int, int> P;

/*
	No.92 逃走経路

	街 a, b 間の通行料金を c とすると
	multimap<int,pair<int,int> > cost で 通行料金 c とペアの街 (a,b), (b,a ) を保持。
	
	vector<P> cand[k] で 犯人が支払った通行料金と一致するペアの街 を保持。
	
	後は cand[i].second == cand[i+1].first となる経路を深さ優先探索で探索。
	結果を set<int> res に保持。

	計算量：O(M^K) やばい？ 1000^1000

	枝刈りしたら最悪ケースで大丈夫になった。

*/
const int MAX_N = 102;
const int MAX_K = 1005;
vector<P> cand[MAX_K];
bool visited[MAX_K][MAX_N];

set<int> res;

int N, M, K;

void dfs (int from, int depth ){

	if (depth == K ){
		res.insert (from );
		return;
	} // end if
	if (visited[depth][from] ) return;

	visited[depth][from] |= true;

	rep (j, cand[depth].size() ){
		P next = cand[depth][j];
		if (from == next.first ){
			dfs (next.second, depth + 1 );
	 	} // end if
	} // end rep
}

int main()
{
	memset (visited, false, sizeof (visited ) );
	ios_base::sync_with_stdio(0);
	cin >> N >> M >> K;
	multimap<int,P> cost; cost.clear();
	rep (i, M ){
		int a, b, c; cin >> a >> b >> c;
		cost.insert (make_pair(c, P (a, b ) ) );
		cost.insert (make_pair(c, P (b, a ) ) );
	} // end rep

	vector<int> d(K, 0 );
	rep (i, K ) cin >> d[i];

	rep (i, K ) cand[i].clear();
	multimap<int,P>::iterator it;
	rep (i, K ){
		int n = cost.count (d[i] );
		it = cost.find (d[i] );
		rep (j, n ){
			cand[i].push_back (it->second );
			it++;
		} // end rep
	} // end rep

	res.clear();
	rep (i, cand[0].size() ){
		int to = cand[0][i].second;
		dfs (to, 1 );
	} // end if

	cout << (int)res.size() << endl;
	set<int>::iterator jt = res.begin();
	for (; jt != res.end(); jt++ ){
		cout << (*jt) << ' ';
	} // end for
	if (!res.empty() )
		cout << endl;

	return 0;
}
0