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

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

入力	結果	実行時間; 実行使用メモリ
testcase_00	WA	-
testcase_01	RE	-
testcase_02	RE	-
testcase_03	AC	1 ms; 5,248 KB
testcase_04	AC	2 ms; 5,248 KB
testcase_05	AC	891 ms; 5,248 KB
testcase_06	AC	8 ms; 5,248 KB
testcase_07	AC	8 ms; 5,248 KB
testcase_08	AC	2 ms; 5,248 KB
testcase_09	AC	13 ms; 5,248 KB
testcase_10	AC	123 ms; 5,248 KB
testcase_11	AC	103 ms; 5,248 KB
testcase_12	AC	505 ms; 5,248 KB
testcase_13	AC	7 ms; 5,248 KB
testcase_14	WA	-
testcase_15	WA	-
testcase_16	WA	-
testcase_17	WA	-
testcase_18	WA	-
testcase_19	WA	-

ソースコード

#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();
	}
	return s;
}

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

	for(int i=0; i<=n; 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;
}