POJ3321 Apple Tree(DFS序 + 樹狀陣列)
題目連結:傳送門
題意:
給定一顆樹,初始的時候每個節點的值為1。
修改操作:C X 如果點X的值為1則變成0,如果點X的值為0則變成1.
查詢操作:Q X 查詢X這個節點以及以它為根的子樹的所有節點的和。
分析:
DFS序:根據dfs的時候節點進棧與出棧的時間點。in[u]/out[u]代表點u進/出棧的時間,那麼(in[u],out[u])之間的節點就是點u子樹上的節點。
我們根據DFS序來維護一個樹狀陣列就可以了。
程式碼如下:
#include <cstring>
#include <algorithm>
#include <cstdio>
#include <vector>
using namespace std;
const int maxn = 1e5+10;
int has[maxn];
int n;
struct Tree{
struct nod{
int to,next;
}edge[maxn];
int in[maxn],out[maxn],tt;
int head[maxn],ip;
void init(){
tt=0,ip=0;
memset(head,-1,sizeof(head));
}
void add(int u,int v){
edge[ip].to= v;
edge[ip].next = head[u];
head[u]= ip++;
}
void dfs(int u){
in[u]=++tt;
for(int i=head[u];i!=-1;i=edge[i].next){
int v = edge[i].to;
dfs(v);
}
out[u]=tt;
}
}G;
struct BIT {
int sum[maxn];
void init(){
memset(sum,0,sizeof(sum));
}
int lowbit(int x) {
return x&(-x);
}
void update(int pos,int x){
while(pos<=n){
sum[pos]+=x;
pos+=lowbit(pos);
}
}
int query(int pos){
int ans = 0;
while(pos>0){
ans+=sum[pos];
pos-=lowbit(pos);
}
return ans;
}
}T;
int main() {
while(~scanf("%d",&n)){
G.init();
T.init();
for(int i=1;i<n;i++){
int u,v;
scanf("%d%d",&u,&v);
G.add(u,v);
}
G.dfs(1);
for(int i=1;i<=n;i++){
T.update(G.in[i],1);
has[i]=1;
}
int x,m;
scanf("%d",&m);
char s[2];
while(m--){
scanf("%s%d",s,&x);
if(s[0]=='C'){
if(has[x])
T.update(G.in[x],-1);
else
T.update(G.in[x],1);
has[x]^=1;
}
else
printf("%d\n",T.query(G.out[x])-T.query(G.in[x]-1));
}
}
return 0;
}
相關文章
- POJ 3321 Apple Tree(dfs+樹狀陣列)APP陣列
- POJ 3321-Apple Tree(樹狀陣列)APP陣列
- hdu 4836 The Query on the Tree(線段樹or樹狀陣列)陣列
- AC自動機+字典序+樹狀陣列陣列
- 樹狀陣列陣列
- bzoj1146: [CTSC2008]網路管理Network(dfs序+主席樹+樹狀陣列)陣列
- 樹的DFS序
- 解析樹狀陣列陣列
- 樹狀陣列詳解陣列
- 樹狀陣列基礎陣列
- poj 2481 樹狀陣列陣列
- hdu 3874 樹狀陣列陣列
- 二維樹狀陣列陣列
- 樹狀陣列模板題 & (樹狀陣列 1:單點修改,區間查詢)陣列
- Codefroces 1328E Tree Querie(dfs序)
- 樹狀陣列和逆序對陣列
- hdu 5147 樹狀陣列陣列
- 樹狀陣列快速入門陣列
- Codeforces Round #225 (Div. 2)(B思維題,E:dfs+樹狀陣列)陣列
- 樹狀陣列模板+習題集陣列
- 樹狀陣列3種基本操作陣列
- 學習筆記----樹狀陣列筆記陣列
- 樹狀陣列upc1976陣列
- CSU 4441 Necklace (樹狀陣列/LIS)陣列
- 樹狀陣列(我是真小白)陣列
- 資料結構——樹狀陣列資料結構陣列
- 線段樹+差分——【模板】樹狀陣列2陣列
- POJ 2486 Apple Tree(樹形dp)APP
- 10:Challenge 3(樹狀陣列直接修改)陣列
- POJ 3928 Ping pong(樹狀陣列)陣列
- HDU 1556 Color the ball(線段樹|樹狀陣列)陣列
- CF 293 E Close Vertices (樹的分治+樹狀陣列)陣列
- HDU 1166 敵兵佈陣 (樹狀陣列)陣列
- HDU 1166 敵兵佈陣(樹狀陣列)陣列
- CF1967C. Fenwick Tree-運算元展開,樹狀陣列的結構陣列
- POJ-2352 Stars(樹狀陣列)陣列
- 【luogu3368】模板 樹狀陣列 2陣列
- 【二維樹狀陣列】poj 2155 Matrix陣列