NOIP2014联合权值

枚举+乘法结合律

枚举每个点左右相邻的两个点乘积和，做(a+b+c...)(a+b+c...)的时候展开减掉a^2+b^2+c^2再除以2就是答案了

注意爆int取模

#include<iostream>
#include<cstdio>
#define N 2000005
using namespace std;
struct node{
    long long u,v;
}e[N];
long long first[N],nxt[N],cnt,max0,ans,w[N];
void add(long long u,long long v){
    e[++cnt].u=u;
    e[cnt].v=v;
    nxt[cnt]=first[u];
    first[u]=cnt;
} 
void sum(long long x){
    long long s[200005],tot=0,num=0,maxv=0,tmaxv=0;
    for(long long i=first[x];i;i=nxt[i]){
        s[++tot]=e[i].v,num+=w[e[i].v];
        num%=10007;
        if(w[e[i].v]>maxv){
            tmaxv=maxv;
            maxv=w[e[i].v];
        } 
        else{
            tmaxv=max(tmaxv,w[e[i].v]);
        }
    }
    num%=10007;
    max0=max(max0,maxv*tmaxv);
    ans+=num*num%10007;
    for(long long i=1;i<=tot;i++)
        ans-=(w[s[i]]*w[s[i]])%10007;
    if(ans<0)ans+=10007;
    ans%=10007;
}
int main(){
    //freopen("link.in","r",stdin);
    //freopen("link.out","w",stdout);
    ios::sync_with_stdio(false);
    long long n;
    cin>>n;
    for(long long i=1;i<n;i++){
        long long a,b;
        cin>>a>>b;
        add(a,b);
        add(b,a);
    }
    for(long long i=1;i<=n;i++)cin>>w[i];
    for(long long i=1;i<=n;i++)
        sum(i);
    cout<<max0<<" "<<ans;
    return 0;
}