luogu1351 [NOIp2014]联合权值 (dfs)

有两种情况：一个点到它的父亲的父亲（要算两次）、一个点的子节点之间互相到达

#include<bits/stdc++.h>
#define pa pair<int,int>
#define CLR(a,x) memset(a,x,sizeof(a))
using namespace std;
typedef long long ll;
const int maxn=2e5+10,mod=10007;

inline ll rd(){
    ll x=0;char c=getchar();int neg=1;
    while(c<'0'||c>'9'){if(c=='-') neg=-1;c=getchar();}
    while(c>='0'&&c<='9') x=x*10+c-'0',c=getchar();
    return x*neg;
}

int N,v[maxn];
int eg[maxn*2][2],egh[maxn],ect;
int fa[maxn],mans,ans;

inline void adeg(int a,int b){
    eg[++ect][0]=b;eg[ect][1]=egh[a];egh[a]=ect;
}

void dfs(int x){
    int mm=0;
    if(fa[fa[x]]){
        mans=max(mans,v[x]*v[fa[fa[x]]]);
        ans=(ans+v[x]*v[fa[fa[x]]]*2)%mod;
    }
    int sum=0;
    for(int i=egh[x];i;i=eg[i][1]){
        int b=eg[i][0];if(b==fa[x]) continue;
        sum=(sum+v[b])%mod;
        mans=max(mans,mm*v[b]);
        mm=max(mm,v[b]);
        fa[b]=x;dfs(b);
    }
    for(int i=egh[x];i;i=eg[i][1]){
        int b=eg[i][0];if(b==fa[x]) continue;
        ans=(ans+v[b]*(sum-v[b]))%mod;
    }
}

int main(){
    //freopen(".in","r",stdin);
    int i,j,k;
    N=rd();
    for(i=1;i<N;i++){
        int a=rd(),b=rd();
        adeg(a,b);adeg(b,a);
    }
    for(i=1;i<=N;i++) v[i]=rd();
    dfs(1);
    printf("%d %d
",mans,(ans%mod+mod)%mod);
    return 0;
}