显然,对于一个点x,如果有两条边x-y x-z,那么yz就是所要求的。实际上题目中的yz和zy算不同的(真是坑爹害得我考了long long,本来都想用2的逆了o(╯□╰)o),所以实际上是2yz。我们知道平方和展开
(Σ(i=1,n)a[i])^2=Σ(i=1,n)a[i]^2+Σ(i=1,n)Σ(j=i+1,n)2a[i]*a[j],而加号右边就是联合权值,另外两部分都是可是求出来的,减一减就行了。
AC代码如下:
#include<iostream>
#include<cstdio>
#define mod 10007
#define ll long long
#define N 400005
using namespace std;
int n,tot,a[N],pnt[N],nxt[N],fst[N];
void add(int aa,int bb){
pnt[++tot]=bb; nxt[tot]=fst[aa]; fst[aa]=tot;
}
int main(){
scanf("%d",&n); int i;
for (i=1; i<n; i++){
int x,y; scanf("%d%d",&x,&y);
add(x,y); add(y,x);
}
for (i=1; i<=n; i++) scanf("%d",&a[i]);
ll ans=0; int mx=0;
for (i=1; i<=n; i++){
int p,mx1=0,mx2=0; ll sum1=0,sum2=0;
for (p=fst[i]; p; p=nxt[p]){
int j=pnt[p]; sum1+=a[j]; sum2+=a[j]*a[j];
if (a[j]>mx1){ mx2=mx1; mx1=a[j]; } else
if (a[j]>mx2) mx2=a[j];
}
if (mx1*mx2>mx) mx=mx1*mx2;
sum1=(sum1*sum1-sum2)>>1;
ans=(ans+sum1)%mod;
}
printf("%d %lld\n",mx,(ans<<1)%mod);
return 0;
}
by lych
2015.12.22