提出 #29486825

---

ソースコード 拡げる

#include <bits/stdc++.h>
using namespace std;
void sort_list(vector<int> &x, vector<set<int> > &r, int n,vector<vector<int> > &P){
  vector<int> l = {x.at(n)};
  for (int m : r.at(n)){
    r.at(m).erase(n);
    sort_list(x,r,m,P);
    l.insert(l.end(), P.at(m).begin(), P.at(m).end());
  }
  sort(l.rbegin(), l.rend());
  vector<int> L(20);
  int s = min(20, (int)l.size());
  for (int i=0; i<s; ++i){
    L.at(i) = l.at(i);
  }
  P.at(n) = L;
}

int main() {
  int N, Q;
  cin >> N >> Q;
  vector<int> X(N);
  
  for (int i=0; i<N; ++i) cin >> X.at(i);
  
  vector<set<int> > r(N);
  int a, b;
  for (int i=0; i<N-1; ++i){
    cin >> a >> b;
    r.at(a-1).insert(b-1);
    r.at(b-1).insert(a-1);
  };
  
  vector<vector<int> > P(N);
  sort_list(X,r,0,P);
  
  int v, k;
  for (int i=0; i<Q; ++i){
    cin >> v >> k;
    cout << P.at(v-1).at(k-1) << endl;
  }
  return 0;
}

提出情報

ジャッジ結果