結果

問題 No.2316 Freight Train
ユーザー yuusaanyuusaan
提出日時 2024-06-30 20:19:31
言語 C++23
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 342 ms / 2,000 ms
コード長 4,592 bytes
コンパイル時間 6,002 ms
コンパイル使用メモリ 320,008 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-06-30 20:20:01
合計ジャッジ時間 15,666 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,812 KB
testcase_01 AC 2 ms
6,944 KB
testcase_02 AC 2 ms
6,940 KB
testcase_03 AC 317 ms
6,944 KB
testcase_04 AC 175 ms
6,940 KB
testcase_05 AC 123 ms
6,940 KB
testcase_06 AC 43 ms
6,944 KB
testcase_07 AC 300 ms
6,940 KB
testcase_08 AC 176 ms
6,940 KB
testcase_09 AC 241 ms
6,940 KB
testcase_10 AC 263 ms
6,940 KB
testcase_11 AC 217 ms
6,940 KB
testcase_12 AC 278 ms
6,940 KB
testcase_13 AC 328 ms
6,944 KB
testcase_14 AC 338 ms
6,940 KB
testcase_15 AC 342 ms
6,940 KB
testcase_16 AC 329 ms
6,944 KB
testcase_17 AC 332 ms
6,944 KB
testcase_18 AC 328 ms
6,940 KB
testcase_19 AC 323 ms
6,940 KB
testcase_20 AC 326 ms
6,940 KB
testcase_21 AC 328 ms
6,944 KB
testcase_22 AC 336 ms
6,944 KB
testcase_23 AC 312 ms
6,940 KB
testcase_24 AC 320 ms
6,940 KB
testcase_25 AC 317 ms
6,940 KB
testcase_26 AC 304 ms
6,940 KB
testcase_27 AC 273 ms
6,940 KB
testcase_28 AC 2 ms
6,940 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <string>
#include <stdio.h>
#include <algorithm>
#include <vector>
#include <tuple>
#include <map>
#include<set>
#include<queue>
#include<stack>
#include <unordered_set>
#include<thread>
#include<bits/stdc++.h>

#include <atcoder/all>
#include <cstdio>

#pragma GCC target("avx")
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")

//if(a < 0 || h <= a || b < 0 || w <= b)return;
// string abc = "abcdefghijklmnopqrstuvwxyz";
// string abc = "ABCDEFGHIJKLMNOPQRSTUVWXYZ";

using namespace std;
using namespace atcoder;
using ll = long long;
using ld = long double;
using ull = unsigned long long;
//using mint = modint998244353;
using mint = modint1000000007;
template<typename T> using pq = priority_queue<T>;//降順?(最大取り出し)
template<typename T> using pqg = priority_queue<T, vector<T>, greater<T>>;//昇順?(最小取り出し)
template<typename T> using vector2 = vector<vector<T>>;
template<typename T> using vector3 = vector<vector<vector<T>>>;
template<typename T> using vector4 = vector<vector<vector<vector<T>>>>;
template<typename T> using vector5 = vector<vector<vector<vector<vector<T>>>>>;
template<typename T> using vector6 = vector<vector<vector<vector<vector<vector<T>>>>>>;
template<typename T> using pairs = pair<T,T>;
#define rep(i, n) for (ll i = 0; i < ll(n); i++)
#define rep1(i,n) for(int i = 1;i <= int(n);i++)
#define repm(i, m, n) for (int i = (m); (i) < int(n);(i)++)
#define repmr(i, m, n) for (int i = (m) - 1; (i) >= int(n);(i)--)
#define rep0(i,n) for(int i = n - 1;i >= 0;i--)
#define rep01(i,n) for(int i = n;i >= 1;i--)




// ユークリッドの互除法による最大公約数算出
ll GCD(ll a,ll b){
    if(b == 0)return a;
    return GCD(b, a % b);
}
//拡張ユークリッドの互除法による(ax + by = GCD(a,b))を満たすx,yの算出
pair<long long, long long> extgcd(long long a, long long b) {
  if (b == 0) return make_pair(1, 0);
  long long x, y;
  tie(y, x) = extgcd(b, a % b);
  y -= a / b * x;
  return make_pair(x, y);
}
struct UnionFind {
    vector<int> par; // par[i]:iの親の番号　(例) par[3] = 2 : 3の親が2
    
    UnionFind(int N) : par(N) { //最初は全てが根であるとして初期化
        for(int i = 0; i < N; i++) par[i] = i;
    }
    
    int root(int x) { // データxが属する木の根を再帰で得る：root(x) = {xの木の根}
        if (par[x] == x) return x;
        return par[x] = root(par[x]);
    }
    
    void unite(int x, int y) { // xとyの木を併合
        int rx = root(x); //xの根をrx
        int ry = root(y); //yの根をry
        if (rx == ry) return; //xとyの根が同じ(=同じ木にある)時はそのまま
        par[rx] = ry; //xとyの根が同じでない(=同じ木にない)時：xの根rxをyの根ryにつける
    }
    
    bool same(int x, int y) { // 2つのデータx, yが属する木が同じならtrueを返す
        int rx = root(x);
        int ry = root(y);
        return rx == ry;
    }
};
ll n;


//座標圧縮
vector<ll> Ccomp(vector<ll> a){
    vector<ll> b = a;
    sort(b.begin(),b.end());
    b.erase(unique(b.begin(),b.end()),b.end());//ダブり消去
    vector<ll> rtn;
    rep(j,a.size()){
        ll pb = lower_bound(b.begin(),b.end(),a[j]) - b.begin();
        rtn.push_back(pb);
    }
    return rtn;
}
/// ここから////////////////////////////////////////////




using F = ll;
using S = ll;
string s;

ll modPow(ll a, ll n, ll mod) { if(mod==1) return 0;ll ret = 1; ll p = a % mod; while (n) { if (n & 1) ret = ret * p % mod; p = p * p % mod; n >>= 1; } return ret; }
void cincout(){
  ios::sync_with_stdio(false);
    std::cin.tie(nullptr);
  cout<< fixed << setprecision(15);
}
//seg,遅延segの設定-----ここから
S op(S a,S b){return max(a,b);}//何を求めるか(最大値とか)

S e(){return 0;}//モノイド(初期値)

S mapping (F a,S b){return a + b;}//遅延処理

F composition (F a,F b){return a + b;}//遅延中の枝にさらに処理

F id(){return 0;}//遅延のモノイド

vector<int> Op(vector<int> a,vector<int> b){a.insert(a.end(),b.begin(),b.end()); return a;}

vector<int> E(){return vector<int> (0);}

//segここまで
ll INF = ll(2e18);

int main() {
    cincout();
    ll q;
    cin >> n >> q;
    UnionFind uf(n);
    rep(j,n){
        int p;
        cin >> p;
        if(p != -1){
            p--;
            uf.unite(p,j);
        }
    }
    rep(j,q){
        int a,b;
        cin >> a >> b;
        a--,b--;
        if(uf.same(a,b))cout << "Yes" << endl;
        else cout << "No" << endl;
    }
    return 0;
}
0