codeforces 343D - Water Tree

这题卡了我两天了，最后受不了，还是看了网上的题解。

一开始我是用两棵线段树来维护各个节点更新操作的时间。

一棵是记录dfs节点u的到它的最远子节点区间用来更新操作1，另一棵是树链剖分记录节点u到根节点的路径用来更新操作2。TLE了。。。。。。

网上的题解，看漏了一点，没记录节点u的父亲，导致一直wa在第六组。

原理：

线段树中保存这个区间有没有水，有水为1，没有水为0；

先对原树进序dfs标号，记录下节点u的出现的时间pos[u]（线段树中的位置）和son[u]。[pos[u], son[u]]为节点u的子节点区间加上本身，为操作1的更新区间。

对于操作2，只需要单点更新u就行了，因为对于节点v，如果区间[pos[v], son[v]]有0， 则节点v必没有水。

需要注意的是，进行操作1时，要判断区间[pos[u], son[u]]是不是为0，如果是0， 则还需要把u的父亲结点进行2操作，为什么呢？

例如:

2

1 2

4

1 1

2 2

1 2

3 1

对于这个案例，如果没有对父亲节点进行2操作，则输出为1，但是结果应为0。因为我们操作2只是单点更新，没有实际上更新到根。

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
#include <stack>
#include <cmath>
#include <cstdlib>
#include <vector>
#include <map>
#include <iostream>
using namespace std;
#define lson l, m, rt<<1
#define rson m+1, r, rt<<1|1

const int maxn = 500010;
int idx;
struct Edge
{
    int v, next;
}e[maxn * 2];
int g[maxn];
int fa[maxn];
void add(int u, int v)
{
    e[++idx].v = v;
    e[idx].next = g[u];
    g[u] = idx;
}
int pos[maxn], son[maxn];
void dfs(int u)
{
   pos[u] = ++idx; 
   for(int i = g[u]; i; i = e[i].next)
   {
       int v = e[i].v;
       if(pos[v] == 0)
       {
           fa[v] = u;
           dfs(v);
       }
   }
   son[u] = idx;
}
int tree[maxn<<2];
void PushUp(int rt)
{
    tree[rt] = tree[rt << 1] & tree[rt<<1|1] & tree[rt]; 
}
void PushDown(int rt)
{
    if(tree[rt])
        tree[rt<<1|1] = tree[rt<<1] = 1;
}
void UpdatePoint(int l, int r, int rt, int P)
{
    if(l == r)
    {
        tree[rt] = 0;
        return;
    }
    PushDown(rt);
    int m = (l + r) >> 1;
    if(P <= m)
        UpdatePoint(lson, P);
    else UpdatePoint(rson, P);
    PushUp(rt);

}
void UpdateSeg(int l, int r, int rt, int L, int R)
{
    if(L <= l && r <= R)
    {
        tree[rt] = 1;
        return;
    }
    PushDown(rt);
    int m = (l + r) >> 1;
    if(L <= m)
        UpdateSeg(lson, L, R);
    if(m < R)
        UpdateSeg(rson, L, R);
    PushUp(rt);
}
int Query(int l, int r, int rt, int L, int R)
{
    if(L <= l && r <= R)
    {
        return tree[rt];
    }
        PushDown(rt);
    int m = (l + r) >> 1;
    int ret = 1;
    if(L <= m)
        ret &= Query(lson, L, R);
    if(m < R)
        ret &= Query(rson, L, R);
    return ret;
}
int main()
{
    int n, m;
    int u, v;
    while(~scanf("%d", &n))
    {
        idx = 0;
        memset(g, 0, sizeof(g));
        memset(pos, 0, sizeof(pos));
        memset(tree, 0, sizeof(tree));
        for(int i = 1; i < n; i++)
        {
            scanf("%d%d", &u, &v);
            add(u, v);
            add(v, u);
        }
        idx = 0;
        fa[1] = -1;
        dfs(1);
        scanf("%d", &m);
        while(m--)
        {
            scanf("%d%d", &u, &v);
            if(u == 1)
            {
                int tmp = Query(1, n, 1, pos[v], son[v]);
                UpdateSeg(1, n, 1, pos[v], son[v]);
                if(tmp == 0 && fa[v] != -1)
                    UpdatePoint(1, n, 1, pos[fa[v]]);
            }
            else if(u == 2)
            {
                UpdatePoint(1, n, 1, pos[v]);
            }
            else if(u == 3)
            {
                printf("%d\n", Query(1, n, 1, pos[v], son[v]));
            }
        }
    }
    return 0;
}