ZOJ 3686 A Simple Tree Problem (线段树)

题意类似于POJ3321，但是这里每次操作的时候更新了子树中的所有节点值，而且每次查询需要进行延迟标记，所以如果跟POJ3321一样用树状数组不适合，这里用线段树来写，开始先dfs一边用时间戳进行节点标记。

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
#define MAX 100001
#define lson (rt<<1)
#define rson (rt<<1|1)
#define ll int
int head[MAX],to[MAX<<1],next[MAX<<1],edge;
int begin[MAX],end[MAX],ind,n,m;
bool vis[MAX];
struct Node
{
    int l,r;
    ll sum;
    int add;
    int len()
    {
        return r-l+1;
    }
} tree[MAX<<2];
void init()
{
    memset(head,-1,sizeof(head));
    memset(vis,0,sizeof(vis));
    edge=ind=0;
}
inline void add(int u,int v)
{
    to[edge]=v,next[edge]=head[u],head[u]=edge++;
    to[edge]=u,next[edge]=head[v],head[v]=edge++;
}
inline void pushUp(int rt)
{
    tree[rt].sum=tree[lson].sum+tree[rson].sum;
}
inline void pushDown(int rt)
{
    int add=tree[rt].add%2;
    if(add)
    {
        tree[lson].add+=add;
        tree[lson].sum=tree[lson].len()-tree[lson].sum;
        tree[rson].add+=add;
        tree[rson].sum=tree[rson].len()-tree[rson].sum;
        tree[rt].add=0;
    }
}
void build(int rt,int l,int r)
{
    tree[rt].l=l,tree[rt].r=r,tree[rt].add=0;
    if(l==r)
    {
        tree[rt].sum=0;
        return ;
    }
    int mid=(l+r)>>1;
    build(lson,l,mid);
    build(rson,mid+1,r);
    pushUp(rt);
}
void update(int rt,int l,int r)
{
    if(l<=tree[rt].l&&r>=tree[rt].r)
    {
        tree[rt].add++;
        tree[rt].sum=tree[rt].len()-tree[rt].sum;
        return;
    }
    pushDown(rt);
    int mid=(tree[rt].l+tree[rt].r)>>1;
    if(l<=mid) update(lson,l,r);
    if(r>mid) update(rson,l,r);
    pushUp(rt);
}
ll query(int rt,int l,int r)
{
    if(l<=tree[rt].l&&r>=tree[rt].r)
    {
        return tree[rt].sum;
    }
    pushDown(rt);
    int mid=(tree[rt].l+tree[rt].r)>>1;
    ll res=0;
    if(l<=mid) res+=query(lson,l,r);
    if(r>mid) res+=query(rson,l,r);
    return res;
}
void dfs(int now)
{
    begin[now]=++ind;
    vis[now]=1;
    for(int i=head[now]; ~i; i=next[i])
        if(!vis[to[i]])
            dfs(to[i]);
    end[now]=ind;
}
int main()
{
    int i,u,m;
    char ch[2];
    while(~scanf("%d%d",&n,&m))
    {
        init();
        build(1,1,n);
        for(i=2; i<=n; ++i)
        {
            scanf("%d",&u);
            add(u,i);
        }
        dfs(1);
        while(m--)
        {
            scanf("%s",ch);
            if(ch[0]=='q')
            {
                scanf("%d",&u);
                printf("%d\n",query(1,begin[u],end[u]));
            }
            else
            {
                scanf("%d",&u);
                update(1,begin[u],end[u]);
            }
        }
        printf("\n");
    }
    return 0;
}