无向连通图 G 有 n 个点，n−1 条边。点从 1 到 n 依次编号, 编号为 i 的点的权值为 Wi，每条边的长度均为 1。图上两点 (u,v) 的距离定义为 u 点到 v 点的最短距离。对于图 G 上的点对 (u,v)，若它们的距离为 2，则它们之间会产生Wv×Wu 的联合权值。
请问图 G 上所有可产生联合权值的有序点对中，联合权值最大的是多少？所有联合权值之和是多少？
第一行包含 1 个整数 n。
接下来 n−1 行, 每行包含 2 个用空格隔开的正整数 u,v，表示编号为 u 和编号为 v 的点之间有边相连。
最后 1 行, 包含 n 个正整数，每两个正整数之间用一个空格隔开，其中第 i 个整数表示图 G 上编号为 i 的点的权值为 Wi。
输出共 1 行, 包含 2 个整数，之间用一个空格隔开, 依次为图 G 上联合权值的最大值和所有联合权值之和。由于所有联合权值之和可能很大，输出它时要对10007取余。
1
2
3
4
5
6
| 5
1 2
2 3
3 4
4 5
1 5 2 3 10
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
| #include "ybwhead/ios.h"
#define maxn (int)2e5 + 100
#include "ybwhead/edge.h"
int n, m;
Edge e;
int ww[maxn];
long long ans, tot;
const int mod = 10007;
int main()
{
yin >> n;
for (int i = 1; i < n; i++)
{
int x, y;
yin >> x >> y;
e.add(x, y);
// cout << i + 1 << " " << x << " " << y << endl;
}
for (int i = 1; i <= n; i++)
yin >> ww[i];
// puts("!!!");
for (int i = 1; i <= n; i++)
{
long long sum = 0, ma = 0;
for (int j = e.head[i]; j; j = e.e[j].nxt)
{
int v = e.e[j].v;
long long w = ww[v];
ans += sum * w % mod;
ans %= mod;
tot = max(tot, ma * w);
sum += w;
sum %= mod;
ma = max(ma, w);
}
}
cout << tot << " " << ans * 2 % mod << endl;
return 0;
}
|
v1.4.14