CodeForces 343D Water Tree dfs序 + 线段树

题目：http://codeforces.com/problemset/problem/343/D

题意：给定一个树，树上有n个点，每个点是一个蓄水池，初始全为空。首先输入一个n，然后输入n - 1行，每行两个点，代表两点之间有边，然后输入一个m，接下来m行操作，操作有3种：1 a，把a及a的所有子孙注水。2 a，把a及a的所有祖先放水。3 a，询问a点有没有水，有输出1，否则0

思路：首先dfs扫一遍，记录每个点的访问时间in[v]和访问完其子孙的时间out[v]，那么其所有子孙必然包含在in[v]到out[v]中间，就转化成线性关系，用线段树去操作。对于操作2，要把其所有祖先更新一遍很麻烦，于是我们先单点更新，并在之后去做一系列处理。对于操作1，用线段树lazy操作，在向下更新时，要判断当前点的子孙中有没有为空的，若有说明这个子孙的所有祖先全为空，于是当前点及其子孙全注水，把当前点的父节点标记为空。对于操作3，要判断其所有子孙中有没有为空的，因此，我们用线段树维护区间内有没有为空的点。

总结：对于树形的题目，第一步要先想办法转换成线性，之后的操作就容易了一些

#include <cstdio>  
#include <cstring>  
#include <vector>  
#include <algorithm>  
using namespace std;  
  
const int N = 500010;  
struct edge  
{  
    int to, next;  
}g[N*2];  
struct node  
{  
    int l, r, val, mark;  
}s[N*4];  
int head[N], in[N], out[N], fat[N];  
int n, cnt, num;  
bool f;  
void add_edge(int v, int u)  
{  
    g[cnt].to = u;  
    g[cnt].next = head[v];  
    head[v] = cnt++;  
}  
void dfs(int v, int fa)  
{  
    in[v] = ++num;  
    for(int i = head[v]; i != -1; i = g[i].next)  
    {  
        int u = g[i].to;  
        if(u != fa)  
        {  
            fat[u] = v;  
            dfs(u, v);  
        }  
    }  
    out[v] = num;  
}  
void push_up(int k)  
{  
    if(s[k<<1].val == 1 && s[k<<1|1].val == 1) s[k].val = 1;/*子孙全有水则此节点也有水，否则便为空*/  
    else s[k].val = 0;  
}  
void push_down(int k)  
{  
    if(s[k].mark)  
    {  
        s[k<<1].val = s[k<<1|1].val = s[k].mark;  
        s[k<<1].mark = s[k<<1|1].mark = s[k].mark;  
        s[k].mark = 0;  
    }  
}  
void build(int l, int r, int k)  
{  
    s[k].l = l, s[k].r = r, s[k].val = 0, s[k].mark = 0;  
    if(l == r) return;  
    int mid = (l + r) >> 1;  
    build(l, mid, k << 1);  
    build(mid + 1, r, k << 1|1);  
}  
void update(int l, int r, int c, int k)  
{  
    if(l == 0) return;  
    if(l <= s[k].l && s[k].r <= r)  
    {  
        if(s[k].val == 0) f = true; /*其子孙中有空点，标记，之后把父节点置空*/  
        s[k].val = c;  
        if(c == 1) s[k].mark = c;  
        return;  
    }  
    push_down(k);  
    int mid = (s[k].l + s[k].r) >> 1;  
    if(l <= mid) update(l, r, c, k << 1);  
    if(r > mid) update(l, r, c, k << 1|1);  
    push_up(k);  
}  
void query(int l, int r, int k)  
{  
    if(l <= s[k].l && s[k].r <= r)  
    {  
        if(s[k].val == 0) f = true;/*其子孙中有空点，被查询的点也是空点*/
        return;  
    }  
    push_down(k);  
    int mid = (s[k].l + s[k].r) >> 1;  
    if(l <= mid) query(l, r, k << 1);  
    if(r > mid) query(l, r, k << 1|1);  
}  
int main ()  
{  
    int a, b;  
    scanf("%d", &n);  
    cnt = num = 0;  
    memset(head, -1, sizeof head);  
    for(int i = 1; i <= n - 1; i++)  
    {  
        scanf("%d%d", &a, &b);  
        add_edge(a, b);  
        add_edge(b, a);  
    }  
    fat[1] = 0;  
    dfs(1, 0);  
    build(1, num, 1);  
    int m;  
    scanf("%d", &m);  
    while(m--)  
    {  
        scanf("%d%d", &a, &b);  
        if(a == 1)  
        {  
            f = false;  
            update(in[b], out[b], 1, 1);  
            if(f) update(in[fat[b]], in[fat[b]], 0, 1);  
        }  
        else if(a == 2) update(in[b], in[b], 0, 1);  
        else  
        {  
            f = false;  
            query(in[b], out[b], 1);  
            if(f) printf("0\n");  
            else printf("1\n");  
        }  
    }  
    return 0;  
}