[ContestHunter Round#55]LCA的统计

题目地址：



http://ch.ezoj.tk/contest/CH%20Round%20%2355%20-%20Streaming%20%236%20(NOIP%E6%A8%A1%E6%8B%9F%E8%B5%9Bday2)/LCA%E7%9A%84%E7%BB%9F%E8%AE%A1

描述

萌蛋有一棵n个节点的有根树，其根节点为1。除此之外，节点i的父节点为p_i。每个点上都有一个权值，节点i的权值是w_i。
萌蛋知道你一定知道什么叫做祖先（从根到某个点的路径上的每个点都是这个点的祖先，包括它本身），也一定知道什么叫做最近公共祖先（两个点的最近公共祖先是某个点，这个点同时是两个点的祖先，且离根最远）。
现在给出这棵树，你需要求出：
∑_(i=1)^n▒∑_(j=1)^n▒〖w_i w_j w_LCA(i,j)  〗
其中LCA(i,j)表示点i与点j的最近公共祖先。
由于答案可能很大，你只需要输出它对1,000,000,007取模的结果。

 

输入格式

第一行为两个整数n w_1。
第二行到第n行，第i行有两个整数p_i  w_i。

 

输出格式

输出只有一行，为一个整数，表示所求答案对1,000,000,007取模的结果。

样例输入

2 2
1 1

样例输出

17

数据范围与约定

对于30%的数据，n≤100,w_i≤10。
对于60%的数据，n≤1,000,w_i≤1,000。
对于100%的数据，1≤n≤100,000,0≤w_i≤1,000,000,000,1≤p_i<i。

样例解释

1×1×1+1×2×2+2×1×2+2×2×2=17。

解题报告：http://pan.baidu.com/s/1o6jwbUy

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <algorithm>

#define MAXE 100010
#define MAXV 100010
#define MOD 1000000007

typedef long long int LL;

using namespace std;

struct edge
{
    int v,next;
}edges[MAXE];

LL head[MAXV],nCount=0,ans=0;
LL w[MAXV],sum[MAXV];

void AddEdge(int U,int V)
{
    edges[++nCount].v=V;
    edges[nCount].next=head[U];
    head[U]=nCount;
}

void DFS(int x) //从点x开始树形DP
{
    for(int p=head[x];p!=-1;p=edges[p].next) //先DFS求出这个店的儿子的sum值
    {
        int v=edges[p].v;
        DFS(v);
        sum[x]=(sum[x]+sum[v])%MOD;
    }
    //此时A=sum[x]
    LL tmp=(sum[x]*2LL+w[x])%MOD*w[x]%MOD;
    for(int p=head[x];p!=-1;p=edges[p].next)
    {
        int son=edges[p].v;
        tmp=(tmp+(sum[x]+MOD-sum[son])%MOD*sum[son])%MOD; //case 2:LCA(i,j)的i和j分布在LCA(i,j)的不同儿子子树下
    }
    ans=(ans+tmp*w[x])%MOD;
    sum[x]=(sum[x]+w[x])%MOD; //这个时候sum[x]把x的权值算进去
}

int main()
{
    int n;
    memset(head,-1,sizeof(head));
    scanf("%d%lld",&n,&w[1]);
    for(int i=2;i<=n;i++)
    {
        int p;
        scanf("%d%lld",&p,&w[i]);
        w[i]%=MOD;
        AddEdge(p,i);
    }
    DFS(1);
    printf("%lld\n",ans%MOD);
    return 0;
}