結果

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

ソースコード

diff #

#include <bits/stdc++.h>
 
using namespace std;
struct DataType {
  long long lv;
  int parent_id;
  char is_visited;
};

long long dfs(vector<DataType>& a, const int& id, map<int, int>& mp) {
  if (a[id].is_visited == 1) return a[id].lv;
  if (mp[id] == 1) { // loop
    a[id].lv = -1;
    a[id].is_visited = 1;
    return -1;
  }
  mp[id] = 1;
  long long lv = dfs(a, a[id].parent_id, mp);
  if (lv == -1) a[id].lv = -1;
  a[id].lv = max(a[id].lv, lv);
  a[id].is_visited = 1;
  return a[id].lv;
}

int main() {
  ios::sync_with_stdio(false);
  cin.tie(0);
  int n, q;
  cin >> n;
  vector<DataType> a(n);
  a[0].lv = 0; a[0].parent_id = -1; a[0].is_visited = 1;
  for (int i=1;i<n;++i) {
    long long lv;
    int id;
    cin >> lv >> id;
    --id;
    a[i].lv = lv; a[i].parent_id = id; a[i].is_visited = 0;
  }
  cin >> q;
  vector<pair<long long, long long>> query(q);
  for (int i=0;i<q;++i) cin >> query[i].first >> query[i].second;

  map<int, int> mp;

  for (int i=0;i<n;++i) {
    if (a[i].is_visited == 1) continue;
    a[i].lv = dfs(a, i, mp);
    //for (const auto& e : a) cout << e.lv << ' ';
    //cout << '\n';
  }
  vector<long long> skills;
  for (int i=0;i<n;++i) {
    if (a[i].lv == -1) continue;
    skills.push_back(a[i].lv);
  }
  sort(skills.begin(), skills.end());

  //for (const auto& e : skills) cout << e << ' ';
  //cout << '\n';

  for (int i=0;i<q;++i) {
    if (query[i].first == 1) {
      auto lit = upper_bound(skills.begin(), skills.end(), query[i].second);
      int cnt = lit - skills.begin();
      cout << cnt << '\n';
    } else {
      cout << a[query[i].second-1].lv << '\n';
    }
  }
}
0