NOIP 2014 联合权值

评测传送

70分：枚举中间点，然后再两层循环枚举这个点的所有邻接点，两两相乘，加入和中。（可能退化为n^2）
100分：在70分的基础上加一个优化，当我们枚举了中间点，然后再枚举它的邻接点 j 时，用sum[i]表示 i 的所有邻接点的权值的和，那么以 i 作为中间点，j这个点对答案的贡献为 (sum[i]-w[j])*w[j]。
最大值则用两个数组来记 i 的邻接点的权值的最大值和次大值。时间复杂度O(4*n)

70分

#include<iostream>//O(n^2) 
#include<cstring>
#include<cstdio>
#include<cmath>
#include<vector>
#include<ctime>
const int N=200009;
const int MOD=10007;
using namespace std;
int n,w[N],ans,maxn;
//int head[N],nxt[2*N],to[2*N],tot;
vector <int> t[N];
void add(int x,int y)
{
    t[x].push_back(y);
    t[y].push_back(x);
}
int main()
{
    freopen("link.in","r",stdin);
    scanf("%d",&n);
    for(int i=1;i<n;i++)
    {
        int u,v;
        scanf("%d%d",&u,&v);
        add(u,v);
    }
    for(int i=1;i<=n;i++) scanf("%d",&w[i]);
    for(int i=1;i<=n;i++)
    {
        for(int j=0;j<t[i].size();j++)
         for(int k=0;k<t[i].size();k++)
          if(j!=k){
            int c=w[t[i][j]]*w[t[i][k]];
            ans=(ans+c)%MOD;
            maxn=max(maxn,c);   
        }
    }
    printf("%d %d",maxn,ans);
    //printf("\n%d",clock());
    return 0;
} 

100分

#include<iostream>//O(4*n) 
#include<cstring>
#include<cstdio>
#include<cmath>
#include<vector>
#include<ctime>
#define LL long long
const int N=200009;
const int MOD=10007;
using namespace std;
int n,maxn,m1[N],m2[N];
int head[N],nxt[2*N],to[2*N],tot,f[N];
bool vis[N];
int sum[N],w[N],ans;
void add(int x,int y)
{
    to[++tot]=y;
    nxt[tot]=head[x];
    head[x]=tot;
}
int main()
{
    //freopen("link.in","r",stdin);
    //freopen("link.out","w",stdout);
    scanf("%d",&n);
    for(int i=1;i<n;i++) 
    {
        int u,v;
        scanf("%d%d",&u,&v);
        add(u,v);add(v,u);
    }
    for(int i=1;i<=n;i++) scanf("%d",&w[i]);

    for(int i=1;i<=n;i++)
    {
        for(int j=head[i];j;j=nxt[j])
        {
            sum[i]+=w[to[j]];
            if(w[to[j]]>m1[i]) m2[i]=m1[i],m1[i]=w[to[j]];
            else if(w[to[j]]>m2[i]) m2[i]=w[to[j]]; 
        }   
    } 
    for(int i=1;i<=n;i++)
    {
        for(int j=head[i];j;j=nxt[j])
        {
            ans=(ans+((sum[i]-w[to[j]])%MOD*w[to[j]]))%MOD;
        }
        maxn=max(maxn,m1[i]*m2[i]);
    }
    printf("%d %d",maxn,ans);
    //printf("\n%d",clock());
    return 0;
}