bzoj 1103: [POI2007]大都市meg【dfs序+树状数组】

很明显的暗示,就是在树的dfs序上维护树状数组,加减的时候差分即可

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=500005;
int n,m,h[N],cnt,s[N],top,tot,t[N],fa[N],l[N],r[N];
struct qwe
{
int to,ne;
}e[N];
int read()
{
int r=0,f=1;
char p=getchar();
while(p>'9'||p<'0')
{
if(p=='-')
f=-1;
p=getchar();
}
while(p>='0'&&p<='9')
{
r=r*10+p-48;
p=getchar();
}
return r*f;
}
void add(int u,int v)
{
cnt++;
e[cnt].ne=h[u];
e[cnt].to=v;
h[u]=cnt;
}
void update(int x,int val)
{
for(int i=x;i<=n+n;i+=(i&(-i)))
t[i]+=val;
}
int ques(int x)
{
int r=-1;
for(int i=x;i;i-=(i&(-i)))
r+=t[i];
return r;
}
int main()
{
scanf("%d",&n);
for(int i=1;i<n;i++)
{
int x=read(),y=read();
add(x,y),add(y,x);
}
s[++top]=1;
while(top)
{
int u=s[top],f=fa[top--];
if(!l[u])
{
l[u]=++tot;
s[++top]=u;
for(int i=h[u];i;i=e[i].ne)
if(e[i].to!=f)
{
s[++top]=e[i].to;
fa[top]=u;
}
}
else
r[u]=++tot;
}
for(int i=1;i<=n;i++)
{
update(l[i],1);
update(r[i],-1);
}
scanf("%d",&m);
for(int i=1;i<=n+m-1;i++)
{
char ch[5];
scanf("%s",ch);
if(ch[0]=='A')
{
int x=read(),y=read();
update(l[y],-1);
update(r[y],1);
}
else
{
int x=read();
printf("%d\n",ques(l[x]));
}
}
return 0;
}
上一篇:db2备份还原


下一篇:iOS百度地图SDK集成详细步骤