1782: [Usaco2010 Feb]slowdown 慢慢游_usaco slowdown-优快云博客

本文链接：https://blog.youkuaiyun.com/Mmh2000/article/details/72954539

本文介绍了一种解决特定路径计数问题的算法实现，通过构建树状结构并使用深度优先搜索（DFS）来统计每头牛经过其他牛的数量。利用线性结构与区间更新技巧优化查询效率。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目大意：有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;
}