P1351 联合权值
【问题描述】
无向连通图 G 有 n 个点,n-1 条边。点从 1 到 n 依次编号,编号为 i 的点的权值为 Wi, 每条边的长度均为 1。图上两点(u, v)的距离定义为 u 点到 v 点的最短距离。对于图 G 上的点对(u, v),若它们的距离为 2,则它们之间会产生Wu×Wv的联合权值。请问图 G 上所有可产生联合权值的有序点对中,联合权值最大的是多少?所有联合权值之和是多少?
【样例输入】
5
1 2
2 3
3 4
4 5
1 5 2 3 10
【样例输出】
20 74
分析:不难发现这是一棵树,求出每个点所有儿子的最值以及sum,然后枚举每个儿子计算即可。
代码
#include <cstdio>
#include <algorithm>
#define N 500005
#define mo 10007
using namespace std;
struct arr
{
int to,nxt;
}a[N];
int n,val[N],mx[N],sum[N],pos[N];
int ls[N],l;
int read()
{
int x = 0, f = 1;
char ch = getchar();
while (ch < '0' || ch > '9') {if (ch == '-') f = -1; ch = getchar();}
while (ch >= '0' && ch <= '9') x = x * 10 + ch - '0', ch = getchar();
return x * f;
}
void add(int x, int y)
{
a[++l].to = y;
a[l].nxt = ls[x];
ls[x] = l;
}
int main()
{
scanf("%d", &n);
for (int i = 1; i < n; i++)
{
int x, y;
x = read(), y = read();
add(x, y);
add(y, x);
}
for (int i = 1; i <= n; i++) val[i] = read();
for (int i = 1; i <= n; i++)
for (int j = ls[i]; j; j = a[j].nxt)
{
if (val[a[j].to] > mx[i])
{
mx[i] = val[a[j].to];
pos[i] = a[j].to;
}
sum[i] += val[a[j].to];
}
int ans1 = 0, ans2 = 0;
for (int i = 1; i <= n; i++)
for (int j = ls[i]; j; j = a[j].nxt)
{
if (pos[i] != a[j].to) ans1 = max(ans1, mx[i] * val[a[j].to]);
ans2 = (ans2 + ((sum[i] - val[a[j].to])) % mo * val[a[j].to] % mo) % mo;
}
printf("%d %d", ans1, ans2);
}