P1351 联合权值

题面:https://www.luogu.org/problem/P1351

本题可以直接枚举每个点所相连的点,因为子-->父-->子距离刚好是2,所以最大权值就由儿子中的权值最大值和权值次大值的乘积来提供,而和可以直接统计,最后乘2即可,因为联合权值的产生是成对的.

Code:
#include<cstdio>
#include<iostream>
using namespace std;
const int N=200005,mod=10007;
int n,ans,maxans,cnt,head[N],w[N];
struct Node{
	int v,nxt;
}edge[N*2];
void add(int u,int v){
	++cnt;
    edge[cnt].v=v;
    edge[cnt].nxt=head[u];
    head[u]=cnt;
}
void dfs(int u){
    int sum=0,maxf=0,maxs=0;
    for(int i=head[u];i;i=edge[i].nxt){
		int v=edge[i].v;
        if(w[v]>maxf){
			maxs=maxf;
			maxf=w[v];
		}
		else if(w[v]>maxs){
			maxs=w[v];
		}
        ans=(ans+sum*w[v])%mod;
        sum=(sum+w[v])%mod;
    }
    maxans=max(maxans,maxf*maxs);
}
int main(){
	int u,v;
    scanf("%d",&n);
    for(int i=1;i<n;i++){
        scanf("%d%d",&u,&v);
        add(u,v);
		add(v,u);
    }
    for(int i=1;i<=n;i++){
		scanf("%d",&w[i]);
	}
    for(int i=1;i<=n;i++){
		dfs(i);
	}
    printf("%d %d\n",maxans,(ans*2)%mod);
}