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 + ;
int a[M] , pos[M] , leng[M] , pre[M] , cnt;
vector<int> vc[M];
struct TnT {
    int l , r , sum , lazy;
}T[M << ];
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 +  - T[lson].sum;
        T[rson].sum = T[rson].r - T[rson].l +  - T[rson].sum;
        T[lson].lazy ^= ;
        T[rson].lazy ^= ;
        T[i].lazy = ;
    }
}
void build(int l , int r , int i) {
    int mid = (l + r) >> ;
    T[i].l = l , T[i].r = r , T[i].sum =  , T[i].lazy = ;
    if(l == r) {
        T[i].sum = a[pre[l]];
        return ;
    }
    push_down(i);
    build(l , mid , lson);
    build(mid +  , r , rson);
    push_up(i);
}
void update(int l , int r , int i) {
    int mid = (T[i].l + T[i].r) >> ;
    if(T[i].l == l && T[i].r == r) {
        T[i].sum = (T[i].r - T[i].l + ) - T[i].sum;
        T[i].lazy ^= ;
        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 +  , r , rson);
    }
    push_up(i);
}
int query(int l , int r , int i) {
    int mid = (T[i].l + T[i].r) >> ;
    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 +  , r , rson);
    }
}
void dfs(int u) {
    int len = vc[u].size();
    for(int i =  ; 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[];
    scanf("%d" , &n);
    for(int i =  ; i <= n ; i++) {
        scanf("%d" , &x);
        vc[x].push_back(i);
    }
    for(int i =  ; i <= n ; i++) {
        scanf("%d" , &a[i]);
    }
    scanf("%d" , &q);
    cnt = ;
    pos[] = ++cnt;
    pre[cnt] = ;
    leng[] = n;
    dfs();
    build( , n , );
    while(q--) {
        scanf("%s" , s);
        if(s[] == 'g') {
            scanf("%d" , &x);
            printf("%d\n" , query(pos[x] , leng[x] , ));
        }
        else {
            scanf("%d" , &x);
            update(pos[x] , leng[x] , );
        }
    }
    return ;
}