結果

問題 No.2325 Skill Tree
ユーザー みどみど
提出日時 2023-05-28 15:44:29
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
TLE  
実行時間 -
コード長 2,329 bytes
コンパイル時間 2,671 ms
コンパイル使用メモリ 221,968 KB
実行使用メモリ 32,256 KB
最終ジャッジ日時 2024-06-08 08:24:22
合計ジャッジ時間 14,232 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
10,752 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 2 ms
5,376 KB
testcase_04 AC 2 ms
5,376 KB
testcase_05 AC 3 ms
5,376 KB
testcase_06 AC 2 ms
5,376 KB
testcase_07 AC 717 ms
7,168 KB
testcase_08 AC 432 ms
15,104 KB
testcase_09 AC 661 ms
11,392 KB
testcase_10 AC 780 ms
23,376 KB
testcase_11 AC 1,022 ms
17,408 KB
testcase_12 TLE -
testcase_13 AC 2,449 ms
32,140 KB
testcase_14 AC 2,142 ms
32,100 KB
testcase_15 AC 2,654 ms
31,956 KB
testcase_16 TLE -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
testcase_20 -- -
testcase_21 -- -
testcase_22 -- -
testcase_23 -- -
testcase_24 -- -
testcase_25 -- -
testcase_26 -- -
testcase_27 -- -
testcase_28 -- -
testcase_29 -- -
testcase_30 -- -
testcase_31 -- -
testcase_32 -- -
testcase_33 -- -
testcase_34 -- -
testcase_35 -- -
testcase_36 -- -
testcase_37 -- -
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#define mrep(i, s, n) for (int i = (s); i < (int)(n); i++)
#define rep(i, n) mrep(i, 0, n)
using namespace std;
using ll = long long;
const int INF = int(1e+9)+1;
const ll INFL = ll(1e+18)+1;

ll N;
vector<ll> L, A;
vector<vector<pair<ll,ll>>> graph, graph_rev;

vector<ll> memo_L, memo_A;

int main() {
    cin >> N;
    L.resize(N+2, 0), A.resize(N+2, 1);
    rep(i, N-1) {cin >> L[i+2] >> A[i+2];}
    graph = vector<vector<pair<ll, ll>>>(N+2);
    graph_rev = vector<vector<pair<ll, ll>>>(N+2);

    rep(i, N) {
        int u = i+2, v = A[i+2];
        graph[v].push_back(make_pair(u, L[i+2]));
        graph_rev[u].push_back(make_pair(v, L[i+2]));
    }

    vector<ll> level_sorted(N);
    copy(L.begin()+2, L.end(), level_sorted.begin());
    sort(level_sorted.begin(), level_sorted.end());

    memo_L.resize(N, 0);
    memo_A.resize(N, 0);
    for (int i = 0; i<N; i++) {
        ll lvl = level_sorted[i];
        int ans = 0;
        queue<ll> q;
        q.push(1);
        while(!q.empty()) {
            ans++;
            int s = q.front(); q.pop();
            // 隣接している頂点を追加
            for (auto un: graph[s]) {
                auto [u, l] = un;
                if (lvl < l) {continue;} // レベル不足
                q.push(u);
            }
        }
        memo_L[i] = lvl;
        memo_A[i] = ans;
    }

    int Q;
    cin >> Q;
    rep(_, Q) {
        int op, val;
        cin >> op >> val;
        if (op == 1) {
            auto it = lower_bound(memo_L.begin(), memo_L.end(), ll(val));
            
            if (*it != val) {it--;}
            int idx = it-memo_L.begin();
            cout << memo_A[idx]-1 << endl;
        } else {
            bool ac = false;
            vector<bool> done(N+2, false);
            ll ans = -1;
            queue<ll> q;
            q.push(val);
            while(!q.empty()) {
                int s = q.front(); q.pop();
                done[s] = true;
                if (s == 1) {ac = true; break;}
                for(auto ul: graph_rev[s]) {
                    auto [u, l] = ul;
                    if (done[u]) {continue;}
                    ans = max(ans, l);
                    q.push(u);
                }
            }
            if (!ac) {ans = -1;}
            cout << ans << endl;
        }
    }

}
0