题目和数据在评论中给出地址。或者自行在网上搜索,这里就不再给出。
第一眼看起来很复杂。
所以我想了个对链、和树不同情况都分别处理的算法。。orz。。复杂度太高最后4组T了。。
所以。。转换思路。。
联合权值 w[i]*w[j]。i,j距离为2.也就是i,j中间间隔了一个节点,所以我们在遍历这个中间节点的时候处理。
题目求最大和总和,
先考虑最大,我们只需要对节点U维护一个_max1,_max2最大值和次大值。
然后_max1*_max2 和总的 maxw比。maxw取大。
然后就是两两和。
先考虑U点只有3个子节点 a,b,c
则 tot = (a * b + b * c + c * a)*2 % mod
而(a + b + c)^2 = a^2 + b^2 + c^2 + 2ab + 2ac +2bc ;
2ab + 2ac +2bc = (a + b + c)^2 - (a^2 + b^2 + c^2 ).
所以..
#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <vector>
#define ma