P1351-联合权值
题目:
题目描述:
无向连通图 $G$ 有 $n$ 个点，$n-1$ 条边。点从 $1$ 到 $n$ 依次编号, 编号为 $i$ 的点的权值为 $W_i$，每条边的长度均为 $1$。图上两点 $(u, v)$ 的距离定义为 $u$ 点到 $v$ 点的最短距离。对于图 $G$ 上的点对 $(u, v)$，若它们的距离为 $2$，则它们之间会产生$W_v \times W_u$ 的联合权值。
请问图 $G$ 上所有可产生联合权值的有序点对中，联合权值最大的是多少？所有联合权值之和是多少？
输入格式:
第一行包含 $1$ 个整数 $n$。
接下来 $n-1$ 行, 每行包含 $2$ 个用空格隔开的正整数 $u, v$，表示编号为 $u$ 和编号为 $v$ 的点之间有边相连。
最后 $1$ 行, 包含 $n$ 个正整数，每两个正整数之间用一个空格隔开，其中第 $i$ 个整数表示图 $G$ 上编号为 $i$ 的点的权值为 $W_i$。
输出格式:
输出共 $1$ 行, 包含 $2$ 个整数，之间用一个空格隔开, 依次为图 $G$ 上联合权值的最大值和所有联合权值之和。由于所有联合权值之和可能很大，输出它时要对$10007$取余。
样例:
样例输入 1:
1
2
3
4
5
6
| 5
1 2
2 3
3 4
4 5
1 5 2 3 10
|
样例输出 1:
思路:
实现:
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;
}
|