結果

問題 No.2316 Freight Train
ユーザー Be11T_Be11T_
提出日時 2023-05-26 21:27:04
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 340 ms / 2,000 ms
コード長 2,949 bytes
コンパイル時間 2,351 ms
コンパイル使用メモリ 208,380 KB
実行使用メモリ 4,784 KB
最終ジャッジ日時 2023-08-26 10:10:36
合計ジャッジ時間 11,731 ms
ジャッジサーバーID
（参考情報） 		judge11 / judge13
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
4,380 KB
testcase_01 AC 2 ms
4,384 KB
testcase_02 AC 2 ms
4,376 KB
testcase_03 AC 328 ms
4,612 KB
testcase_04 AC 175 ms
4,376 KB
testcase_05 AC 121 ms
4,376 KB
testcase_06 AC 44 ms
4,376 KB
testcase_07 AC 294 ms
4,376 KB
testcase_08 AC 176 ms
4,544 KB
testcase_09 AC 246 ms
4,376 KB
testcase_10 AC 269 ms
4,380 KB
testcase_11 AC 223 ms
4,380 KB
testcase_12 AC 281 ms
4,424 KB
testcase_13 AC 333 ms
4,600 KB
testcase_14 AC 337 ms
4,780 KB
testcase_15 AC 330 ms
4,584 KB
testcase_16 AC 337 ms
4,632 KB
testcase_17 AC 329 ms
4,528 KB
testcase_18 AC 340 ms
4,568 KB
testcase_19 AC 338 ms
4,552 KB
testcase_20 AC 327 ms
4,540 KB
testcase_21 AC 332 ms
4,784 KB
testcase_22 AC 333 ms
4,608 KB
testcase_23 AC 310 ms
4,588 KB
testcase_24 AC 322 ms
4,616 KB
testcase_25 AC 319 ms
4,532 KB
testcase_26 AC 307 ms
4,676 KB
testcase_27 AC 279 ms
4,376 KB
testcase_28 AC 1 ms
4,380 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
//#include <atcoder/all>
//using namespace atcoder;
#pragma GCC target("avx2")
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")

#define rep(i, n) for(ll i = 0; i < n; i++)
#define rep2(i, m, n) for(ll i = m; i <= n; i++)
#define rrep(i, m, n) for(ll i = m; i >= n; i--)
#define all(a) a.begin(), a.end()
#define rall(a) a.rbegin(), a.rend()
#define MAX(x) *max_element(all(x))
#define MIN(x) *min_element(all(x))
#define SZ(a) ((ll)(a).size())
#define bit(n, k) ((n >> k) & 1)
using namespace std;
using ll = long long;
using P = pair<ll, ll>;
const int dx[4] = { 1, 0, -1, 0};
const int dy[4] = { 0, -1, 0, 1};
const int inf = 1e9 + 1;
const ll infll = 1e18;
//const double pi = acos(-1);
     
using Graph = vector<vector<ll>>;
using REV_PQ = priority_queue<ll, vector<ll>, greater<ll>>;
using PQ = priority_queue<ll>;

const int SIZE = 400010;
ll powll(ll n, ll x) {
    // return n ^ x;
    ll ret = 1;
    rep(i, x) ret *= n;
    return ret;
}
void OutputYesNo(bool val) {
    if (val) cout << "Yes" << endl;
    else cout << "No" << endl;
}
void OutputTakahashiAoki(bool val) {
    if (val) cout << "Takahashi" << endl;
    else cout << "Aoki" << endl;
}
void OutPutInteger(ll x) { cout << x << endl; }
void OutPutString(string x) { cout << x << endl; }

int pop_count(ll n){
    int ret = 0;
    while(n > 0){
        if(n % 2 == 1) ret++;
        n /= 2; 
    }
    return ret; 
}

//using PAIR_REV_PQ = priority_queue<pair<ll, ll>, vector<pair<ll, ll>>, greater<pair<ll, ll>>>;
//typedef atcoder::modint998244353 mint;
//typedef atcoder::modint1000000007 mint;
//typedef modint mint;
//typedef vector<mint> ModintVec;
//typedef vector<ModintVec> ModintMat; 

struct UnionFind{
    vector<int> parent;
    vector<int> rank; // tree's depth

    UnionFind(int n){
        parent.resize(n);
        rank.resize(n);
        init(n);
    }

    // initialize
    void init(int n){
        for(int i = 0; i < n; i++){
            parent[i] = i;
            rank[i] = 0;
        }
    }
    // find root
    int find(int x){
        if(parent[x] == x){
            return x;
        }else{
            return parent[x] = find(parent[x]);
        }
    }

    void unite(int x, int y){
        x = find(x);
        y = find(y);
        if(x == y) return;

        if(rank[x] < rank[y]){
            parent[x] = y;
        }else{
            parent[y] = x;
            if(rank[x] == rank[y]) rank[x]++;
        }
    }

    bool same(int x, int y){
        return find(x) == find(y);
    }
};
int main(){
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    int n, q; cin >> n >> q;
    UnionFind uf(n);

    rep(i, n){
        int x; cin >> x;
        if(x == -1) continue;
        x--;
        if(!uf.same(i, x)) uf.unite(i, x) ;
    }
    rep(i, q){
        int x, y; cin >> x >> y;
        x--, y--;
        if(uf.same(x, y)) cout << "Yes" << endl;
        else cout << "No" << endl;
    }

    
}
0