結果
問題 | No.2316 Freight Train |
ユーザー |
|
提出日時 | 2023-05-26 21:29:04 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 86 ms / 2,000 ms |
コード長 | 1,410 bytes |
コンパイル時間 | 774 ms |
コンパイル使用メモリ | 77,664 KB |
最終ジャッジ日時 | 2025-02-13 05:50:09 |
ジャッジサーバーID (参考情報) |
judge3 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 26 |
ソースコード
#include<iostream> #include<vector> #include<algorithm> using namespace std; using ll=long long; #define rep(i,n) for(int i=0;i<n;i++) #define rrep(i,n) for(int i=(n)-1;i>=0;i--) #define all(v) v.begin(),v.end() #define rall(v) v.rbegin(),v.rend() template<class T> bool chmax(T &a, T b){if (a < b){a = b;return true;} else return false;} template<class T> bool chmin(T &a, T b){if (a > b){a = b;return true;} else return false;} struct UnionFind { vector<int> par, siz; UnionFind(int n) : par(n,-1), siz(n,1) {} int find(int x){ if (par[x] == -1) return x; else{ par[x] = find(par[x]); return par[x]; } } bool same(int x, int y){ return find(x) == find(y); } int size(int x){ return siz[find(x)]; } void merge(int x, int y){ x = find(x); y = find(y); if (x == y) return; if (siz[x] < siz[y]) swap(x,y); siz[x] += siz[y]; par[y] = x; } }; int main(){ ios::sync_with_stdio(false); cin.tie(nullptr); int N,Q; cin>>N>>Q; vector<int>p(N); UnionFind uf(N); rep(i,N){ cin>>p[i]; if(p[i]!=-1){ p[i]--; uf.merge(i,p[i]); } } while(Q--){ int a,b; cin>>a>>b; a--; b--; if(uf.same(a,b))cout<<"Yes"<<"\n"; else cout<<"No"<<"\n"; } }