hdu 4366 Successor (線段樹xdfs序x排序預處理)

題意：

給一顆樹，每個節點兩個屬性（x，y）。

每次查詢一個子樹中，x大于root（子樹的根）且y最大的節點。

思路：

查詢子樹，很容易想到先求出dfs序然後用線段樹維護y值。

但如何保證 x 呢？

如果讓線段樹初始為空，葉子的值是-1。

将節點按 x 從大到小排序，x 相等則 dfs序小的在前 （這裡子樹中root的dfs序最小）

然後隻要按序查詢每個節點對應的子樹，然後将節點更新到線段樹上。就預處理出了每個子樹的答案。

int tr[N<<], 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) >> ;
    if ( qR <= m ) return query(lc, l, m);
    if ( qL > m )  return query(rc, m+, r);
    return max ( query(lc, l, m), query(rc, m+, r) );
}

void modify(int o, int l, int r, int x, int v) {
    if ( l == r ) {
        tr[o] = v; return;
    }
    int m = (l + r) >> ;
    if ( x <= m ) modify(lc, l, m, x, v);
    else modify(rc, m+, r, x, v);
    tr[o] = max ( tr[lc], tr[rc] );
}

vector<int> graph[N];
int ans[N];

int L[N], R[N], loyaltyToId[], timer;
void dfs(int u) {
    L[u] = timer ++;
    for(int v:graph[u]) dfs(v);
    R[u] = timer - ;
}

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, , n-) vector<int>().swap(graph[i]);
        rep(i, , n-) {
            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 = ;
        dfs();

        memset(tr, -, sizeof(tr));
        memset(ans, -, sizeof(ans));
        sort(as + , as + n);
        for (int i = ; i < n; ++ i) {
            int v = as[i].id;
            qL = L[v], qR = R[v];
            int res = query(, , n);
            if ( res != - ) ans[v] = loyaltyToId[res];
            modify(, , n, L[as[i].id], as[i].x);
        }

        rep(i, , m) {
            int x;
            scanf("%d", &x);
            printf("%d\n", ans[x]);
        }
    }
    return ;
}