题意简述
题解
首先,每次修改完点权后,重新考虑一遍所有路径显然是不现实的,所以我们考虑求出经过每个点的两端同色的简单路径数,这样权值和容易统计和修改。
接下来分析仙人掌上的简单路径性质。一条简单路径上的边,可以分为桥边和环上的边。过桥只有一种方法,而过同一个环有两个方向的环边可选。
现在,我们选取一个关键点,强制简单路径必须通过该点。如果关键点在一个环上,且该简单路径在该环上的第一个和最后一个点均不是关键点,那么该环只有一个方向的环边可选。
建立原仙人掌的圆方树。
如果定义方点贡献 2 种方案,圆点贡献 1 种方案,一条路径贡献方案数为所有点贡献方案数之积,那么原仙人掌上两个同色点间的路径数转化为对应圆点间的简单路径方案数。
在所有的树上简单路径中,
如果它通过关键点,则贡献值为该路径的方案数;
如果它不通过关键点,但是通过关键点的某相邻方点,则贡献该路径方案数的一半。
考虑用何算法求出经过每个点的两端同色的简单路径数(记过 u u u的简单路径数为 c o e [ u ] coe[u] coe[u]):
法一:暴力枚举
按题意枚举所有同色点对,可解决 n ≤ 100 n≤100 n≤100的子任务。
法二:树形DP
首先枚举每一种颜色,每次只考虑当前颜色的同色点对的贡献。
对于树上任意一点 u u u,我们可以求出:
d p [ u ] = dp[u]= dp[u]= u u u的子树内当前色的圆点到 u u u的路径数
u p [ u ] = up[u]= up[u]= u u u的子树外当前色的圆点到 u u u的路径数
(换根DP)
那么过 u u u的简单路径数 c o e [ u ] + = ( d p [ u ] + u p [ u ] + 1 ) 2 − d p [ u ] 2 − d p [ v ] 2 − 1 2 2 coe[u]+=\frac{(dp[u]+up[u]+1)^2-dp[u]^2-dp[v]^2-1^2}{2} coe[u]+=2(dp[u]+up[u]+1)2−dp[u]2−dp[v]2−12
这样可以解决 n ≤ 1 0 5 , k ≤ 5 n\leq10^5,k\leq5 n≤105,k≤5的子任务
优化:建虚树
处理树上询问点两两间路径的并的问题,一个常见的套路是构建询问点的虚树,因为虚树中同一条边上的点受询问点的影响一致,把这些点一起处理可大大提高效率
于是,我们构建同色点的虚树,通过前缀和,预处理虚树每条边上压缩的点贡献的方案数,作为边的方案数。
对于过虚树节点的简单路径数,可以在树形dp时直接统计出来;
对于过虚树边上压缩掉的点的简单路径数,可以利用树上差分与前缀和快速处理。
那么接下来仅需考虑 过关键点相邻的方点的路径总方案数 即可:
计算通过了每个相邻方点的路径总方案数,再减去通过了它们之间相邻边的方案数。
不难发现,上述树上差分与前缀和的方法,已经预处理了通过各边的路径方案数。
此算法时间复杂度 O ( n l o g n ) O(n log n) O(nlogn)、空间复杂度 O ( n ) O(n) O(n)
#include<bits/stdc++.h>
using namespace std;
const int mod=1e9+7;
const int inv2=(mod+1)/2;
namespace modular{
int add(int a,