結果

問題 No.2316 Freight Train
ユーザー Be11T_Be11T_
提出日時 2023-05-26 21:27:04
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 317 ms / 2,000 ms
コード長 2,949 bytes
コンパイル時間 2,294 ms
コンパイル使用メモリ 209,372 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-06-07 05:25:19
合計ジャッジ時間 11,023 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 302 ms
5,376 KB
testcase_04 AC 177 ms
5,376 KB
testcase_05 AC 117 ms
5,376 KB
testcase_06 AC 40 ms
5,376 KB
testcase_07 AC 272 ms
5,376 KB
testcase_08 AC 160 ms
5,376 KB
testcase_09 AC 227 ms
5,376 KB
testcase_10 AC 248 ms
5,376 KB
testcase_11 AC 212 ms
5,376 KB
testcase_12 AC 254 ms
5,376 KB
testcase_13 AC 302 ms
5,376 KB
testcase_14 AC 312 ms
5,376 KB
testcase_15 AC 305 ms
5,376 KB
testcase_16 AC 305 ms
5,376 KB
testcase_17 AC 311 ms
5,376 KB
testcase_18 AC 298 ms
5,376 KB
testcase_19 AC 304 ms
5,376 KB
testcase_20 AC 305 ms
5,376 KB
testcase_21 AC 309 ms
5,376 KB
testcase_22 AC 317 ms
5,376 KB
testcase_23 AC 285 ms
5,376 KB
testcase_24 AC 294 ms
5,376 KB
testcase_25 AC 282 ms
5,376 KB
testcase_26 AC 285 ms
5,376 KB
testcase_27 AC 262 ms
5,376 KB
testcase_28 AC 2 ms
5,376 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