結果

問題 No.2325 Skill Tree
ユーザー umezo
提出日時 2023-05-28 14:19:05
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 176 ms / 3,000 ms
コード長 1,126 bytes
コンパイル時間 2,030 ms
コンパイル使用メモリ 206,032 KB
最終ジャッジ日時 2025-02-13 11:13:02
ジャッジサーバーID
（参考情報） 		judge3 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 36
権限があれば一括ダウンロードができます

ソースコード

diff # 

#define rep(i,n) for(int i=0;i<(int)(n);i++)
#define ALL(v) v.begin(),v.end()
typedef long long ll;
 
#include<bits/stdc++.h>
using namespace std;

vector<int> G[200200];
const int INF=1e9+7;
 
int main(){
  ios::sync_with_stdio(false);
  std::cin.tie(nullptr);
  
  int n;
  cin>>n;
  vector<int> L(n),A(n);
  for(int i=1;i<n;i++){
    cin>>L[i]>>A[i];
    A[i]--;
    G[i].push_back(A[i]);
    G[A[i]].push_back(i);
  }
  vector<int> P(n,INF);
  
  auto dfs=[&](auto dfs,int v,int p,int d)->void{
    d=max(d,L[v]);
    P[v]=d;
    for(auto nv:G[v]){
      if(nv==p) continue;
      dfs(dfs,nv,v,d);
    }
  };
  
  dfs(dfs,0,-1,0);
  map<int,int> m;
  rep(i,n) m[P[i]]++;
  int k=m.size();
  vector<int> B,C;
  for(auto [b,c]:m){
    B.push_back(b);
    C.push_back(c);
  }
  vector<int> S(k);
  S[0]=C[0];
  for(int i=1;i<k;i++) S[i]=S[i-1]+C[i];
  
  int q;
  cin>>q;
  while(q--){
    int c,x;
    cin>>c>>x;
    if(c==1){
      auto a=upper_bound(ALL(B),x)-B.begin();
      a--;
      cout<<S[a]<<'\n';
    }
    else{
      x--;
      if(P[x]==INF) cout<<-1<<'\n';
      else cout<<P[x]<<'\n';
    }
  }

  return 0;
}
0