http://poj.org/problem?id=3321
这道题涉及到怎么把多叉树的查询转化为树状数组的前缀和查询
思想:通过某种办法将x转化为一个区间[L,R];
一个连续的区间可以代表一棵子树
代码块语法遵循标准markdown代码,例如:
#include <iostream>
#include <cstdio>
#include <vector>
using namespace std;
/*
id[i]编号为i的树杈对应的后序遍历次序
size[i]编号为i的树杈对应的子树规模
转化为BIT问题,i->[L,R]:R=id[i],L=R-size[i]+1
x[id[i]]代表是否有果
前缀和sum(id[i])代表所有后序遍历在id[i]之前的减少的苹果数量
*/
int id[100010]={0},size[100010]={0};
int BIT[100010]={0},x[100010]={0};
typedef vector<int> ve;
void postOrder(int subroot,vector<ve> &adj,bool book[],int& subsize,int &order){
//cout<<"traverse subroot "<<subroot<<endl;
//subsize当前子树规模
bool leave=1;//当前树是否为叶子
for(int i=0;i<adj[subroot].size();i++){
if(!book[adj[subroot][i]]){
leave=0;
book[adj[subroot][i]]=1;
int tempsize=0;//子树返回的规模
postOrder(adj[subroot][i],adj,book,tempsize,order);
//cout<<"tempsize "<<adj[subroot][i]<<": "<<tempsize<<endl;
subsize+=tempsize;
}
}
if (leave)
{
subsize=1;
size[subroot]=1;
}
else{
subsize++;
size[subroot]=subsize;
}
id[subroot]=++order;
//cout<<"traverse subroot "<<subroot<<" end"<<endl;
}
void add(int,int,int);
int Sum(int);
int main(){
int n;
cin>>n;
//邻接表用于进行后序遍历
//vector<vector<int>> adj(n+1);
vector<ve> adj(100002);
for(int i=1;i<n;i++){
int f1,f2;
scanf("%d%d",&f1,&f2);
adj[f1].push_back(f2);
adj[f2].push_back(f1);
}
//后序遍历
bool book[100010]={0};
book[1]=1;
int tsize=0,order=0;
postOrder(1,adj,book,tsize,order);
/*
for(int i=1;i<=n;i++){
cout<<"id "<<i<<": "<<id[i]<<endl;
cout<<"size "<<i<<": "<<size[i]<<endl;
}
*/
int m,fork;
char ins;
cin>>m;
while(m--){
cin>>ins;
cin>>fork;
int postid=id[fork];
if(ins=='C'){
if(x[postid]==0){
x[postid]=1;
add(postid,1,n);
}else{
x[postid]=0;
add(postid,-1,n);
}
/*
for(int i=1;i<=n;i++){
cout<<"x["<<i<<"] "<<x[i]<<endl;
cout<<"BIT["<<i<<"] "<<BIT[i]<<endl;
}
*/
}else{
int sumR=Sum(postid);
//cout<<"sumR: "<<sumR<<endl;
int sumL=Sum(postid-size[fork]);
//cout<<"sumL: "<<sumL<<endl;
cout<<size[fork]-sumR+sumL<<endl;
}
}
}
void add(int i,int num,int n){
while(i<=n){
BIT[i]+=num;
i=i+(-i&i);
}
}
int Sum(int i){
int res=0;
while(i>0){
res+=BIT[i];
i=i-(-i&i);
}
return res;
}
不懂的地方:
有一个地方会卡超时:
vector<vector<int>> adj(100010);
这样就会超时,不知道为什么,有知道为什么的人可以解答一下。