zoj 3686 简单线段树更新 好久没写这个东东了!!!

/*
感觉是颗线段树,一开始没什么好的思路,偷窥了网上的转换方法!
题目大意:
一棵树有n个结点(1<=N<=100000),根为1,每个结点有一个标签,初始值为0.
现在有2种操作
o操作:(o,x)对结点x及其所有子孙结点的标签值取反(和1异或)。
q操作:(q,x)输出结点x及其所以子孙结点的标签值为1的个数。
操作次数为m次(1<=M<=10000)。
题目思路:


从根开始中序遍历每一个结点,给每一个结点按照遍历顺序重新编号(beg)。


这样子可以使得重新编号的结点,根及其连接的子树的结点是连续的,这样我们就可以操作时区间更新和区间询问了。


并且遍历时记录下某根子树中最大的结点编号(end),这样我们更新或询问某结点及其子树的范围即


[ beg[x] , end[x] ]。
*/
#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
using namespace std;
#define lson rt<<1,l,mid
#define rson rt<<1|1,mid+1,r
const int maxn=100002;
int beg[maxn],end[maxn],head[maxn],sum[maxn<<2],col[maxn<<2];
int NE=0,n,m,tot;
char op[2];
struct node
{
    int u,v,next;
}Edge[maxn<<2];
void addEdge(int u,int v)
{
    Edge[NE].u=u,Edge[NE].v=v,Edge[NE].next=head[u];
    head[u]=NE++;
}
int dfs(int rt)
{
    beg[rt]=++tot;
    for(int i=head[rt];i!=-1;i=Edge[i].next)dfs(Edge[i].v);
    end[rt]=tot;
}
void PushUp(int rt)
{
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
void build(int rt,int l,int r)
{
    col[rt]=sum[rt]=0;
    if(l==r)return;
    int mid=(l+r)>>1;
    build(lson);
    build(rson);
   // PushUp(rt);
}
void PushDown(int rt,int l,int r)
{
    if(col[rt]!=0)
    {
        int mid=(l+r)>>1;
        col[rt<<1]^=col[rt];
        col[rt<<1|1]^=col[rt];
        sum[rt<<1]=mid-l+1-sum[rt<<1];
        sum[rt<<1|1]=r-mid-sum[rt<<1|1];
        col[rt]=0;
    }
}
void update(int rt,int l,int r,int L,int R)
{
    if(L<=l&&r<=R)
    {
        col[rt]^=1;
        sum[rt]=r-l+1-sum[rt];
        return;
    }
    PushDown(rt,l,r);
    int mid=(l+r)>>1;
    if(mid>=L)update(lson,L,R);
    if(mid<R)update(rson,L,R);
    PushUp(rt);
}
int query(int rt,int l,int r,int L,int R)
{
    if(L<=l&&r<=R)return sum[rt];
    PushDown(rt,l,r);
    int mid=(l+r)>>1;
    int res=0;
    if(mid>=L)res+=query(lson,L,R);
    if(mid<R)res+=query(rson,L,R);
    return res;
}
int main()
{
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        NE=tot=0;
        int i,j,u,v;
        memset(head,-1,sizeof(head));
        for(v=2;v<=n;v++)
        {
            cin>>u;
            addEdge(u,v);
        }
        dfs(1);
        build(1,1,n);
        while(m--)
        {
            cin>>op>>u;
            if(op[0]=='o')update(1,1,n,beg[u],end[u]);
            else printf("%d\n",query(1,1,n,beg[u],end[u]));
        }
        puts("");
    }
    return 0;
}


你可能感兴趣的:(zoj 3686 简单线段树更新 好久没写这个东东了!!!)