NOIP2014 Day1T2 联和权值 数学&图论

       显然，对于一个点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