結果

問題 No.2325 Skill Tree
ユーザー kyawakyawa
提出日時 2023-05-28 16:29:58
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 1,025 ms / 3,000 ms
コード長 4,170 bytes
コンパイル時間 3,127 ms
コンパイル使用メモリ 218,648 KB
実行使用メモリ 116,224 KB
最終ジャッジ日時 2024-06-08 09:35:47
合計ジャッジ時間 31,847 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 3 ms
5,376 KB
testcase_02 AC 3 ms
5,376 KB
testcase_03 AC 2 ms
5,376 KB
testcase_04 AC 2 ms
5,376 KB
testcase_05 AC 2 ms
5,376 KB
testcase_06 AC 3 ms
5,376 KB
testcase_07 AC 323 ms
18,560 KB
testcase_08 AC 316 ms
49,792 KB
testcase_09 AC 396 ms
35,072 KB
testcase_10 AC 463 ms
81,664 KB
testcase_11 AC 467 ms
58,800 KB
testcase_12 AC 936 ms
116,096 KB
testcase_13 AC 919 ms
116,096 KB
testcase_14 AC 914 ms
116,096 KB
testcase_15 AC 934 ms
116,224 KB
testcase_16 AC 918 ms
116,096 KB
testcase_17 AC 867 ms
116,224 KB
testcase_18 AC 861 ms
116,096 KB
testcase_19 AC 879 ms
116,096 KB
testcase_20 AC 873 ms
116,224 KB
testcase_21 AC 862 ms
116,096 KB
testcase_22 AC 975 ms
116,096 KB
testcase_23 AC 952 ms
116,096 KB
testcase_24 AC 948 ms
116,096 KB
testcase_25 AC 950 ms
116,096 KB
testcase_26 AC 973 ms
116,096 KB
testcase_27 AC 925 ms
63,944 KB
testcase_28 AC 925 ms
63,952 KB
testcase_29 AC 926 ms
64,072 KB
testcase_30 AC 924 ms
64,080 KB
testcase_31 AC 903 ms
64,072 KB
testcase_32 AC 757 ms
64,040 KB
testcase_33 AC 770 ms
63,944 KB
testcase_34 AC 772 ms
63,948 KB
testcase_35 AC 975 ms
63,900 KB
testcase_36 AC 914 ms
63,948 KB
testcase_37 AC 1,025 ms
64,076 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

/* author: kyawa */
#include<bits/stdc++.h>
using namespace std;

/*
 考察
 */

class WaveletMatrix{
    using i64 = int64_t;
public:
    explicit WaveletMatrix() = default;
    WaveletMatrix(vector<i64> &_v){ __WaveletMatrix(_v);};
    i64 range_freq(i64 L, i64 R, i64 lower, i64 upper){ return __range_freq(L, R, upper) - __range_freq(L, R, lower);}
    i64 range_Kthmin(i64 L, i64 R, i64 K){ return __range_Kthmin_assignXOR(L, R, K, 0);}
    i64 range_Kthmax(i64 L, i64 R, i64 K){ return __range_Kthmin_assignXOR(L, R, R - L - K - 1, 0);}
    i64 range_successor(i64 L, i64 R, i64 value){ i64 C = __range_freq(L, R, value + 1); return C == R - L ? -1 : __range_Kthmin_assignXOR(L, R, C, 0);}
    i64 range_predecessor(i64 L, i64 R, i64 value){ i64 C = __range_freq(L, R, value); return C == 0 ? -1 : __range_Kthmin_assignXOR(L, R, C - 1, 0);}
    i64 range_min_assignXOR(i64 L, i64 R, i64 value){ return __range_Kthmin_assignXOR(L, R, 0, value);}
    i64 range_max_assignXOR(i64 L, i64 R, i64 value){ return __range_Kthmin_assignXOR(L, R, R - L - 1, value);}
    i64 range_Kthmin_assignXOR(i64 L, i64 R, i64 K, i64 value){ return __range_Kthmin_assignXOR(L, R, K, value);}
    i64 range_Kthmax_assignXOR(i64 L, i64 R, i64 K, i64 value){ return __range_Kthmin_assignXOR(L, R, R - L - K - 1, value);}
private:
    vector<i64> Matrix;
    i64 bitsize;
    i64 prefixsize;
    void __WaveletMatrix(vector<i64> v){
        for(auto &e : v) assert(0 <= e and e < INT64_MAX);
        i64 v_max = *max_element(v.begin(), v.end()) + 1;
        bitsize = 64 - __builtin_clzll(v_max);
        prefixsize = v.size()+1;
        Matrix.resize(prefixsize * bitsize);
        for(i64 h = bitsize - 1; h >= 0; h--){
            for(i64 i = 0; i < v.size(); i++){
                Matrix[i+1 + prefixsize * h] = Matrix[i + prefixsize * h] + (v[i] >> h & 1);
            }
            stable_sort(v.begin(), v.end(), [&](auto a, auto b){ return (a >> h & 1) < (b >> h & 1);});
        }
    };
    i64 __range_freq(i64 L, i64 R, i64 upper){
        if((uint64_t)upper >= ((uint64_t)1 << bitsize)) return R - L;
        if(upper < 0) return 0;
        i64 res = 0;
        for(i64 h = bitsize - 1; h >= 0; h--){
            auto [L0, R0, L1, R1] = decomp(L, R, h);
            if(upper & ((i64)1 << h)){
                res += R0 - L0;
                tie(L, R) = {L1, R1};
            }else{
                tie(L, R) = {L0, R0};
            }
        }
        return res;
    };
    i64 __range_Kthmin_assignXOR(i64 L, i64 R, i64 K, i64 value){
        i64 res = 0;
        for(i64 h = bitsize - 1; h >= 0; h--){
            auto [L0, R0, L1, R1] = decomp(L, R, h);
            if(value & ((i64)1 << h)){
                swap(L0, L1); swap(R0, R1);
            }
            if(K < R0 - L0){
                tie(L, R) = {L0, R0};
            }else{
                K -= (R0 - L0);
                res |= (i64)1 << h;
                tie(L, R) = {L1, R1};
            }
        }
        return res;
    }
    tuple<i64,i64,i64,i64> decomp(i64 L, i64 R, i64 h){
        return tuple<i64,i64,i64,i64>({
            L - Matrix[L + prefixsize * h],
            R - Matrix[R + prefixsize * h],
            prefixsize - 1 - Matrix[prefixsize * (h+1) - 1] + Matrix[L + prefixsize * h],
            prefixsize - 1 - Matrix[prefixsize * (h+1) - 1] + Matrix[R + prefixsize * h]
        });
    }
};

int main(){
    int64_t N; cin >> N;
    vector<int64_t> resy(N, INT64_MAX-1); resy[0] = 0;
    vector<vector<int64_t>> edge(N);
    vector<int64_t> L(N, 0);
    for(int64_t i = 1; i < N; i++){
        cin >> L[i];
        int64_t a; cin >> a; a--;
        edge[a].push_back(i);
    }
    auto dfs = [&](auto &&self, int64_t vis, int64_t prv) -> void {
        resy[vis] = max(resy[prv], L[vis]);
        for(auto &nxt : edge[vis]){
            self(self, nxt, vis);
        }
    };
    dfs(dfs, 0, 0);
    WaveletMatrix wm(resy);
    int64_t Q; cin >> Q;
    while(Q--){
        int64_t t, x; cin >> t >> x;
        if(t == 1){
            cout << wm.range_freq(0, N, 0, x+1) << '\n';
        }else{
            cout << (resy[x-1] == INT64_MAX-1 ? -1 : resy[x-1]) << '\n';
        }
    }
}
0