codeforces 877 E. Danil and a Part-time Job(线段树(dfs序))
题目链接:http://codeforces.com/contest/877/problem/E
题解:显然一看就感觉要么树链剖分要么线段树+dfs序,题目要求的操作显然用线段树+dfs序就可以实现。然后就敲一下线段树+dfs序就行挺简单的只要dfs一遍记录当前节点的下表然后再加一个leng数组来存子树最多到达几然后更新或者求值的时候只要查询(pos[x],leng[x])即可。
#include <iostream> #include <cstring> #include <cstdio> #include <vector> #define lson (i << 1) #define rson ((i << 1) | 1) using namespace std; const int M = 2e5 + 10; int a[M] , pos[M] , leng[M] , pre[M] , cnt; vector<int> vc[M]; struct TnT { int l , r , sum , lazy; }T[M << 2]; void push_up(int i) { T[i].sum = T[lson].sum + T[rson].sum; } void push_down(int i) { if(T[i].lazy) { T[lson].sum = T[lson].r - T[lson].l + 1 - T[lson].sum; T[rson].sum = T[rson].r - T[rson].l + 1 - T[rson].sum; T[lson].lazy ^= 1; T[rson].lazy ^= 1; T[i].lazy = 0; } } void build(int l , int r , int i) { int mid = (l + r) >> 1; T[i].l = l , T[i].r = r , T[i].sum = 0 , T[i].lazy = 0; if(l == r) { T[i].sum = a[pre[l]]; return ; } push_down(i); build(l , mid , lson); build(mid + 1 , r , rson); push_up(i); } void update(int l , int r , int i) { int mid = (T[i].l + T[i].r) >> 1; if(T[i].l == l && T[i].r == r) { T[i].sum = (T[i].r - T[i].l + 1) - T[i].sum; T[i].lazy ^= 1; return ; } push_down(i); if(mid < l) { update(l , r , rson); } else if(mid >= r) { update(l , r , lson); } else { update(l , mid , lson) , update(mid + 1 , r , rson); } push_up(i); } int query(int l , int r , int i) { int mid = (T[i].l + T[i].r) >> 1; if(T[i].l == l && T[i].r == r) { return T[i].sum; } push_down(i); if(mid < l) { return query(l , r , rson); } else if(mid >= r) { return query(l , r , lson); } else { return query(l , mid , lson) + query(mid + 1 , r , rson); } } void dfs(int u) { int len = vc[u].size(); for(int i = 0 ; i < len ; i++) { int v = vc[u][i]; pos[v] = ++cnt; pre[cnt] = v; dfs(v); leng[v] = cnt; } } int main() { int n , x , q; char s[123]; scanf("%d" , &n); for(int i = 2 ; i <= n ; i++) { scanf("%d" , &x); vc[x].push_back(i); } for(int i = 1 ; i <= n ; i++) { scanf("%d" , &a[i]); } scanf("%d" , &q); cnt = 0; pos[1] = ++cnt; pre[cnt] = 1; leng[1] = n; dfs(1); build(1 , n , 1); while(q--) { scanf("%s" , s); if(s[0] == 'g') { scanf("%d" , &x); printf("%d\n" , query(pos[x] , leng[x] , 1)); } else { scanf("%d" , &x); update(pos[x] , leng[x] , 1); } } return 0; }