树上路径

[编程题]树上路径

时间限制：C/C++ 2秒，其他语言4秒空间限制：C/C++ 256M，其他语言512M
算法知识视频讲解

给出一个n个点的树，1号节点为根节点，每个点有一个权值
你需要支持以下操作
1.将以u为根的子树内节点(包括u)的权值加val

2.将(u, v)路径上的节点权值加val

3.询问(u, v)路径上节点的权值两两相乘的和

输入描述:

第一行两个整数n, m，表示树的节点个数以及操作个数

接下来一行n个数，表示每个节点的权值

接下来n - 1行，每行两个整数(u, v)，表示(u, v)之间有边

接下来m行
开始有一个数opt，表示操作类型
若opt = 1，接下来两个整数表示u, val
若opt = 2，接下来三个整数表示(u, v), val
若opt = 3，接下来两个整数表示(u, v)
含义均如题所示

输出描述:

对于每个第三种操作，输出一个数表示答案，对取模

示例1

输入

输出

说明

第一组询问结果：3 * 5 = 15
第二组询问结果：1 * 5 = 5
第三组询问结果：3 * 5 + 1 * 5 + 3 * 1 = 23

备注:

对于的数据，
对于的数据，
设表示读入的第i个节点的权值以及每次修改的权值，保证
保证不会有负数

算法知识视频讲解

耕云种月

发表于 2022-01-30 19:49:45

原题解链接： ∑1⩽i,j⩽Nai∗aj=(∑ai)2−∑ai2\sum_{1 \leqslant i, j \leqslant N} a_{i} * a_{j}=\left(\sum a_{i}\right)^{2}-\sum a_{i}^{2}∑1⩽i,j⩽Nai∗aj=(∑ai)2−∑ 展开全文

Karashi

发表于 2022-02-24 15:53:05

题3 - 树上路径题目支持3种操作 1.将以u为根的子树内节点(包括u)的权值加val 2.将(u, v)路径上的节点权值加val 3.询问(u, v)路径上节点的权值两两相乘的和思路：很明显，唯一有难度的就是操作3。我们换位思考，要计算一个数组内元素两两相乘之和，其实就等于（元素之和的平方- 展开全文