[noip2014 d1t2] 联合权值

感觉自己越来越弱了...

题目链接

对于60%的数据，朴素枚举即可

对于100%的数据，考虑优化

我们注意到每条边的长度均为1，而要求点对距离为2，可见和同一个点有连边的两点之间就会产生权值

对于第一问，显然可以贪心保存最大和次大值从而避免n³的查询；

对于第二问，一个讲的清楚的-->链接

记sum为与一个结点相连的所有点的权值之和，则对于与之相连的每个点i，由结合律，它对答案的贡献就是w[i]*(sum-w[i])

抓住题目的特殊条件，从分析数据间的关系入手，便可有效地避免冗杂的运算，将时间复杂度降在O(k*n)，k为较小的常数

#include<cstdio>
#include<cstring>
#include<vector>
#define maxn 200200
#define MOD 10007
using namespace std;
int n;
int weight[maxn];
vector<int> con[maxn];
int read(){
    int x=0,f=1;
    char ch=getchar();
    while (ch<'0'||ch>'9'){
        if (ch=='-') f=-1;
        ch=getchar();
    }
    while (ch>='0'&&ch<='9'){
        x=x*10+ch-'0';
        ch=getchar();
    }
    return x*f;
}
int main(){
    int ans1=0,ans2=0;
    n=read();
    for (int i=1;i<n;i++){
        int u=read(),v=read();
        con[u].push_back(v);
        con[v].push_back(u);
    }
    memset(weight,0,sizeof(weight));
    for (int i=1;i<=n;i++)
      weight[i]=read();
    for (int i=1;i<=n;i++){
        int m1=0,m2=0,sum=0;
        for (int j=0;j<con[i].size();j++){
            if (weight[con[i][j]]>=m1){
                m2=m1;m1=weight[con[i][j]];
            }
            else if (weight[con[i][j]]>m2){
                m2=weight[con[i][j]];
            }
            sum+=weight[con[i][j]];
        }
        sum%=MOD;
        ans1=max(ans1,m1*m2);
        for (int j=0;j<con[i].size();j++){
            ans2+=(weight[con[i][j]]*(sum-weight[con[i][j]]))%MOD;
            ans2=(ans2+MOD)%MOD;
        }
    }
    printf("%d %d
",ans1,ans2);
    return 0;
}

 黄学长树规的做法也值得借鉴