[dfs序+线段树] codeforces 343D. Water Tree

题意：
给出一棵树，节点权值为true或者false，初始每个节点权值为false，维护两种操作。

• 形如1 x：把x节点及其子树的值赋为true。
• 形如2 x：把x节点及其到根节点的祖先赋值为false。
• 形如3 x：问x节点的值。

题解：
由于要对子树进行操作，先求出dfs序，用线段树即可快速维护子树。
然而发现还要对祖先操作，dfs序对子树和祖先显然不能同时维护。
稍微转换下思路，要维护祖先，也就是祖先下面的所有子节点都能影响到祖先，并且操作１和操作２对一个节点的值的影响是互斥的，最后一次影响的就是当前的值了。
于是用两棵线段树维护操作１的时间戳，操作２的时间戳，查询时比较哪个操作的时间戳更靠后。
操作１的时间戳不需要往上更新，操作２的时间戳需要往上更新。

#include<bits/stdc++.h>
#define lson rt<<1,l,mid
#define rson rt<<1|1,mid+1,r
using namespace std;
const int N = 5e5+15;
vector<int>G[N];
int lazy[N<<2], tree[N<<2], lazy1[N<<2], tree1[N<<2];
int stp = 1, in[N], out[N], id[N];
void dfs(int rt, int f){
    in[rt] = stp, id[stp++] = rt;
    for(auto& v : G[rt]){
        if(v != f) dfs(v, rt);
    }
    out[rt] = stp-1;
}
inline void push_up(int rt, int *tree, int* lazy){
    tree[rt] = max(tree[rt<<1], tree[rt<<1|1]);
}
void build(int rt, int l, int r, int *tree, int *lazy){
    tree[rt] = lazy[rt] = 0;
    if(l == r) return;
    int mid = (l+r) >> 1;
    build(lson, tree, lazy);
    build(rson, tree, lazy);
}
inline void push_down(int rt, int* tree, int* lazy){
    if(lazy[rt] != 0){
        tree[rt<<1] = tree[rt<<1|1] = lazy[rt<<1] = lazy[rt<<1|1] = lazy[rt];
        lazy[rt] = 0;
    }
}
void update(int rt, int l, int r, int ql, int qr, int val, int* tree, int* lazy, int f){
    if(ql <= l && qr >= r){
        lazy[rt] = val;
        tree[rt] = val;
        return;
    }
    push_down(rt, tree, lazy);
    int mid = (l+r) >> 1;
    if(ql <= mid) update(lson, ql, qr, val, tree, lazy, f);
    if(qr > mid) update(rson, ql, qr, val, tree, lazy, f);
    if(f) push_up(rt, tree, lazy);
}
int query(int rt, int l, int r, int ql, int qr,int *tree, int *lazy, int f){
    if(ql <= l && qr >= r){
        return tree[rt];
    }
    push_down(rt, tree, lazy);
    int mid = (l+r) >> 1;
    int res = 0;
    if(ql <= mid) res = max(res, query(lson, ql, qr, tree, lazy, f));
    if(qr > mid) res = max(res, query(rson, ql, qr, tree, lazy, f));
    if(f) push_up(rt, tree, lazy);
    return res;
}
int main(){
    int n;
    scanf("%d", &n);
    for(int i = 1; i < n; ++i){
        int a, b;
        scanf("%d%d", &a, &b);
        G[a].push_back(b);
        G[b].push_back(a);
    }
    dfs(1, -1);
    --stp;
    build(1, 1, stp, tree, lazy);
    build(1, 1, stp, tree1, lazy1);
    int q;
    scanf("%d", &q);
    for(int i = 1; i <= q; ++i){
        int t, v;
        scanf("%d%d", &t, &v);
        if(t == 1) update(1, 1, stp, in[v], out[v], i, tree, lazy, 0);
        else if(t == 2) update(1, 1, stp, in[v], in[v], i, tree1, lazy1, 1);
        else{
            int a = query(1, 1, stp, in[v], in[v], tree, lazy, 0);
            int b = query(1, 1, stp, in[v], out[v], tree1, lazy1, 1);
            printf("%d\n", a>b);
        }
    }
}