题目大意:有n头牛从树上的一个点到达另一个点,求每头牛经过其他牛的个数
题解:当一头牛到达一个点x后,所有终点在以x为根的子树中的牛答案+1,区间修改,单点查询dfs序就好了。Orz hzwer写法
我的收获:劲
#include<iostream>
#include<cstdio>
#include<cstring>
#define N 100005
using namespace std;
inline int read()
{
int x=0;char ch=getchar();
while(ch<'0'||ch>'9')ch=getchar();
while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
return x;
}
int n,cnt;
int p[N],ans[N],head[N],t[N];
struct data{int to,next;}e[N+N];
void ins(int u,int v)
{e[++cnt].to=v;e[cnt].next=head[u];head[u]=cnt;}
void insert(int u,int v)
{ins(u,v);ins(v,u);}
inline int lowbit(int x){return x&(-x);}
void update(int x,int val)
{for(int i=x;i<=n;i+=lowbit(i))t[i]+=val;}
int ask(int x)
{
int sum=0;
for(int i=x;i>0;i-=lowbit(i))
sum+=t[i];
return sum;
}
void dfs(int x,int fa)
{
ans[p[x]]=ask(p[x]);
update(p[x],1);//x点对询问比它晚(p[x]值大),x及的子树(dfs搜索顺序保证)的点有贡献
for(int i=head[x];i;i=e[i].next)
if(e[i].to!=fa)
dfs(e[i].to,x);
update(p[x],-1);
}
int main()
{
n=read();
for(int i=1;i<n;i++)
insert(read(),read());
for(int i=1;i<=n;i++)
p[read()]=i;//按询问次序排列
dfs(1,0);
for(int i=1;i<=n;i++)
printf("%d\n",ans[i]);
return 0;
}