链式前向星+b/dfs

题目链接:Wannfly挑战赛4-D

题目大意:

题目描述

wyf非常喜欢树。一棵有根数树上有N个节点,1号点是他的根,每条边都有一个距离,而wyf是个爱问奇怪问题的熊孩子,他想知道对于某个点x,以x为根的子树上,所有与x距离大于等于k的点与x的距离之和。

输入描述:

 
   

第一行一个正整数N

接下来N-1描述这棵树,每行两个数第i行两个数p和D表示树上有一条p到i+1长度为D的边。(p<=i)

下面一行一个正整数Q表示wyf的询问次数。

接下来Q行每行两个正整数x和k。 (1<=N,Q<=2x105,1<=D,K<=106)

输出描述:

对于每次询问x,k输出以x为根的子树上,所有与x距离大于等于k的点与x的距离之和。(若不存在这样的点,则输出应为0)
示例1

输入

3
1 2
1 3
2
1 3
1 2

输出

3
5
题目思路:

用链接前向星存储图,然后开几个数组存一下,然后搜索dfs和bfs就可以了,

#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#define ri(n) scanf("%d",&n)
#define oi(n) printf("%d\n",n)
#define rl(n) scanf("%lld",&n)
#define ol(n) printf("%lld\n",n)
#define rep(i,l,r) for(i=l;i<=r;i++)
#define rep1(i,l,r) for(i=l;ip;
    p.push(x);
    ll ans=0;
    while(!p.empty())
    {
        int u=p.front();
        p.pop();
        for(int i=head[u];~i;i=e[i].next)
        {
            int v=e[i].to;
            if(v==fa[u])
                continue;//可有可无,有可以减少时间复杂度
            if(s[v]-s[x]>=k)//通过点v和x到点1之间的距离差确定两点之间距离
            {
                ans+=sum[v]+(s[v]-s[x])*sz[v];
            }
            else
                p.push(v);
        }
    }
    return ans;
}
int main()
{
    int n;
    while(scanf("%d",&n)==1)
    {
        init();
        for(int i=2; i<=n; i++)
        {
            int p,w;
            scanf("%d%d",&p,&w);
            add(p,i,w);
        }
        fa[1]=-1;
        s[1]=0;
        dfs(1);
        //for(int i=1;i<=n;i++)
            //cout<<"fa["<


你可能感兴趣的:(dfs)