#99421 (C++11) No.92 逃走経路

• 問題: No.92 逃走経路
• ユーザー: Tatamo
• 提出日時: 2016-06-22 11:37:40
• 言語: C++11 (gcc 11.4.0)
• 結果: RE
• 実行時間: -
• コード長: 1,523 bytes
• コンパイル時間: 899 ms
• コンパイル使用メモリ: 82,516 KB
• 実行使用メモリ: 5,248 KB
• 最終ジャッジ日時: 2024-10-11 19:16:55
• 合計ジャッジ時間: 5,610 ms
• ジャッジサーバーID （参考情報）: judge1 / judge3

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	21 ms; 5,248 KB
testcase_01	AC	1 ms; 5,248 KB
testcase_02	AC	1 ms; 5,248 KB
testcase_03	RE	-
testcase_04	RE	-
testcase_05	AC	1,703 ms; 5,248 KB
testcase_06	AC	8 ms; 5,248 KB
testcase_07	AC	7 ms; 5,248 KB
testcase_08	RE	-
testcase_09	RE	-
testcase_10	AC	159 ms; 5,248 KB
testcase_11	AC	121 ms; 5,248 KB
testcase_12	AC	906 ms; 5,248 KB
testcase_13	RE	-
testcase_14	AC	17 ms; 5,248 KB
testcase_15	AC	23 ms; 5,248 KB
testcase_16	AC	16 ms; 5,248 KB
testcase_17	AC	18 ms; 5,248 KB
testcase_18	AC	13 ms; 5,248 KB
testcase_19	AC	10 ms; 5,248 KB

ソースコード

#include <iostream>
#include <vector>
#include <utility>
#include <map>
#include <algorithm>

#define DBG(x) cerr << x << endl

using namespace std;

class Graph{
	public:
		vector<pair<int, int> >edges;
		Graph(){
		}
		void setEdge(int id, int cost){
			edges.push_back(make_pair(id, cost));
		}
};

vector<Graph> g;
vector<int> d;

map<int, Graph> search(){
	int count = 0;
	map<int, Graph> s;
	for(int i=1; i<=g.size(); i++){
		s.insert(map<int, Graph>::value_type(i, g[i]));
	}
	map<int, Graph> s_next;
	for(int c=0; c<d.size(); c++){
		for(auto itr = s.begin(); itr!=s.end(); itr++){
			for(int i=0; i < itr->second.edges.size(); i++){
				int cost = itr->second.edges[i].second;
				if(cost == d[count]){
					int index = itr->second.edges[i].first;
					s_next.insert(map<int, Graph>::value_type(index, g[index]));
				}
			}
		}
		s = s_next;
		s_next.clear();
		count += 1;
	}
	return s;
}

int main(){
	int n,m,k;
	cin >> n >> m >> k;

	for(int i=0; i<=m; i++){
		g.push_back(Graph());
	}
	for(int i=0; i<m; i++){
		int a, b, c;
		cin >> a >> b >> c;
		g[a].setEdge(b, c);
		g[b].setEdge(a, c);
	}

	for(int i=0; i<k; i++){
		int tmp;
		cin >> tmp;
		d.push_back(tmp);
	}

/*
	for(int i=0; i<g.size(); i++){
		for(int ii=0; ii<g[i].edges.size(); ii++){
			DBG(g[i].edges[ii].first);
			DBG(g[i].edges[ii].second);
			DBG("");
		}
	}
*/

	map<int, Graph> result = search();
	cout << result.size() << endl;
	for(auto itr=result.begin(); itr!=result.end(); itr++){
		cout << itr->first << " ";
	}
	cout << endl;

}