結果
問題 | No.92 逃走経路 |
ユーザー |
|
提出日時 | 2020-06-12 12:08:41 |
言語 | C++17(gcc12) (gcc 12.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 8 ms / 5,000 ms |
コード長 | 3,613 bytes |
コンパイル時間 | 13,078 ms |
コンパイル使用メモリ | 298,236 KB |
最終ジャッジ日時 | 2025-01-11 01:46:31 |
ジャッジサーバーID (参考情報) |
judge4 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 18 |
ソースコード
#pragma GCC target("avx")#pragma GCC optimize("O3")#pragma GCC target("avx")#pragma GCC optimize("O3")#pragma GCC optimize("unroll-loops")#include <bits/stdc++.h>using namespace std;using ll = long long;using ld = long double;using VI = vector<int>;using VL = vector<ll>;using VS = vector<string>;template<class T> using PQ = priority_queue<T, vector<T>, greater<T>>;#define FOR(i,a,n) for(int i=(a);i<(n);++i)#define eFOR(i,a,n) for(int i=(a);i<=(n);++i)#define rFOR(i,a,n) for(int i=(n)-1;i>=(a);--i)#define erFOR(i,a,n) for(int i=(n);i>=(a);--i)#define each(i, a) for(auto &i : a)#define SORT(i) sort(i.begin(),i.end())#define rSORT(i) sort(i.rbegin(),i.rend())#define fSORT(i,a) sort(i.begin(),i.end(),a)#define all(i) i.begin(),i.end()#define out(y,x) ((y)<0||h<=(y)||(x)<0||w<=(x))#define line cout << "-----------------------------\n"#define ENDL(i,n) ((i) == (n) - 1 ? "\n" : " ")#define stop system("pause")constexpr ll INF = 1000000000;constexpr ll LLINF = 1LL << 60;constexpr ll mod = 1000000007;constexpr ll MOD = 998244353;constexpr ld eps = 1e-10;constexpr ld pi = 3.1415926535897932;template<class T>inline bool chmax(T& a, const T& b) { if (a < b) { a = b; return true; }return false; }template<class T>inline bool chmin(T& a, const T& b) { if (a > b) { a = b; return true; }return false; }inline void init() { cin.tie(nullptr); cout.tie(nullptr); ios::sync_with_stdio(false); cout << fixed << setprecision(15); }template<class T>inline istream& operator>>(istream& is, vector<T>& v) { for (auto& a : v)is >> a; return is; }template<class T>inline istream& operator>>(istream& is, deque<T>& v) { for (auto& a : v)is >> a; return is; }template<class T, class U>inline istream& operator>>(istream& is, pair<T, U>& p) { is >> p.first >> p.second; return is; }template<class T>inline vector<T> vec(size_t a) { return vector<T>(a); }template<class T>inline vector<T> defvec(T def, size_t a) { return vector<T>(a, def); }template<class T, class... Ts>inline auto vec(size_t a, Ts... ts) { return vector<decltype(vec<T>(ts...))>(a, vec<T>(ts...)); }template<class T, class... Ts>inline auto defvec(T def, size_t a, Ts... ts) { return vector<decltype(defvec<T>(def, ts...))>(a, defvec<T>(def, ts...)); }template<class T>inline void print(const T& a) { cout << a << "\n"; }template<class T, class... Ts>inline void print(const T& a, const Ts&... ts) { cout << a << " "; print(ts...); }template<class T>inline void print(const vector<T>& v) { for (int i = 0; i < v.size(); ++i)cout << v[i] << (i == v.size() - 1 ? "\n" : " "); }template<class T>inline void print(const vector<vector<T>>& v) { for (auto& a : v)print(a); }inline string reversed(const string& s) { string t = s; reverse(all(t)); return t; }int main() {init();int n, m, k; cin >> n >> m >> k;vector<vector<pair<int, int>>> g(n);FOR(i, 0, m) {int a, b, c; cin >> a >> b >> c;--a, --b;g[a].emplace_back(b, c);g[b].emplace_back(a, c);}VI d(k); cin >> d;auto dp = vec<bool>(n, k + 1);queue<pair<int, int>> bfs;FOR(i, 0, n) {dp[i][0] = true;bfs.emplace(i, 0);}while (!bfs.empty()) {auto [cur, num] = bfs.front();bfs.pop();each(dest, g[cur]) {auto [to, cost] = dest;if (num == k || cost != d[num] || dp[to][num + 1])continue;dp[to][num + 1] = true;bfs.emplace(to, num + 1);}}VI ans;FOR(i, 0, n)if (dp[i][k])ans.push_back(i + 1);print(ans.size());print(ans);return 0;}