可以推出第z个点的联合权值为与这个点相连的点的权值和的平方减去这些点的权值的平方和(网上有);最大联合权值格外设变量保存。由于N过大,所以用vector数组。
#include<bits/stdc++.h>
using namespace std;
long long l,m,n;
vector <long long> ha[200010];
long long kao[200010];
int main(){
// freopen("link.in","r",stdin);
// freopen("link.out","w",stdout);
scanf("%lld",&n);
for(long long z=1;z<n;++z){
long long a,b;
scanf("%lld%lld",&a,&b);
ha[a].push_back(b);
ha[b].push_back(a);
}
for(long long z=1;z<=n;++z) scanf("%lld",&kao[z]);
for(long long z=1;z<=n;++z){
long long q=0,w=0;
long long e=0,r=0;
for(long long y=0;y<ha[z].size();++y){
if(q<kao[ha[z][y]]){
w=q;
q=kao[ha[z][y]];
}
else if(w<kao[ha[z][y]]) w=kao[ha[z][y]];
e+=kao[ha[z][y]];
l-=(kao[ha[z][y]]*kao[ha[z][y]]);
}
if(m<q*w) m=q*w;
l+=(e*e);
l%=10007;
}
printf("%lld %lld",m,l);
}