luogu 1351

水题

这里记录个步骤 

对某节点x,其子节点为y

求 a[y1]*(a[y2]+a[y3]+ ... +a[yn])  + a[y2]*(a[y1]+a[y3]+...a[yn]) + .... +

a[yn]*(a[y1]+a[y2]+...+a[y(n-1)] ) 

观察一下，很容易推出 answer= S^2- sum{a[i]^2}  其中S=sum{a[i] }

#include <iostream>
#include <vector>
using namespace std; 

#define int long long
 const int N=2e5+4,mod=10007;
 vector<int> g[N];
 int n,a[N];
 int mx=0,S=0;
 
 void dp(int x,int par){
 	int i,y,k1=0,k2=0;
 	for(i=0;i<g[x].size();i++){
 	    y=g[x][i]; if(par==y) continue;
 		dp(y,x);
 		
 		mx=max(mx,a[par]*a[y]); S+=2*a[par]*a[y]; S%=mod;
 		if(a[k1]<a[y]) k1=y;
	 }
	for(i=0;i<g[x].size();i++){
 		y=g[x][i]; if(par==y) continue;
 		
 		if(y!=k1&&a[k2]<a[y]) k2=y; 
	 }
	mx=max(mx,a[k1]*a[k2]);
	int t=0;
	for(i=0;i<g[x].size();i++){
 		y=g[x][i]; if(par==y) continue;
 		t+=a[y];
 		S-=a[y]*a[y];
	 }
	S+=t*t; S%=mod;
 }
 main(){ 
    int i,x,y;
    cin>>n;
    
    for(i=1;i<n;i++){
    	cin>>x>>y,g[y].push_back(x),g[x].push_back(y);
	}
	for(i=1;i<=n;i++) cin>>a[i];
	dp(1,0);
	cout<<mx<<' '<<S%mod;
 }