[省选] [线段树] [DFS序] [矩阵快速幂] [HLOI2015] Tree

Description 题目描述

Chavo有一颗水晶树，树上会结出魔法水晶。包括树根在内，树一共有$n$个分叉的节点，其中$1$号节点为树根，初始状态下每一个节点都有一个生长值$a_i$。Chavo有一根法杖，法杖可以对某个节点$u$释放生长法术，使得以$u$为根的子树的每个节点的生长值更新为$f(k)$。对于每一个被更新的节点来说，$f(x)$的定义都不一样，其中$f(1)$是在释放法术之前被更新节点的生长值，$f(2)$是在释放法术之前节点$v$的生长值。但是对于所有被更新的节点来说，$f(x)$的递推公式都一样，是$f(x)=f(x-1)+f(x-2)+k$。Chavo想要释放$m$次生长法术，每次法术都给定$u，v，k$，她想问问你释放法术之后所有节点的生长值的和是多少？结果要对$1000000007$取模后输出。

Input 输入

第一行输入整数$n$和$m$
第二行输入$n$个整数，第$i$个整数代表第$i$个节点上初始值$a_i$
接下来$n-1$行，每一行有两个正整数$i,j$，代表树上第$i$个节点和第$j$个节点有树杈相连。
接下来$m$行，每一行有$3$个正整数$u,v,k$，如题目描述

Output 输出

输出一个整数代表经过所有操作后整棵树上所有节点数的和，结果要对$1000000007$取模后输出。

Sample Input 样例输入

3 2
3 4 5
1 2
1 3
1 2 2
2 3 3

Sample Output 样例输出

19

Limits 限制

对于$30\%$的数据，$1\le n,m,k\le 100,1\le u,v,i,j\le n,0\le a_i\le 100$
对于$60\%$的数据，$1\le n,m\le 10^3,1\le k\le 10^9,1\le u,v,i,j\le n,0\le a_i\le 10^9$
对于$100\%$的数据，$1≤n,m≤10^5,1≤k≤109,1≤u,v,i,j≤n,0≤a_i≤10^9 $
Time Limit : $2s$ & Memory Limit : $128MB$

Hints 提示

第一次操作时，更新$1、2、3$号点，对于$1$号点，$f(1)=3,f(2)=4$,更新后$1$号点值为$f(2)=4$，同理$2、3$号点值为$4$
第二次操作时，更新$2$号点，$f(1)=4,f(2)=4,f(3)=11$,更新后$2$号点为$11$
最后求和为$4+11+4=19$

注意，$f(1)$是对于子树内每个点说的……每个点的$f(1)$不一定相同，不过都是该点的权值。
树上实现子树修改和查询，搞出来个DFS序就变成了区间问题，线段树即可解决。
但是这什么鬼畜修改。类似于斐波那契数列的递推关系，可以考虑构造矩阵，做矩阵快速幂。
先看看$f(x)$的一些值有什么规律：

$$\begin{array}{rl}f(1)& =u\\ f(2)& =v\\ f(3)& =u+v+k\\ f(4)& =u+2v+k\\ f(5)& =2u+3v+4k\\ f(6)& =3u+5v+7k\\ f(7)& =5u+8v+12k\\ & \dots \end{array}$$

如果我们定义一个类斐波那契数列$F_i$，$F_{-1}=1,F_0=0,F_1=1$，$F_i=F_{i-1}+F_{i-2}(i>1)$，$f(T)$就可以这样描述了：
$$f(T)=F_{T-2}u+F_{T-1}v+(F_T-1)k$$
怎么构造矩阵呢？
可以分为两种矩阵，线段树里的矩阵和处理影响的矩阵。
线段树里的矩阵，叶子节点的$S$这样构造：
$$S=\left[\begin{array}{ccc}u& 0& 1\\ 0& 0& 0\\ 0& 0& 0\end{array}\right]$$
其中，$u$为此时这个节点的权值。
非叶子节点的$S$为单位矩阵。
处理影响的矩阵$T$是两个矩阵的乘积：
$$T=\left[\begin{array}{ccc}1& 0& 0\\ 0& 0& 0\\ 0& v& 1\end{array}\right]\times {\left[\begin{array}{ccc}0& 1& 0\\ 1& 1& 0\\ 0& k& 1\end{array}\right]}^{k-1}$$
其中，$v$是节点$v$的生长值，$k$意义与题中相同。
那么这么乘能乘出来些什么？
为了方便，令：
$$A=\left[\begin{array}{ccc}0& 1& 1\\ 1& 1& 0\\ 0& k& 1\end{array}\right],B=\left[\begin{array}{ccc}1& 0& 0\\ 0& 0& 0\\ 0& v& 1\end{array}\right]$$
发现$A^c$这个矩阵是这样的：
$$A^c=\left[\begin{array}{ccc}{F}_{c-1}& F_c& 0\\ F_c& F_{c+1}& 0\\ (F_{c+1}-1)k& (F_{c+2}-1)k& 1\end{array}\right]$$
那么，$A^{k-1}$就是这样的：
$$A^{k-1}=\left[\begin{array}{ccc}{F}_{k-2}& F_{k-1}& 0\\ F_{k-1}& F_k& 0\\ (F_k-1)k& (F_{k+1}-1)k& 1\end{array}\right]$$
$B\times A^{k-1}$就是这样的：
$$B\times A^{k-1}=\left[\begin{array}{ccc}{F}_{k-2}& F_{k-1}& 0\\ 0& 0& 0\\ v{F}_{k-1}+(F_k-1)k& v{F}_k+(F_{k+1}-1)k& 1\end{array}\right]$$
更新一个叶子节点后，叶节点的矩阵就是这样的：
$$S\times B\times A^{k-1}=\left[\begin{array}{ccc}u{F}_{k-2}+v{F}_{k-1}+(F_k-1)k& u{F}_{k-1}+v{F}_k+(F_{k+1}-1)k& 1\\ 0& 0& 0\\ 0& 0& 0\end{array}\right]$$
$u$的位置顺利被$f(k)$替代，构造成功。
那么，如何更新答案？
对于线段树矩阵，更新一个区间时$S$矩阵乘$T$
每次查询节点$v$的生长值时，将这一区间矩阵下传，之后清为单位矩阵。
最后，将线段树上非叶子节点全部下传清为单位矩阵后，统计答案。
即将$S$矩阵中的$u$加起来。
Code
总时间复杂度为$O(m({\log }_{2}n+{\log }_{2}k))$