题意:
给定一棵树,以及定义了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;
}