nodeoj2000——Freda's Chess

DFS序加线段树，很模板

---

 1 #include <cstdio>
 2 #include <cstring>
 3 #include <iostream>
 4 #include <cmath>
 5 #include <algorithm>
 6 using namespace std;
 7 const int maxn=200005;
 8 int n,m,y,x,idx,cnt,in[maxn],head[maxn],out[maxn];
 9 int ct,root=1;
10 char ju;
11 struct edge{
12     int v,nxt;
13 }d[maxn];
14 inline void add(int a,int b){
15     d[++cnt].v=b;d[cnt].nxt=head[a];head[a]=cnt;
16     d[++cnt].v=a;d[cnt].nxt=head[b];head[b]=cnt;
17 }
18 struct TREE{
19     int l,r,sum;
20 }t[maxn<<2];
21 void dfs(int u,int fa){
22     in[u]=++idx;
23     for(int i=head[u];i;i=d[i].nxt){
24         int v=d[i].v;
25         if(v==fa)continue;
26         dfs(v,u);
27     }
28     out[u]=idx;
29 }
30 void build(int ro,int l,int r){
31     t[ro].l=l,t[ro].r=r;
32     if(l==r){
33         t[ro].sum=1;
34         return;
35     }
36     int mid=l+r>>1;
37     build(ro<<1,l,mid);
38     build(ro<<1|1,mid+1,r);
39     t[ro].sum=t[ro<<1].sum+t[ro<<1|1].sum;
40 }
41 void update(int ro,int pos){
42     if(t[ro].l==pos&&t[ro].r==pos){
43         t[ro].sum^=1;
44         return;
45     }
46     int mid=t[ro].l+t[ro].r>>1;
47     if(pos<=mid)update(ro<<1,pos);
48     else update(ro<<1|1,pos);
49     t[ro].sum=t[ro<<1].sum+t[ro<<1|1].sum;
50 }
51 int query(int ro,int l,int r){
52     if(t[ro].l==l&&t[ro].r==r){
53         return t[ro].sum;
54     }
55     int mid=t[ro].l+t[ro].r>>1;
56     if(r<=mid)return query(ro<<1,l,r);
57     else if(l>mid)return query(ro<<1|1,l,r);
58     else return query(ro<<1,l,mid)+query(ro<<1|1,mid+1,r);
59 }
60 
61 
62 int main(){
63     ios::sync_with_stdio(false);
64     cin>>n;
65     for(int i=1;i<n;++i){
66         cin>>x>>y;
67         add(x,y);
68     }
69     dfs(1,0);
70     build(root,1,idx);
71     cin>>m;
72     while(m--){
73         cin>>ju>>x;
74         if(ju=='C')update(root,in[x]);
75         else printf("%d\n",query(root,in[x],out[x]));
76     }
77     return 0;
78 }

 

转载于:https://www.cnblogs.com/AT-HENS/p/7756869.html