結果
問題 | No.2316 Freight Train |
ユーザー |
|
提出日時 | 2023-10-25 23:51:31 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 447 ms / 2,000 ms |
コード長 | 1,415 bytes |
コンパイル時間 | 1,817 ms |
コンパイル使用メモリ | 172,156 KB |
実行使用メモリ | 6,944 KB |
最終ジャッジ日時 | 2024-09-25 07:55:17 |
合計ジャッジ時間 | 13,740 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 26 |
ソースコード
#include<bits/stdc++.h>using namespace std;using ll = long long;#define rep(i,m,n) for(int i=m; i<n; ++i)#define repl(i,m,n) for(ll i=m; i<n; ++i)struct UnionFind{vector<int> size, parent;UnionFind(){}UnionFind(int n){size.resize(n,0);parent.resize(n,0);for(int i=0; i<n; ++i){make_tree(i);}}void make_tree(int x){parent[x] = x;size[x] = 1;}bool is_same(int x, int y){return find_root(x) == find_root(y);}bool unite(int x, int y){x = find_root(x);y = find_root(y);if(x == y) return false;if(size[x] > size[y]){parent[y] = x;size[x] += size[y];}else{parent[x] = y;size[y] += size[x];}return true;}int find_root(int x){if(x != parent[x]) parent[x] = find_root(parent[x]);return parent[x];}int tree_size(int x){return size[find_root(x)];}};int main(){int N, Q;cin >> N >> Q;UnionFind uf(N);rep(i, 0, N){int P;cin >> P;if(P == -1) continue;uf.unite(i+1, P);}vector<int> A(Q), B(Q);rep(i, 0, Q) cin >> A[i] >> B[i];rep(i, 0, Q){bool f = uf.is_same(A[i], B[i]);cout << (f ? "Yes" : "No") << endl;}return 0;}