結果
問題 |
No.1023 Cyclic Tour
|
ユーザー |
![]() |
提出日時 | 2020-04-11 18:15:27 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 217 ms / 2,000 ms |
コード長 | 4,297 bytes |
コンパイル時間 | 1,219 ms |
コンパイル使用メモリ | 95,388 KB |
実行使用メモリ | 33,920 KB |
最終ジャッジ日時 | 2024-09-19 05:51:28 |
合計ジャッジ時間 | 9,309 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 49 |
ソースコード
#include<iostream> #include<vector> #include<string> #include<set> #include<map> #define rep(i, start, end) for (int i = (int)start; i < (int)end; ++i) #define rrep(i, start, end) for (int i = (int)start - 1; i >= (int)end; --i) #define all(x) (x).begin(), (x).end() using namespace std; using ll = long long; template<typename T> inline bool chmax(T& a, T b) {if (a < b) {a = b; return true;} return 0;} template<typename T> inline bool chmin(T& a, T b) {if (a > b) {a = b; return true;} return 0;} class UnionFind { private: vector<int> parent_; vector<int> node_rank_; vector<int> sizes_; public: UnionFind(int node_num): parent_(vector<int>(node_num)), node_rank_(vector<int>(node_num)), sizes_(vector<int>(node_num)) { for (int i = 0; i < node_num; ++i) { parent_[i] = i; node_rank_[i] = 0; sizes_[i] = 1; } } int getRoot(int u) { return parent_[u] == u ? u : parent_[u] = getRoot(parent_[u]); } bool isSame(int u, int v) { return getRoot(u) == getRoot(v); } void unite(int u, int v) { u = getRoot(u); v = getRoot(v); if (u == v) return; if (node_rank_[u] < node_rank_[v]) { parent_[u] = v; sizes_[v] += sizes_[u]; } else { parent_[v] = u; sizes_[u] += sizes_[v]; if (node_rank_[u] == node_rank_[v]) { node_rank_[u]++; } } } int getSize(int u) { return sizes_[getRoot(u)]; } }; bool dfs1(const vector<vector<int>>& graph, vector<bool>& seen, vector<bool>& finished, int node, int parent) { seen[node] = true; for (auto& next_node : graph[node]) { if (next_node == parent) { continue; } if (finished[next_node]) { continue; } if (seen[next_node] && !finished[next_node]) { return true; } if (dfs1(graph, seen, finished, next_node, node)) { return true; } } finished[node] = true; return false; } bool dfs2(const vector<vector<int>>& graph, vector<bool>& seen, vector<bool>& finished, int node) { seen[node] = true; for (auto& next_node : graph[node]) { if (finished[next_node]) { continue; } if (seen[next_node] && !finished[next_node]) { return true; } if (dfs2(graph, seen, finished, next_node)) { return true; } } finished[node] = true; return false; } int main() { cin.tie(0); ios::sync_with_stdio(false); int N, M; cin >> N >> M; vector<int> A(M), B(M), C(M); rep(i, 0, M) { cin >> A[i] >> B[i] >> C[i]; --A[i], --B[i]; } // 無向辺のみで構築されるグラフに閉路があるかを確かめる vector<vector<int>> graph1(N); rep(i, 0, M) { if (C[i] == 1) { graph1[A[i]].push_back(B[i]); graph1[B[i]].push_back(A[i]); } } vector<bool> seen1(N, false), finished1(N, false); rep(u, 0, N) { if (!seen1[u] && dfs1(graph1, seen1, finished1, u, -1)) { cout << "Yes" << endl; return 0; } } // 無向辺で連結される頂点をまとめる UnionFind uf(N); rep(i, 0, M) { if (C[i] == 1) { uf.unite(A[i], B[i]); } } set<int> S; rep(i, 0, N) { S.insert(uf.getRoot(i)); } map<int, int> node_map; int node_num = 0; for (auto s : S) { node_map[s] = node_num++; } // まとめた頂点と有向辺でグラフを構築する vector<vector<int>> graph2(node_num); rep(i, 0, M) { if (C[i] == 2) { int u = node_map[uf.getRoot(A[i])]; int v = node_map[uf.getRoot(B[i])]; graph2[u].push_back(v); } } // 新しく構築したグラフにおいて閉路検出を行う vector<bool> seen2(node_num, false); vector<bool> finished2(node_num, false); rep(u, 0, node_num) { if (!seen2[u] && dfs2(graph2, seen2, finished2, u)) { cout << "Yes" << endl; return 0; } } cout << "No" << endl; return 0; }