cogs1804 联合权值 dp

填坑……链接：http://cogs.pro/cogs/problem/problem.php?pid=1804

题意：给出一棵树，定义两个距离为2的点权值乘积为这两个点产生的联合权值，求出整棵树最大联合权值及联合权值之和。

实际上早就做过……现在重做旧题感觉还是很有必要……

我们可以枚举每一个点作为可以产生联合权值的两点的中点，对于这个点来说，他周围的点权值之和就是

\sum\limits_{i=1}^N(\sum\limits_{u,v \in adj_i, u \neq v} W_u \times W_v)

然后对于每一个中点，他做出的贡献也就可以表示成

S_i = \sum_{j \in adj_i} (W_j \times (\sum_{k \in adj_i} W_k) - W_j)

然后大力化简可得

S_i ={ (\sum_{j \in adj_i} W_j)} ^ 2 - \sum_{j \in adj_i} W_j^2

这也就是我们需要的求和式子。

至于最大的嘛……直接找到最大的和次大的相乘比较就可以了……

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
const int mod=10007,maxn=200005;
struct node
{
    int from,to,next;
}edge[maxn<<1];
int head[maxn],tot,n,weight[maxn];
void addedge(int u,int v)
{
    edge[++tot]=(node){u,v,head[u]};head[u]=tot;
}
int haha()
{
    freopen("linkb.in","r",stdin);
    freopen("linkb.out","w",stdout);
    scanf("%d",&n);
    for(int i=1;i<n;i++)
    {
        int x,y;scanf("%d%d",&x,&y);
        addedge(x,y);addedge(y,x);
    }
    for(int i=1;i<=n;i++)scanf("%d",&weight[i]);
    int maxx=0;long long sum=0;
    for(int i=1;i<=n;i++)
    {
        int firmax=0,secmax=0;long long sigma=0,pow2sigma=0;
        for(int j=head[i];j;j=edge[j].next)
        {
            int v=edge[j].to;
            if(weight[v]>firmax)secmax=firmax,firmax=weight[v];
            else secmax=max(secmax,weight[v]);
            sigma=(sigma+weight[v]);pow2sigma=(pow2sigma+weight[v]*weight[v])%mod;
        }
        maxx=max(maxx,firmax*secmax);
        sum=(sum+sigma*sigma-pow2sigma)%mod;
    }
    printf("%d %lld",maxx,sum);
}
int sb=haha();
int main(){;}