結果
問題 | No.2316 Freight Train |
ユーザー | AnchorBlues |
提出日時 | 2023-09-29 14:30:57 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 85 ms / 2,000 ms |
コード長 | 4,438 bytes |
コンパイル時間 | 2,288 ms |
コンパイル使用メモリ | 207,856 KB |
実行使用メモリ | 6,944 KB |
最終ジャッジ日時 | 2024-07-22 08:42:52 |
合計ジャッジ時間 | 7,846 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge5 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
6,816 KB |
testcase_01 | AC | 2 ms
6,940 KB |
testcase_02 | AC | 2 ms
6,944 KB |
testcase_03 | AC | 85 ms
6,940 KB |
testcase_04 | AC | 43 ms
6,940 KB |
testcase_05 | AC | 32 ms
6,940 KB |
testcase_06 | AC | 10 ms
6,944 KB |
testcase_07 | AC | 55 ms
6,940 KB |
testcase_08 | AC | 53 ms
6,940 KB |
testcase_09 | AC | 57 ms
6,940 KB |
testcase_10 | AC | 57 ms
6,944 KB |
testcase_11 | AC | 60 ms
6,944 KB |
testcase_12 | AC | 68 ms
6,944 KB |
testcase_13 | AC | 82 ms
6,944 KB |
testcase_14 | AC | 79 ms
6,944 KB |
testcase_15 | AC | 81 ms
6,940 KB |
testcase_16 | AC | 81 ms
6,940 KB |
testcase_17 | AC | 85 ms
6,944 KB |
testcase_18 | AC | 84 ms
6,944 KB |
testcase_19 | AC | 83 ms
6,944 KB |
testcase_20 | AC | 84 ms
6,940 KB |
testcase_21 | AC | 81 ms
6,944 KB |
testcase_22 | AC | 79 ms
6,944 KB |
testcase_23 | AC | 58 ms
6,940 KB |
testcase_24 | AC | 65 ms
6,944 KB |
testcase_25 | AC | 59 ms
6,940 KB |
testcase_26 | AC | 56 ms
6,940 KB |
testcase_27 | AC | 31 ms
6,940 KB |
testcase_28 | AC | 2 ms
6,940 KB |
ソースコード
#include <bits/stdc++.h> using namespace std; using ll = long long; template <typename T> using min_priority_queue = priority_queue<T, vector<T>, greater<T>>; using pii = pair<int, int>; using pll = pair<ll, ll>; using Graph = vector<vector<int>>; const ll INF = 1LL << 60; template <class T> void chmax(T& a, T b) { if (b > a) a = b; } template <class T> void chmin(T& a, T b) { if (b < a) a = b; } template <typename T, typename S> std::ostream& operator<<(std::ostream& os, const pair<T, S>& x) noexcept { return os << "(" << x.first << ", " << x.second << ")"; } template <typename T> void print_vector(vector<T> a) { cout << '['; for (int i = 0; i < a.size(); i++) { cout << a[i]; if (i != a.size() - 1) { cout << ", "; } } cout << ']' << endl; } class UnionFind { public: // コンストラクタ UnionFind(); UnionFind(int); // 根を求める int root(int x) const; // x と y が同じグループに属するかどうか (根が一致するかどうか) bool issame(int x, int y) const; // x を含むグループと y を含むグループとを併合する bool unite(int x, int y); // 要素の追加。まだ追加されていない要素だったらその要素だけから成る集合を作る void add_elem(int x); // x を含むグループのサイズ int size(int x) const; // 島の個数 int n_unions() const; // 要素xがすでに追加されているかどうかを判定 bool is_added(int x) const; private: // 経路圧縮する。根を返す int path_compression(int x); std::vector<int> parents; std::vector<int> sizes; }; // コンストラクタ UnionFind::UnionFind() {} UnionFind::UnionFind(int N) { parents = std::vector<int>(N, -1); sizes = std::vector<int>(N, -1); } // 根を求める int UnionFind::root(int x) const { if (parents[x] == -1) return x; // x が根の場合は x を返す return root(parents[x]); } // x と y が同じグループに属するかどうか (根が一致するかどうか) bool UnionFind::issame(int x, int y) const { return root(x) == root(y); } // x を含むグループと y を含むグループとを併合する bool UnionFind::unite(int x, int y) { // x, y をそれぞれ根まで移動する if (!is_added(x)) sizes[x] = 1; if (!is_added(y)) sizes[y] = 1; int root_x = path_compression(x); int root_y = path_compression(y); // すでに同じグループのときは何もしない if (root_x == root_y) return false; if (!is_added(root_x)) sizes[root_x] = 1; if (!is_added(root_y)) sizes[root_y] = 1; // union by size (y 側のサイズが小さくなるようにする) if (sizes[root_x] < sizes[root_y]) std::swap(root_x, root_y); // y を x の子とする parents[root_y] = root_x; sizes[root_x] += sizes[root_y]; return true; } void UnionFind::add_elem(int x) { // すでに追加されていたら何もしない if (is_added(x)) return; sizes[x] = 1; } // x を含むグループのサイズ。x が追加されていない要素の場合は「1」を返す int UnionFind::size(int x) const { if (is_added(x)) return sizes[root(x)]; return 1; } // 島の個数 int UnionFind::n_unions() const { std::unordered_set<int> s; for (size_t i = 0; i < sizes.size(); i++) if (is_added(i)) s.insert(root(i)); return s.size(); } // 経路圧縮をしながら根を求める int UnionFind::path_compression(int x) { if (parents[x] == -1) return x; // x が根の場合は x を返す parents[x] = path_compression(parents[x]); return parents[x]; } bool UnionFind::is_added(int x) const { return sizes[x] != -1; } void solve() { ll N, Q; cin >> N >> Q; auto uf = UnionFind(N); for (int i = 0; i < N; i++) { ll p; cin >> p; if (p == -1) continue; p--; uf.unite(i, p); } while (Q--) { ll a, b; cin >> a >> b; a--; b--; if (uf.issame(a, b)) { std::cout << "Yes" << "\n"; } else { std::cout << "No" << "\n"; } } } int main() { ios::sync_with_stdio(false); std::cin.tie(nullptr); int T = 1; while (T--) { solve(); } return 0; }