結果

問題 No.2325 Skill Tree
ユーザー みどみど
提出日時 2023-05-28 15:58:03
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 2,790 bytes
コンパイル時間 2,568 ms
コンパイル使用メモリ 228,392 KB
実行使用メモリ 41,648 KB
最終ジャッジ日時 2024-06-08 09:03:44
合計ジャッジ時間 21,754 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 WA -
testcase_01 AC 2 ms
5,376 KB
testcase_02 WA -
testcase_03 AC 2 ms
5,376 KB
testcase_04 WA -
testcase_05 WA -
testcase_06 WA -
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
testcase_10 WA -
testcase_11 AC 376 ms
22,184 KB
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 WA -
testcase_16 AC 761 ms
39,596 KB
testcase_17 AC 704 ms
37,416 KB
testcase_18 AC 704 ms
37,288 KB
testcase_19 AC 700 ms
37,292 KB
testcase_20 WA -
testcase_21 AC 705 ms
37,288 KB
testcase_22 AC 694 ms
41,644 KB
testcase_23 AC 695 ms
41,648 KB
testcase_24 AC 703 ms
41,644 KB
testcase_25 AC 692 ms
41,512 KB
testcase_26 AC 687 ms
41,648 KB
testcase_27 TLE -
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;
unordered_map<int, ll> memo0, memo1;

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;
    }
    memo_L.push_back(INFL);
    

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

}
0