剛見到此題時,我認爲此題和poj2352star可以用類似的模型,然後寫了一個程序,結果wa了,後來又仔細讀了遍題目,發現之前題目理解錯誤。
這題難點在於把樹映射到數組中,然而樹節點的序號並不能達到要求,我們可以求出每個節點管理的左值和右值,這裏可以用dfs實現。知道左值與右值,就可以很快得出這點所管理的蘋果的個數sum(right)-sum(left-1)。有個難點在於如何dfs,因爲一個根節點管理的分支數目不定,是個n叉樹。我一開始想到的存儲方法是用指針數組來記錄他的下面的分支。《挑戰》上面的圖的存儲方法正好可以用到這裏,用vector就方便很多。如果對這裏的dfs還是不太瞭解的,可以看看《挑戰》97頁二分圖判定那個例題。
#include<iostream>
#include <cstdio>
#include <cstring>
#include <vector>
#define MAXN 100005
#define mem(a) memset(a, 0, sizeof(a))
using namespace std;
int c[MAXN], Left[MAXN], Right[MAXN], Fork[MAXN];
typedef vector<int> Ve;
vector<Ve>Edge(MAXN);
int N,M;
int key;
void init()//初始化數組和
{
mem(Left); mem(Right);
mem(Fork); mem(c);
for(int i=0;i<MAXN;i++)Edge[i].clear();
}
void DFS(int node)//爲每一個node添加一個左值和右值,表示這個節點所
{
Left[node] = key;
for(int i=0;i<Edge[node].size();i++)
{
key+=1;
DFS(Edge[node][i]);
}
Right[node] = key;
}
int LowBit(int x)//返回的是2^k
{
return x & (x ^ (x-1));
}
void add(int k, int num)//修改節點k,如果是添加一個,代入1,刪除一個代入-1
{
while(k <= N)
{
c[k] += num;
k += LowBit(k);
}
}
int GetSum(int k)//得到1...k的和
{
int sum = 0;
while(k>=1)
{
sum += c[k];
k -= LowBit(k);
}
return sum;
}
void ReadDataAndDo()
{
int a,b;
char ch;
for(int i=1;i<N;i++)//輸入a,b把邊存放在容器裏面
{
scanf("%d%d", &a, &b);
Edge[a].push_back(b);
}
key = 1; DFS(1);//爲每一個節點對應一個左邊界和右邊界,他自己就存放在左邊界裏面,而它的管轄範圍就是左邊界到右邊界
for(int i=1;i<=N;i++)
{
Fork[i] = 1;//最初每個Fork上都有一個蘋果
add(i,1);//同時更新樹狀數組的值
}
scanf("%d%*c", &M);
for(int i=0;i<M;i++)
{
scanf("%c %d%*c", &ch, &b);
if(ch == 'Q')//b的子樹就是[Left[b], right[b]]
{
printf("%d\n", GetSum(Right[b]) - GetSum(Left[b]-1));
}
else
{
if(Fork[b]) add(Left[b],-1);//由於每個節點的編號就是它的左值,所以直接修改左節點
else add(Left[b],1);
Fork[b] = !Fork[b];//變爲相反的狀態
}
}
}
int main()
{
while(~scanf("%d", &N))
{
init();
ReadDataAndDo();
}
return 0;
}