联合权值
联合权值
https://ac.nowcoder.com/acm/problem/16495
分析
每条边的边权为 ,由于输入的图是一颗树,那么
之间有且仅有一条简单路径,所以合法的点对
中间只会有一个点。所以考虑枚举中间的节点。令
表示节点
为中间节点的权值之和,那么
。因为
,所以可以先考虑有序点对的值最后再
就好了。所以
,
表示在
为中间点,
为当前考虑点,已经计算过的权值之和。由于总的枚举数等于度数
,所以时间复杂度为
。
代码
#include<bits/stdc++.h> using namespace std; #define LL long long const int N = 200010,p = 10007; LL read() {LL x;scanf("%lld",&x);return x;} LL w[N],Max = 0,Sum = 0; vector<int> G[N]; int main() { LL n = read(); for(int i = 1;i < n;i++) { LL a = read(),b = read(); G[a].push_back(b);G[b].push_back(a); } for(int i = 1;i <= n;i++) w[i] = read(); for(int i = 1;i <= n;i++) { LL Mx = 0,sum = 0; for(auto y:G[i]) { Max = max(Max,Mx * w[y]);(Sum += sum * w[y]) %= p; sum += w[y];Mx = max(Mx,w[y]); } } printf("%lld %lld\n",Max,(Sum * 2) % p); return 0; }