送你一颗圣诞树

题意

有$m$棵树$T_0$~$T_m$，$T_0$只有一个$0$号节点。对于$T_i(i\in[1,m])$，给出$a_i,b_i,c_i,d_i,l_i$，表示这棵树是由$T_{a_i}$的$c_i$号节点与$T_{b_i}$的$d_i$号节点间连一条长为$l_i$的边构成的。在$T_i$ 中，保持$T_{a_i}$ 中的所有节点编号不变，然后如果$T_{a_i}$ 中有$s$ 个节点，他会把$T_{b_i}$ 中的所有节点的编号加上$s$。
对于$T_i(i\in[1,m])$，求$\sum\limits_{i= 0}^{n-1}\sum\limits_{j = i + 1}^{n-1}d[i][j]$，其中$n$为$T_i$节点个数，$d[i][j]$为该树中$i,j$的最短距离,输出$mod$ $10^9 + 7$。

$0\le a_i,b_i < i,0\le l_i\le10^9$
$1\le m\le60$
数据组数$T\le 100$

$Time$ $Limits:2000ms$
$Memory$ $Limits:512M$

分析

令$ans[i]$为$T_i$的答案，$size[i]$为$T_i$的节点个数。
易得$ans[i] = ans[a_i] + ans[b_i]+size[a_i] * size[b_i] * l_i+top_{a_i,c_i} * size[b_i]+top_{b_i,d_i}*size[a_i]$
$top_{k,x}$表示$T_k$中所有节点到$x$号节点的距离和。
现在我们看看如何求$top_{k,x}$。
考虑把$T_k$分成$T_{a_k},T_{b_k}$。
1° $k=0$时，值为$0$;
2° $x < size[a_k]$时，$x$原来是在$T_{a_k}$中，那么显然$top_{k,x} = top_{a_k,x}+(top_{b_k,d_k}+(l_k+dis_{a_k,c_k,x})*size[b_k])$
其中$dis_{k,x,y}$表示$T_k$中$x,y$的距离。
3°$x >= size[a_k]$时，$x$原来是在$T_{b_k}$中，仿照2°，我们可以得出一个相似的方程。
这样就可以递归求解$top_{k,x}$了。
我们的问题又变成了如何求$dis_{k,x,y}$。
同样考虑递归求解。
如果$x,y$同属于组成$T_k$的两棵树中的一棵，我们可以递归下去；否则令$x<y$，$dis_{k,x,y} = dis_{a_k,x,c_k}+dis_{b_k,y,d_k}+l_k$，这样变成了两个子问题，继续递归求解。边界为$k=0或x=y$ 时，$dis_{k,x,y}=0$。

上面$top,dis$的求值过程可以用记忆化搜索。

求$dis_{k,x,y}$的过程最多递归$m$层，时间复杂度为$O(m)$;
求$top_{k,x,y}$的过程最多有$m$层，每层一个求$dis$的过程，时间复杂度为$O(m^2)$
所以总时间复杂度是$O(m^3)$

代码

#include <cstdio>
#include <algorithm>
#include <cstring>
#include <map>
using namespace std;

typedef long long LL;
typedef pair<LL,LL> pa;
const int N = 66;
const LL P = 1e9 + 7;
int n,a[N],b[N],len[N];
LL size[N],ans[N],c[N],d[N];
map< pa , int > dis[N];

void init() {
    scanf("%d",&n);
    for (int i = 1;i <= n;i ++) {
        scanf("%d%d%lld%lld%d",&a[i],&b[i],&c[i],&d[i],&len[i]);
        dis[i].clear();
    }
}

int get(int k,LL x,LL y) {
    if (k == 0 || x == y) return 0;
    if (y < x) swap(x,y);
    if (y < size[a[k]]) return get(a[k],x,y);
    if (x >= size[a[k]]) return get(b[k],x - size[a[k]],y - size[a[k]]);
    pa cur = make_pair(x,y);
    if (dis[k].find(cur) != dis[k].end()) return dis[k][cur];
    int re = (0LL + get(a[k],x,c[k]) + get(b[k],y - size[a[k]],d[k]) + len[k]) % P;
    return dis[k][cur] = re;
}

int top(int k,LL x) {
    if (k == 0) return 0;
    pa cur = make_pair(x,-1);
    if (dis[k].find(cur) != dis[k].end()) return dis[k][cur];
    int re = 0;
    if (x < size[a[k]]) {
        re = top(a[k],x);
        re = (re + top(b[k],d[k])) % P;
        re = (re + 1LL * (get(a[k],x,c[k]) + len[k]) % P * (size[b[k]] % P) % P) % P;
    }       
    else {
        re = top(b[k],x - size[a[k]]);
        re = (re + top(a[k],c[k])) % P;
        re = (re + 1LL * size[a[k]] % P * (get(b[k],x - size[a[k]],d[k]) + len[k]) % P) % P;
    }
    return dis[k][cur] = re;
}

void solve() {
    for (int i = 1;i <= n;i ++) {
        int l = a[i],r = b[i];
        ans[i] = (ans[l] + ans[r]) % P;
        size[i] = size[l] + size[r];
        ans[i] = (ans[i] + size[l] % P * (size[r] % P) % P * LL(len[i]) % P) % P;
        ans[i] = (ans[i] + 1LL * size[l] % P * top(r,d[i]) % P) % P;
        ans[i] = (ans[i] + 1LL * size[r] % P * top(l,c[i]) % P) % P;
        printf("%lld\n",ans[i]);
    }
}

int main() {
    int T;
    scanf("%d",&T);
    size[0] = 1;
    while (T --) {
        init();
        solve();
    }
}