POJ 3321 Apple Tree

最新推荐文章于 2024-12-28 16:15:10 发布

原创最新推荐文章于 2024-12-28 16:15:10 发布 · 908 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#apple #graph #query #struct #tree #null

http://acm.pku.edu.cn/JudgeOnline/problem?id=3321

#include <iostream>
using namespace std;

const int N=100001;

inline int lowbit(int x) //返回2^k,其中k是x末尾0的个数
{
return x & (x^(x-1)); //x & (-x)
}

struct Node
{
Node(int v=0,Node * n=NULL):vertex(v),next(n){}
int vertex;
Node * next;
};

Node * graph[N];

int DFSBegin[N],DFSEnd[N],depth,TreeArr[N]={0},n;
bool vst[N]={false};

int apple[N]={0};

void Modify(int pos,int d)
{
while(pos<=n)
{
TreeArr[pos]+=d;
pos+=lowbit(pos);
}
}

int Query(int pos)
{
int sum=0;
while(pos>0)
{
sum+=TreeArr[pos];
pos-=lowbit(pos);
}
return sum;
}

void DFS(int v)
{
Node * p;
int be=++depth;
vst[v]=true;
for(p=graph[v];p;p=p->next)
{
if(! vst[p->vertex])
DFS(p->vertex);
}
DFSBegin[v]=be;
DFSEnd[v]=depth;
}

void AddEdge(int a,int b)
{
Node * p1=new Node(b,graph[a]);
Node * p2=new Node(a,graph[b]);
graph[a]=p1;
graph[b]=p2;
}

int main()
{
int i,a,b,t;
char ch;
cin>>n;
memset(graph,0,sizeof(Node *)*n);
depth=0;
for(i=0;i<n-1;i++)
{
  scanf("%d%d",&a,&b);
  AddEdge(a,b);
}
DFS(1);
for(i=1;i<=n;i++)
  Modify(i,1);
cin>>t;
while(t--)
{
  cin>>ch;
  scanf("%d",&a);
  if(ch=='C')
  {
   if(! apple[a])
    Modify(DFSBegin[a],-1);
   else
    Modify(DFSBegin[a],1);
   apple[a]=! apple[a];
  }
  else
   printf("%d/n",Query(DFSEnd[a])-Query(DFSBegin[a]-1));
}
return 0;
}