洛谷 P1351 联合权值

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);
}