无向连通图 GG 有 nn 个点,n-1n−1 条边。点从 11 到 nn 依次编号,编号为 ii 的点的权值为 W_iWi,每条边的长度均为 11。图上两点 (u, v)(u,v) 的距离定义为 uu 点到 vv 点的最短距离。对于图 GG 上的点对 (u, v)(u,v),若它们的距离为 22,则它们之间会产生W_v \times W_uWv×Wu 的联合权值。
请问图 GG 上所有可产生联合权值的有序点对中,联合权值最大的是多少?所有联合权值之和是多少?
由于所有联合权值之和可能很大,输出它时要对1000710007取余。
https://www.luogu.org/problemnew/show/P1351
首先,此题说的比较的隐晦。所谓的有n-1条边,n个点的无相连通图,其实就是一颗无根树。换句话说,没有一个环。所以,要是两个点同于另一个点相连,则它们不可能再有边相连,即它们之间的距离为2
思路:枚举每一个点,取其任意两个点,然后进行组合,然后两两相乘,得到最大值与他们的和。将所有值统计一下,然后注意因为和是组合出来的,所以再乘2,输出就可以了
明白这些之后就是优化问题了
对于一个点,因为用邻接表存储,所以线性扫过所有与他相连的点,然后动态更新目前的和与目前这些点中的最大值,然后再到下一个点时,将下一个点的权值与这两个值相乘,然后做相应的处理就行了
因为一个点会延伸出去很多点,每个点会与其他几个点都组合一次,每一次都记录下他们的和和最大值,这样利用乘法分配律的道理就可以很快解决了
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define mo 10007
using namespace std;
int n;
struct node
{
int next,to;
}e[400001];
long long w[200001];
int head[200001];
int cnt;
long long ansmax,ans,maxn,sum;
void add(int x,int y)
{
e[++cnt].next=head[x];
e[cnt].to=y;
head[x]=cnt;
}
int main()
{
scanf("%d",&n);
for(int i=1;i<=n-1;i++){
int x,y;
scanf("%d%d",&x,&y);
add(x,y);add(y,x);
}
for(int i=1;i<=n;i++){
scanf("%d",&w[i]);
}
for(int i=1;i<=n;i++){
int u=head[i];
sum=(maxn=w[e[u].to])%mo;
u=e[u].next;
for(;u;u=e[u].next){
ans=(ans+sum*w[e[u].to])%mo;
ansmax=max(ansmax,maxn*w[e[u].to]);
sum=(sum+w[e[u].to])%mo;
maxn=max(maxn,w[e[u].to]);
}
}
printf("%lld %lld",ansmax,(ans*2)%mo);
return 0;
}