zoj 3686 A Simple Tree Problem

题目链接：http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemCode=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 <stdlib.h>#include <string.h>#include <stdio.h>#include <ctype.h>#include <math.h>#include <stack>#include <queue>#include <map>#include <set>#include <vector>#include <string>#include <iostream>#include <algorithm>using namespace std;#define ll __int64#define ls rt<<1#define rs ls|1#define lson l,mid,ls#define rson mid+1,r,rs#define middle (l+r)>>1#define clr_all(x,c) memset(x,c,sizeof(x))#define clr(x,c,n) memset(x,c,sizeof(x[0])*(n+1))#define eps (1e-8)#define MOD 1000000007#define INF 0x3f3f3f3f#define PI (acos(-1.0))#pragma comment(linker, "/STACK:102400000,102400000")template <class T> T _max(T x,T y){return x>y? x:y;}template <class T> T _min(T x,T y){return x<y? x:y;}template <class T> T _abs(T x){return (x < 0)? -x:x;}template <class T> T _mod(T x,T y){return (x > 0)? x%y:((x%y)+y)%y;}template <class T> void _swap(T &x,T &y){T t=x;x=y;y=t;}template <class T> void getmax(T& x,T y){x=(y > x)? y:x;}template <class T> void getmin(T& x,T y){x=(x<0 || y<x)? y:x;}int TS,cas=1;const int M=100000+5;int n,m;struct node{int beg,end;}mp[M];int fa[M],nxt[M],vec[M],tot;void dfs(int rt){mp[rt].beg=++tot;for(int i=fa[rt];i!=-1;i=nxt[i]) dfs(vec[i]);mp[rt].end=tot;}int cov[M<<2],sum[M<<2];void build(int l,int r,int rt){cov[rt]=sum[rt]=0;if(l==r) return;int mid=middle;build(lson),build(rson);}void pushDown(int l,int mid,int r,int rt){if(!cov[rt]) return;sum[ls]=(mid-l+1)-sum[ls],cov[ls]^=1;sum[rs]=(r-mid)-sum[rs],cov[rs]^=1;cov[rt]=0;}void pushUp(int rt){sum[rt]=sum[ls]+sum[rs];}void update(int l,int r,int rt,int L,int R){if(L<=l && r<=R){sum[rt]=(r-l+1)-sum[rt];cov[rt]^=1;return;}int mid=middle;pushDown(l,mid,r,rt);if(R<=mid) update(lson,L,R);else if(mid<L) update(rson,L,R);else update(lson,L,mid),update(rson,mid+1,R);pushUp(rt);}int query(int l,int r,int rt,int L,int R){if(L<=l && r<=R) return sum[rt];int mid=middle;pushDown(l,mid,r,rt);if(R<=mid) return query(lson,L,R);else if(mid<L) return query(rson,L,R);else return query(lson,L,mid)+query(rson,mid+1,R);}void run(){int i,j;clr(fa,-1,n+1);tot=0;for(i=2;i<=n;i++){scanf("%d",&j);nxt[tot]=fa[j],vec[tot]=i,fa[j]=tot++;}tot=0;dfs(1);build(1,n,1);while(m--){char op[3];scanf("%s%d",op,&j);if(op[0] == 'o') update(1,n,1,mp[j].beg,mp[j].end);else printf("%d\n",query(1,n,1,mp[j].beg,mp[j].end));}puts("");}void preSof(){}int main(){    //freopen("input.txt","r",stdin);    //freopen("output.txt","w",stdout);    preSof();    //run();while((~scanf("%d%d",&n,&m))) run();     //for(scanf("%d",&TS);cas<=TS;cas++) run();    return 0;}