[luoguP1351] 联合权值（Dfs）

传送门

距离为2的点会产生权值，第一问，只需要在dfs的时候把一个点相邻的点都处理出来就行。

具体处理方式看代码，然而这样只处理了一遍，最后在乘2就好了。

第二问只需要处理一个点相邻的点中最大的和次大的就行。

——代码

#include <cstdio>
#include <cstring>
#include <iostream>
#define LL long long

const int MAXN = 200001, p = 10007;
int n, cnt;
int head[MAXN], to[MAXN << 1], next[MAXN << 1];
LL max, tot, a[MAXN], sum[MAXN], ans[MAXN], max1[MAXN], max2[MAXN];
bool vis[MAXN];

inline int read()
{
    int x = 0, f = 1;
    char ch = getchar();
    for(; !isdigit(ch); ch = getchar()) if(ch == '-') f = -1;
    for(; isdigit(ch); ch = getchar()) x = (x << 1) + (x << 3) + ch - '0';
    return x * f;
}

inline LL Max(LL x, LL y)
{
    return x > y ? x : y;
}

inline void add(int x, int y)
{
    to[cnt] = y;
    next[cnt] = head[x];
    head[x] = cnt++;
}

inline void dfs(int u)
{
    int i, v;
    vis[u] = 1;
    for(i = head[u]; i ^ -1; i = next[i])
    {
        v = to[i];
        ans[u] = (ans[u] + sum[u] * a[v]) % p;
        sum[u] = (sum[u] + a[v]) % p;
        if(a[v] > max1[u]) max2[u] = max1[u], max1[u] = a[v];
        else if(a[v] > max2[u]) max2[u] = a[v];
        if(!vis[v]) dfs(v);
    }
}

int main()
{
    int i, x, y;
    n = read();
    memset(head, -1, sizeof(head));
    for(i = 1; i < n; i++)
    {
        x = read();
        y = read();
        add(x, y);
        add(y, x);
    }
    for(i = 1; i <= n; i++) a[i] = read();
    dfs(1);
    for(i = 1; i <= n; i++)
    {
        tot = (tot + ans[i]) % p;
        max = Max(max, max1[i] * max2[i]);
    }
    printf("%lld %lld
", max, (tot << 1) % p);
    return 0;
}