CodeForces 343D Water Tree(dfs序+线段树区间更新)

题意:

给定一棵树,以及定义了3个操作

1、把v点及其子树灌上水
2、把v点及v到根的路径去掉水
3、询问v点是否有水

解析:

先对这棵树做一遍dfs序,把树转成dfs_clock,这样每个点就可以对其子树进行区间更新。

那么对于点v 出现的时间in[v]和消失的时间out[v] ,一定会把v子树下所有节点都夹在[in[v], out[v]]之中。

对于操作1,就是把 [in[v], out[v]]改成1

对于操作2和3,把路径去掉水,显然是不存在这样直接到达根部的链,所以单点更新 in[v] = 0,若询问[in[u], out[u]]时 区间内存在一个0,则u子树下存在0,即u是没有水的。

值得注意的是,操作1还要注意一点就是如果子树内存在0,那么v的父节点就应该更新为0,因为将[in[v] ,out[v]]更新为1后v的祖先们不会改变(因为一个点是否为0看的是他所代表的区间内是否有0),这样才能保证算法的正确性。

my code

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
#define ls (o<<1)
#define rs (o<<1|1)
#define lson ls, L, M
#define rson rs, M+1, R
#define MID (L + R) >> 1
#define pb push_back
using namespace std;
const int N = (int)5e5 + 10;

int n, m;
vector<int> G[N];

void addEdge(int u, int v) {
    G[u].pb(v);
}

int dfs_clock;
int fa[N], in[N], out[N];
void dfs(int u, int pre) {
    in[u] = ++dfs_clock;
    fa[u] = pre;
    for(int i = 0; i < (int)G[u].size(); i++) {
        int v = G[u][i];
        if(v == pre) continue;
        dfs(v, u);
    }
    out[u] = dfs_clock;
}

int cov[N<<2], minv[N<<2];
void build(int o, int L, int R) {
    minv[o] = cov[o] = 0;
    if(L == R) return ;
    int M = MID;
    build(lson);
    build(rson);
}

inline void pushDown(int o) {
    if(cov[o] != -1) {
        cov[ls] = cov[rs] = cov[o];
        minv[ls] = minv[rs] = cov[o];
        cov[o] = -1;
    }
}

inline void pushUp(int o) {
    minv[o] = min(minv[ls], minv[rs]);
    if(cov[ls] == cov[rs])
        cov[o] = cov[ls];
    else cov[o] = -1;
}

void modify(int o, int L, int R, int ql, int qr, int val) {
    if(ql <= L && R <= qr) {
        minv[o] = cov[o] = val;
        return ;
    }
    int M = MID;
    pushDown(o);
    if(ql <= M) modify(lson, ql, qr, val);
    if(qr > M) modify(rson, ql, qr, val);
    pushUp(o);
}

int query(int o, int L, int R, int ql, int qr) {
    if(ql <= L && R <= qr) return minv[o];
    int M = MID;
    pushDown(o);
    if(qr <= M) return query(lson, ql, qr);
    else if(ql > M) return query(rson, ql, qr);
    else {
        int ret = min(query(lson, ql, M), query(rson, M+1, qr));
        return ret;
    }
}

void init() {
    dfs_clock = 0;
    for(int i = 1; i <= n; i++)
        G[i].clear();
}

int main() {
    int op, x;
    int u, v;
    while(~scanf("%d", &n)) {
        init();
        for(int i = 1; i < n; i++) {
            scanf("%d%d", &u, &v);
            addEdge(u, v);
            addEdge(v, u);
        }
        dfs(1, -1);
        build(1, 1, dfs_clock);

        scanf("%d", &m);
        while(m--) {
            scanf("%d%d", &op, &x);
            if(op == 1) {
                if(fa[x] != -1 && query(1, 1, n, in[fa[x]], out[fa[x]]) <= 0)
                    modify(1, 1, n, in[fa[x]], in[fa[x]], 0);           
                modify(1, 1, dfs_clock, in[x], out[x], 1);
            }else if(op == 2) {
                modify(1, 1, dfs_clock, in[x], in[x], 0);
            }else {
                int ret = query(1, 1, dfs_clock, in[x], out[x]);
                printf("%d\n", ret);
            }
        }
    }
    return 0;
}

你可能感兴趣的:(codeforces,343D)