题意:
给一颗树,每个节点两个属性(x,y)。
每次查询一个子树中,x大于root(子树的根)且y最大的节点。
思路:
查询子树,很容易想到先求出dfs序然后用线段树维护y值。
但如何保证 x 呢?
如果让线段树初始为空,叶子的值是-1。
将节点按 x 从大到小排序,x 相等则 dfs序小的在前 (这里子树中root的dfs序最小)
然后只要按序查询每个节点对应的子树,然后将节点更新到线段树上。就预处理出了每个子树的答案。
int tr[N<<2], qL, qR;
#define lc o<<1
#define rc o<<1|1
int query(int o, int l, int r) {
if ( qL <= l && r <= qR ) return tr[o];
int m = (l + r) >> 1;
if ( qR <= m ) return query(lc, l, m);
if ( qL > m ) return query(rc, m+1, r);
return max ( query(lc, l, m), query(rc, m+1, r) );
}
void modify(int o, int l, int r, int x, int v) {
if ( l == r ) {
tr[o] = v; return;
}
int m = (l + r) >> 1;
if ( x <= m ) modify(lc, l, m, x, v);
else modify(rc, m+1, r, x, v);
tr[o] = max ( tr[lc], tr[rc] );
}
vector<int> graph[N];
int ans[N];
int L[N], R[N], loyaltyToId[1000005], timer;
void dfs(int u) {
L[u] = timer ++;
for(int v:graph[u]) dfs(v);
R[u] = timer - 1;
}
struct Node {
int x, y, id; // loyality, ability
bool operator < (const Node& rhs) const {
if ( y != rhs.y ) return y > rhs.y;
return L[id] < L[rhs.id];
}
};
Node as[N];
int main() {
#ifdef _LOCA_ENV_
freopen("input.in", "r", stdin);
#endif // _LOCA_ENV
int t, n, m;
scanf("%d", &t);
while ( t -- ) {
scanf("%d%d", &n, &m);
rep(i, 0, n-1) vector<int>().swap(graph[i]);
rep(i, 1, n-1) {
int x, y, z;
scanf("%d%d%d", &x, &y, &z);
graph[x].push_back(i);
as[i] = (Node) {y, z, i};
loyaltyToId[y] = i;
}
timer = 1;
dfs(0);
memset(tr, -1, sizeof(tr));
memset(ans, -1, sizeof(ans));
sort(as + 1, as + n);
for (int i = 1; i < n; ++ i) {
int v = as[i].id;
qL = L[v], qR = R[v];
int res = query(1, 1, n);
if ( res != -1 ) ans[v] = loyaltyToId[res];
modify(1, 1, n, L[as[i].id], as[i].x);
}
rep(i, 1, m) {
int x;
scanf("%d", &x);
printf("%d\n", ans[x]);
}
}
return 0;
}