CodeForces - 343D_Water Tree_线段树维护dfs序

题意

n 个水池形成树状结构。当一个节点倒满水时，它的子孙节点也都会被倒满。当一个节点被抽空时，它的祖先节点也都会被抽空。m 个操作，倒满一个节点，抽空一个节点，询问一个节点是不是满的。对于每个询问操作输出答案。

思路

线段树维护dfs序。
倒满操作，区间修改，它和它的子树所在的区间置 1。
清空操作，单点修改，在叶子节点修改后结果上传。
查询操作，查询区间是否都为1。
wa点
线段树维护的是dfs序，而不是原本树的结构。原树中某个节点的祖先可能和它分在了线段树的不同子树上。这里就出问题了。
先把该节点抽空，这时父节点也空了，但是修改操作并没有影响到父节点。然后再把该节点倒满，此时实际情况是该节点是满的，它的父节点是空的。但线段树上显示的父节点仍然是满的。
所以，在倒满操作之前，如果该节点在操作之前是空的，并且它有父亲节点，则应该把它的父亲节点先置空。

链接

https://vjudge.net/contest/176833#problem/F

代码

#include
#include
#include
#include

using namespace std;

const int maxn = 5 * 1e5 + 10;

int n, q;
int x, y;
int tree[maxn * 4], lazy[maxn * 4];
vector<int> vec[maxn];
int id[maxn], End[maxn];
int tot;
int pre[maxn];

void push_down(int p)
{
    if(lazy[p] == 0) return;
    tree[p<<1] = lazy[p<<1] = 1;
    tree[p<<1|1] = lazy[p<<1|1] = 1;
    lazy[p] = 0;
}

void Fill(int p, int l, int r, int x, int y)
{
    if(x <= l && r <= y)
    {
        tree[p] = 1, lazy[p] = 1;
        return;
    }

    if(lazy[p]) return;

    int mid = (l + r) >> 1;

    if(x <= mid) Fill(p<<1, l, mid, x, y);
    if(y > mid) Fill(p<<1|1, mid+1, r, x, y);

    tree[p] = tree[p<<1] & tree[p<<1|1];
}

void Empty(int p, int l, int r, int x)
{
    if(l == r)
    {
        tree[p] = 0, lazy[p] = 0;
        return;
    }

    push_down(p);

    int mid = (l + r) >> 1;

    if(x <= mid) Empty(p<<1, l, mid, x);
    else Empty(p<<1|1, mid+1, r, x);

    tree[p] = tree[p<<1] & tree[p<<1|1];
}

int query(int p, int l, int r, int x, int y)
{
    if(x <= l && r <= y) return tree[p];
    if(lazy[p] == 1) return 1;

    push_down(p);

    int res = 1;
    int mid = (l + r) >> 1;
    if(x <= mid) res &= query(p<<1, l, mid, x, y);
    if(y > mid) res &= query(p<<1|1, mid+1, r, x, y);

    return res;
}

void dfs(int v, int p)
{
    id[v] = ++tot;
    pre[v] = p;

    for(int i= 0; i< vec[v].size(); i++)
    {
        int u = vec[v][i];
        if(id[u] == 0) dfs(u, v);
    }
    End[v] = tot;
}

int main()
{
    //freopen("in.txt", "r", stdin);

    scanf("%d", &n);

    for(int i= 1; i<= n-1; i++)
    {
        scanf("%d %d", &x, &y);
        vec[x].push_back(y);
        vec[y].push_back(x);
    }

    dfs(1, 0);

    scanf("%d", &q);
    while(q--)
    {
        scanf("%d %d", &x, &y);
        if(x == 1)
        {
            int tmp = query(1, 1, n, id[y], End[y]);
            if(tmp == 0 && pre[y] != 0) Empty(1, 1, n, id[pre[y]]);
            Fill(1, 1, n, id[y], End[y]);
        }
        else if(x == 2) Empty(1, 1, n, id[y]);
        else cout << query(1, 1, n, id[y], End[y]) << endl;
    }

    return 0;
}