洛谷P1351 联合权值

70分暴力做法：枚举每一个点和与其距离为2的点，直接计算联合权值并求最大值，累加求和，复杂度为O(n^3)
暴力代码:

#include<cstdio>
#include
using namespace std;
int
head[2000005],n,x,y,ecnt=1,w[2000005],ans1,maxn,maxm;
long long ans2;
struct list{
      int r,n;
}l[2000005];
void add(int a,int b)
{
      l[ecnt].r=b;
      l[ecnt].n=head[a];
      head[a]=ecnt;
      ecnt++;
}
int main()
{
      scanf("%d",&n);
      for(int i=1;i
      {
             scanf("%d%d",&x,&y);
             add(x,y);
             add(y,x);
      }
      for(int i=1;i<=n;i++)
             scanf("%d",&w[i]);
      for(intj=1;j<=n;j++)//搜索每个点
      {
             ans1=0;
             maxn=0;
             for(intk=head[j];k;k=l[k].n)//搜索该点的每条出边
             {
                    for(intp=head[l[k].r];p;p=l[p].n)//搜索每条边的下一个点并计算
                    {
                           if(l[p].r==j)continue;
                           intmul=w[j]*w[l[p].r];
                           if(mul>ans1)ans1=mul;
                           ans2=(ans2+mul)007;
                    }
             }
             ans2=(ans2+w[j]*ans1)007;
             maxm=max(maxm,maxn*w[j]);
      }
      printf("%d%lld",maxm,ans2007);
      return 0;
}

正解：枚举“中点”并计算

如上图，假设遍历顺序为abcdx，则由于abcd四点都只连接了一条边，故不能成为两点的中点。计算x时，维护一个和值sum，sum为现在已找到的连接到“中点”上的点的权值和，模拟计算过程以说明：

Step1: sum=0 找到a
Step2: sum=a 找到b
Step3: sum=a+b 找到c
Step4: sum=a+b+c 找到d

可以看出，联合权值之和ans=ab+ac+ad+bc+bd+cd

所以可推出ans=ab+(a+b)c+(a+b+c)d
即ans=step[i].sum*w[i]，其中step[i].sum为第i步中sum的值,w[i]为第i步中找到的点的权值。

由此，将图中的每个点都作为“中点”枚举并依照上述方法计算，即可得到某个点作为“中点”时可贡献的联合权值，将所有sum累加即可得到ans

需要注意的是，联合权值一定是成对出现的，即若存在(1,3)，则一定存在(3,1)，且他们的值相等。因此，算出最后的ans后，应乘以2

同时，由于需要求余，根据(a+b) mod c=(a mod c+b mod c) modc，我们可以在每次求出sum后就求余，再累加到ans中，并且再输出ans时对ans求余。

AC代码:

#include<cstdio>
using namespace std;
long long head[2000005],n,x,y,ecnt=1,w[2000005];
long long ans,sum,max1,max2,maxn;
struct list{
	int r,n;
}l[2000005];
void add(int a,int b)
{
	l[ecnt].r=b;
	l[ecnt].n=head[a];
	head[a]=ecnt;
	ecnt++;
}
int main()
{
	scanf("%lld",&n);
	for(int i=0;i<n-1;i++)
	{
		scanf("%lld%lld",&x,&y);
		add(x,y);
		add(y,x);
	}
	for(int i=1;i<=n;i++) scanf("%lld",&w[i]);
	for(int i=1;i<=n;i++)
	{
		sum=0;
		max1=0;
		max2=0;
		for(int j=head[i];j;j=l[j].n)
		{
			int s=w[l[j].r];
			ans+=sum*s;
			sum+=s;
			if(max1<s)
			{
				max2=max1;
				max1=s;
			}
			else if(max2<s) max2=s;
		}
		if(maxn<max1*max2) maxn=max1*max2;
	}
	ans*=2;
	printf("%lld %lld",maxn,ans%10007);
	return 0;
}