結果
問題 | No.2316 Freight Train |
ユーザー | nagisa5101 |
提出日時 | 2023-05-26 21:25:54 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 315 ms / 2,000 ms |
コード長 | 2,530 bytes |
コンパイル時間 | 4,106 ms |
コンパイル使用メモリ | 263,396 KB |
実行使用メモリ | 4,788 KB |
最終ジャッジ日時 | 2023-08-26 10:05:42 |
合計ジャッジ時間 | 13,011 ms |
ジャッジサーバーID (参考情報) |
judge12 / judge15 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 1 ms
4,376 KB |
testcase_01 | AC | 2 ms
4,380 KB |
testcase_02 | AC | 2 ms
4,376 KB |
testcase_03 | AC | 299 ms
4,584 KB |
testcase_04 | AC | 168 ms
4,376 KB |
testcase_05 | AC | 110 ms
4,380 KB |
testcase_06 | AC | 39 ms
4,380 KB |
testcase_07 | AC | 269 ms
4,376 KB |
testcase_08 | AC | 158 ms
4,592 KB |
testcase_09 | AC | 222 ms
4,376 KB |
testcase_10 | AC | 246 ms
4,376 KB |
testcase_11 | AC | 201 ms
4,420 KB |
testcase_12 | AC | 257 ms
4,380 KB |
testcase_13 | AC | 301 ms
4,616 KB |
testcase_14 | AC | 305 ms
4,696 KB |
testcase_15 | AC | 298 ms
4,568 KB |
testcase_16 | AC | 309 ms
4,788 KB |
testcase_17 | AC | 306 ms
4,616 KB |
testcase_18 | AC | 301 ms
4,628 KB |
testcase_19 | AC | 304 ms
4,576 KB |
testcase_20 | AC | 315 ms
4,636 KB |
testcase_21 | AC | 307 ms
4,576 KB |
testcase_22 | AC | 305 ms
4,620 KB |
testcase_23 | AC | 285 ms
4,632 KB |
testcase_24 | AC | 294 ms
4,628 KB |
testcase_25 | AC | 272 ms
4,692 KB |
testcase_26 | AC | 277 ms
4,684 KB |
testcase_27 | AC | 265 ms
4,380 KB |
testcase_28 | AC | 2 ms
4,376 KB |
ソースコード
#include <bits/stdc++.h> #include <atcoder/all> #pragma GCC target("avx2") #pragma GCC optimize("O3") #pragma GCC optimize("unroll-loops") using namespace std; using namespace atcoder; #define rep(i, n) for (int i = 0; i < (int)(n); i++) #define repll(i, n) for (long long i = 0; i < (long long)(n); i++) #define rep2(i, n, m) for (int i = n; i < (int)(m); i++) #define repll2(i, n, m) for (long long i = n; i < (long long)(m); i++) #define all(v) v.begin(),v.end() using ll=long long; using ld=long double; using vi=vector<int>; using vvi=vector<vi>; using vvvi=vector<vvi>; using vl=vector<ll>; using vvl=vector<vl>; using vvvl=vector<vvl>; using vld=vector<ld>; using vvld=vector<vld>; int dx[8]={1,0,-1,0,1,1,-1,-1}; int dy[8]={0,1,0,-1,1,-1,1,-1}; const double PI = acos(-1); //const ll MOD=1e9+7; //const ll MOD=998244353; const ll INF=(1LL<<60); const int INF2=(1<<30); //using mint=modint1000000007; //using mint=modint998244353; /* UnionFind:素集合系管理の構造体(union by size) isSame(x, y): x と y が同じ集合にいるか。 計算量はならし O(α(n)) unite(x, y): x と y を同じ集合にする。計算量はならし O(α(n)) treeSize(x): x を含む集合の要素数。 */ struct UnionFind { vector<int> size, parents; UnionFind() {} UnionFind(int n) { // make n trees. size.resize(n, 0); parents.resize(n, 0); for (int i = 0; i < n; i++) { makeTree(i); } } void makeTree(int x) { parents[x] = x; // the parent of x is x size[x] = 1; } bool isSame(int x, int y) { return findRoot(x) == findRoot(y); } bool unite(int x, int y) { x = findRoot(x); y = findRoot(y); if (x == y) return false; if (size[x] > size[y]) { parents[y] = x; size[x] += size[y]; } else { parents[x] = y; size[y] += size[x]; } return true; } int findRoot(int x) { if (x != parents[x]) { parents[x] = findRoot(parents[x]); } return parents[x]; } int treeSize(int x) { return size[findRoot(x)]; } }; int main() { ios::sync_with_stdio(false); std::cin.tie(nullptr); int n,q;cin>>n>>q; UnionFind uf(n); rep(i,n){ int p;cin>>p; if(p==-1)continue; uf.unite(i,p-1); } while(q--){ int a,b;cin>>a>>b; if(uf.isSame(a-1,b-1))cout<<"Yes"<<endl; else cout<<"No"<<endl; } return 0; }