結果
問題 |
No.2316 Freight Train
|
ユーザー |
|
提出日時 | 2023-05-27 14:47:22 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 493 ms / 2,000 ms |
コード長 | 582 bytes |
コンパイル時間 | 1,812 ms |
コンパイル使用メモリ | 194,364 KB |
最終ジャッジ日時 | 2025-02-13 09:17:27 |
ジャッジサーバーID (参考情報) |
judge3 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 26 |
ソースコード
#include <bits/stdc++.h> using namespace std; int par[200020]; int siz[200020]; int root(int a) { return (a == par[a] ? a : root(par[a])); } void merge(int a, int b) { a = root(a); b = root(b); if (a == b) return; if (siz[a] < siz[b]) swap(a, b); par[b] = a; siz[a] += siz[b]; } int main () { int N, Q; cin >> N >> Q; iota(par, par + N, 0); fill(siz, siz + N, 1); for (int i = 0; i < N; i ++) { int p; cin >> p; p --; if (p >= 0) { merge(p, i); } } while (Q--) { int a, b; cin >> a >> b; cout << (root(--a) == root(--b) ? "Yes" : "No") << endl; } }