LOJ2499 联合权值(树，思路)

题目描述

无向连通图$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$ 取余。

样例

样例输入

5
1 2
2 3
3 4
4 5
1 5 2 3 10

样例输出

20 74

思路

枚举每一个点，取其任意两个点，然后进行组合，然后两两相乘，得到最大值与他们的和。将所有值统计一下，然后注意因为和是组合出来的，所以再乘2，输出就可以了

代码

#include <bits/stdc++.h>
using namespace std;
#define mem(a, b) memset(a, b, sizeof(a))
#define mod 10007
const int N = 200000 + 10;
int first[N], tot = 0;
struct edge
{
    int v, next;
} e[N * 2];
int a[N];
void add_edge(int u, int v)
{
    e[tot].v = v;
    e[tot].next = first[u];
    first[u] = tot++;
}
void init()
{
    mem(first, -1);
    tot = 0;
}
int max_ans = 0, ans;
void solve(int u)
{
    int maxx = 0, sum = 0, m = 0;
    for (int i = first[u]; ~i; i = e[i].next)
    {
        int v = e[i].v;
        if (a[v] > maxx)
        {
            m = maxx;
            maxx = a[v];
        }
        else if (a[v] > m)
        {
            m = a[v];
        }
        ans = (ans + sum * a[v]) % mod;
        sum = (sum + a[v]) % mod;
    }
    max_ans = max(max_ans, maxx * m);
}
int main()
{
    int n, u, v;
    scanf("%d", &n);
    init();
    for (int i = 1; i < n; i++)
    {
        scanf("%d%d", &u, &v);
        add_edge(u, v);
        add_edge(v, u);
    }
    for (int i = 1; i <= n; i++)
        scanf("%d", &a[i]);
    for (int i = 1; i <= n; i++)
        solve(i);
    printf("%d %d\n", max_ans, ans * 2 % mod);
    return 0;
}