链式前向星+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 <iostream>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <cstring>
#include <string>
#include <algorithm>
#include <set>
#include <map>
#include <stack>
#include <vector>
#include <queue>
#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;i<r;i++)
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const int epg=10-8;
const int maxn=2*1e5+10;
int head[maxn];
int cnt;
struct node
{
    int w;
    int to;
    int next;
} e[maxn*2];

ll fa[maxn];//记录结点u的上一个结点
ll sum[maxn];//记录结点u到其他下面能到达的所有结点的总路程
ll s[maxn];//记录点1到所有点的路程长度
ll sz[maxn];//记录点u到能到达其他结点的个数，包括自己
void init()
{
    memset(head,-1,sizeof(head));
    memset(fa,0,sizeof(fa));
    memset(sum,0,sizeof(sum));
    memset(s,0,sizeof(s));
    memset(sz,0,sizeof(sz));
    cnt=0;
}
void add(int u,int v,int w)
{
    e[cnt].to=v;
    e[cnt].next=head[u];
    e[cnt].w=w;
    head[u]=cnt++;
}
void dfs(int u)
{
    sz[u]=1;
    for(int i=head[u];~i;i=e[i].next)
    {
        int v=e[i].to;
        if(fa[v])
            continue;//可有可无，有可以减少时间复杂度
        fa[v]=u;
        s[v]=s[u]+e[i].w;
        dfs(v);
        sz[u]+=sz[v];
        //sum[u]+=sum[v]+(s[v]-s[u])*sz[v];//效果和下面一句相同
        sum[u]+=sum[v]+e[i].w*sz[v];
    }
}
ll bfs(int x,ll k)
{
    queue<int>p;
    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["<<i<<"]="<<fa[i]<<"  s["<<i<<"]="<<s[i]<<"  sz["<<i<<"]="<<sz[i]<<"  sum["<<i<<"]="<<sum[i]<<endl;
        int m;
        scanf("%d",&m);
        while(m--)
        {
            int x;
            ll k;
            scanf("%d%lld",&x,&k);
            printf("%lld\n",bfs(x,k));
        }
    }
    return 0;
}

/*
7
1 3
1 2
3 1
4 3
2 4
1 6
*/