hdu 4366 Successor (线段树xdfs序x排序预处理)

题意：
给一颗树，每个节点两个属性（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;
}