結果
問題 | No.2316 Freight Train |
ユーザー |
![]() |
提出日時 | 2023-07-24 13:38:56 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 83 ms / 2,000 ms |
コード長 | 1,206 bytes |
コンパイル時間 | 2,075 ms |
コンパイル使用メモリ | 198,180 KB |
最終ジャッジ日時 | 2025-02-15 18:45:09 |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 26 |
ソースコード
#include <bits/stdc++.h>using namespace std;struct UnionFind {vector<int> par;vector<int> size;UnionFind() {}UnionFind(int n) {par.resize(n);size.resize(n,1);for(int i = 0; i < n; i++) {par[i] = i;}}int find(int x) {if(par[x] == x) {return x;}return par[x] = find(par[x]);}bool same(int x, int y) {return find(x) == find(y);}int consize(int x) {return size[find(x)];}bool unite(int x, int y) {x = find(x);y = find(y);if(x == y) {return false;}if(size[x] > size[y]) {swap(x,y);}par[x] = y;size[y] += size[x];return true;}};int main() {ios::sync_with_stdio(false);cin.tie(nullptr);int N,Q;cin >> N >> Q;UnionFind uf(N);for(int i = 0; i < N; i++) {int P;cin >> P;P--;if(P >= 0) uf.unite(i,P);}for(int i = 0; i < Q; i++) {int a,b;cin >> a >> b;a--;b--;cout << ((uf.same(a,b))?"Yes":"No") << "\n";}}