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 400005using 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]; } elseif (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