CodeForces 343D 線段樹維護dfs序

標籤：

給定一棵樹，初始時樹為空白

操作1，往某個結點注水，那麼該結點的子樹都注滿了水

操作2，將某個結點的水放空，那麼該結點的父親的水也就放空了

操作3，詢問某個點是否有水

 

我們將樹進行dfs， 產生in[u]， 訪問結點u的時間戳記，out[u]，離開結點u的時間戳記

每個結點的in值對應線上段樹中的區間的一點

那麼對於操作1， 只要將區間[in[u],out[u]] 的值都改為1, 但是如果區間[in[u],out[u]] 原先存在為0的點，那麼父區間肯定是空的，這個操作不能

改變父區間的狀態，所以需要將in[fa[u]]置為0,

對於操作2，只要將in[u] 置為0就行了

對於操作3，只要區間[in[u],out[u]]內不存在為0的點，那麼點u就是有水的

  1 #include <stdio.h>  2 #include <string.h>  3 #include <stdlib.h>  4 #include <algorithm>  5 #include <iostream>  6 #include <queue>  7 #include <stack>  8 #include <vector>  9 #include <map> 10 #include <set> 11 #include <string> 12 #include <math.h> 13 using namespace std; 14 #pragma warning(disable:4996) 15 #pragma comment(linker, "/STACK:1024000000,1024000000") 16 typedef long long LL;                    17 const int INF = 1<<30; 18 /* 19 我們對樹進行dfs，求出dfs序列 in[]  和out[]   用線段樹維護這個序列 20 操作1就是將[in[b],out[b]]區間內的點都變為1 ，  如果區間記憶體在0，那麼要將b的父結點更新為0，因為雖然更新了[in[b],out[b]]，但是祖先不會變 21 操作2就是將點in[b]置為0 22 操作3就是看區間[in[b],out[b]]  23  24  25 所以不僅可以將樹進行樹鏈剖分，然後用線段樹維護 26 還可以用線段樹維護樹的dfs序列 27 */ 28 const int N = 500000 + 10; 29 int in[N], out[N], num, fa[N]; 30 int tree[N * 8], laze[N * 8]; 31 vector<int> g[N]; 32 void dfs(int u) 33 { 34     in[u] = ++num; 35     for (int i = 0; i < g[u].size(); ++i) 36     { 37         int v = g[u][i]; 38         if (v != fa[u]) 39         { 40             fa[v] = u; 41             dfs(v); 42         } 43     } 44     out[u] = num; 45 } 46 void pushDown(int rt) 47 { 48     if (laze[rt]==1) 49     { 50         tree[rt << 1] = tree[rt << 1 | 1] = laze[rt << 1] = laze[rt << 1 | 1] = 1; 51         laze[rt] = 0; 52     } 53 } 54 void pushUp(int rt) 55 { 56     if (tree[rt << 1] == 0 || tree[rt << 1 | 1] == 0) 57         tree[rt] = 0; 58     else 59         tree[rt] = 1; 60 } 61 bool flag = false; 62 void update(int l, int r, int rt, int L, int R, int val) 63 { 64     pushDown(rt); 65     if (L<=l && R>=r) 66     { 67         if (tree[rt] == 0) 68             flag = true; 69         tree[rt] = val; 70         laze[rt] = val; 71         return; 72     } 73     int mid = (l + r) >> 1; 74     if (L <= mid) 75         update(l, mid, rt << 1, L, R, val); 76     if (R > mid) 77         update(mid + 1, r, rt << 1 | 1, L, R, val); 78     pushUp(rt); 79 } 80  81 void query(int l, int r, int rt, int L, int R) 82 { 83     pushDown(rt); 84     if (L<=l && R>=r) 85     { 86         //區間內有一個是0，那麼所訪問的點（根）就是0 87         if (tree[rt] == 0) 88             flag = false; 89         return; 90     } 91     int mid = (l + r) >> 1; 92     if (L <= mid) 93         query(l, mid, rt << 1, L, R); 94     if (R > mid) 95         query(mid + 1, r, rt << 1 | 1, L, R); 96     pushUp(rt); 97 } 98 int main() 99 {100     //freopen("d:/in.txt", "r", stdin);101     int n, q, a, b,x, y;102     scanf("%d", &n);103     for (int i = 1; i < n; ++i)104     {105         scanf("%d%d", &a, &b);106         g[a].push_back(b);107         g[b].push_back(a);108     }109     fa[1] = 0;110     dfs(1);111     scanf("%d", &q);112     while (q--)113     {114         scanf("%d%d", &a, &b);115         if (a == 1)116         {117             flag = false;118             //將區間內的點都置為1119             update(1,  n, 1, in[b], out[b],1);120             //如果區間記憶體在0的點，那麼fa[b]肯定為空白，121             if (flag&&fa[b]!=0)122             {123                 update(1,  n, 1, in[fa[b]],in[fa[b]],0);124             }125         }126         else if (a == 2)127         {128             update(1,  n, 1, in[b],in[b],0);129         }130         else131         {132             flag = true;133             query(1,n, 1, in[b], out[b]);134             if (flag)135                 printf("1\n");136             else137                 printf("0\n");138         }139     }140     return 0;141 }

 

CodeForces 343D 線段樹維護dfs序