Codeforces Round #200 (Div. 1)D Water Tree

点击打开链接

题意就是给一棵树，三种操作，操作1是把某一个结点v和它的所有的孩子结点全都变成1，操作2是把一个结点和它的祖先结点全都变成0，操作三是询问某一个结点和它的儿子结点什么的是不是都是1

显而易见可以用线段树来维护，首先深搜出来结点的序列作为一个线性序列来维护。

对于操作1，非常简单，把进入结点时间和出结点时间当做是一个区间进行覆盖就行。

对于操作2，麻烦了，我们无法做到快速地找到一个结点的祖先链，这怎么办呢？不过考虑一下这个问题的性质，我们可以只更新操作的那个结点，如果当前结点不是满的，那它一定有一个孩子结点被操作清空了，那么它就一定应该是空的（是不是满的直接检查是否等于区间长度即可）。

现在又有了一个问题，如果它不是满的，可是它又被填满了怎么办？

很简单，它不是满的，它父亲肯定就不是满的，填满它，跟它父亲没有关系，把它父亲清空了即可。

这样的话这道题就愉快的结束了。

//By qhpeklh5959, contest: Codeforces Round #200 (Div. 1), problem: (D) Water Tree, Accepted, ##include <cstdio>#include <cstring>#include <algorithm>#include <vector>using namespace std;#define lson l, m, rt << 1#define rson m + 1, r, rt << 1 | 1const int maxn = (int)5e5 + 10;int sum[maxn << 2], tag[maxn << 2];vector<int> chd[maxn];int pre[maxn], post[maxn], fa[maxn];void PushDown(int l, int r, int rt){int m = (l + r) >> 1;if (tag[rt] != -1){sum[rt << 1] = m - l + 1;sum[rt << 1 | 1] = r - m;tag[rt << 1] = tag[rt << 1 | 1] = tag[rt];tag[rt] = -1;}}void PushUp(int rt){sum[rt] = sum[rt << 1] + sum[rt << 1 | 1];}void build(int l, int r, int rt){tag[rt] = -1; sum[rt] = 0;if (l == r) return;int m = (l + r) >> 1;build(lson);build(rson);}void update(int ll, int rr, int f, int l, int r, int rt){if (ll <= l && rr >= r){sum[rt] = (r - l + 1) * f;tag[rt] = f;return;}PushDown(l, r, rt);int m = (l + r) >> 1;if (ll <= m) update(ll, rr, f, lson);if (rr > m) update(ll, rr, f, rson);PushUp(rt);}int query(int ll, int rr, int l, int r, int rt){if (ll <= l && rr >= r) return sum[rt];PushDown(l, r, rt);int m = (l + r) >> 1;int ret = 0;if (ll <= m) ret += query(ll, rr, lson);if (rr > m) ret += query(ll, rr, rson);return ret;}int cnt; bool vis[maxn];void dfs(int u){pre[u] = ++cnt; vis[u] = 1;for (int i = 0; i < chd[u].size(); i++){int v = chd[u][i];if (!vis[v]) dfs(v);}post[u] = cnt;}int n, m;int main(){#ifndef ONLINE_JUDGEfreopen("in", "rt", stdin);#endifwhile(scanf("%d", &n) == 1){memset(vis, 0, sizeof(vis));for (int i = 0; i <= n; i++)chd[i].clear();memset(pre, 0, sizeof(pre));memset(post, 0, sizeof(post));memset(fa, 0, sizeof(fa));int x, y;for (int i = 1; i < n; i++){scanf("%d%d", &x, &y);if (x > y) swap(x, y);chd[x].push_back(y);chd[y].push_back(x);fa[y] = x;}fa[1] = 1;cnt = 0;dfs(1);build(1, cnt, 1);scanf("%d", &m);int op;while(m--){scanf("%d%d", &op, &x);if (op == 1){int tt = query(pre[x], post[x], 1, cnt, 1);if (tt != (post[x] - pre[x] + 1)) update(pre[fa[x]], pre[fa[x]], 0, 1, cnt, 1);update(pre[x], post[x], 1, 1, cnt, 1);}else if (op == 2) update(pre[x], pre[x], 0, 1, cnt, 1);else if (op == 3){int tt = query(pre[x], post[x], 1, cnt, 1);if (tt != (post[x] - pre[x] + 1)) puts("0");else puts("1");}}}return 0;}