結果

問題 No.92 逃走経路
ユーザー kurenai3110
提出日時 2016-06-22 17:17:38
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
AC  
実行時間 506 ms / 5,000 ms
コード長 1,315 bytes
コンパイル時間 972 ms
コンパイル使用メモリ 85,516 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-10-11 19:20:16
合計ジャッジ時間 3,099 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <algorithm>
#include <map>
#include <iterator>
#include <set>
using namespace std;

int solve(int n, pair<int, int> &hfs);

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

	vector<pair<long, pair<int, int> > > h;
    h.resize(m);
	vector<long> c;
    c.resize(m);
    int a,b;
	for (int i = 0; i<m; i++) {
		cin >> a >> b >> c[i];
		h[i] = make_pair(c[i],make_pair(a,b));
	}
    sort(h.begin(),h.end());
    sort(c.begin(),c.end());

	long x;
	set<int> nr;
	for (int i = 0; i < k; i++) {
        set<int> r;
		cin >> x;
        int f = distance(c.begin(),lower_bound(c.begin(),c.end(),x));
		while (f<m&&h[f].first==x) {
			if (i == 0) {
				r.insert(h[f].second.first);
				r.insert(h[f].second.second);
			}
			else {
				int y;
				for (auto itnr = nr.begin();itnr!=nr.end();itnr++) {
					y = solve(*(itnr), h[f].second);
					if (y) {
						r.insert(y);
					}
				}
			}
            f++;
		}
        nr.clear();
        nr=r;
	}
    auto itnr = nr.begin(); 
	cout << nr.size() << endl;
	for (int i=0;i<nr.size(); i++) {
		if (i) cout << " ";
		cout << *(itnr);
        itnr++;
	}
	cout << endl;

	return 0;
}

int solve(int n, pair<int, int> &hfs) {
	if (n == hfs.first) {
		return hfs.second;
	}
	else if (n == hfs.second) {
		return hfs.first;
	}
	return 0;
}
0