题目:联合权值
思路:
首先n个点n-1条边的联通图是一颗无根树。
对于每两个距离为2的点,可以知道,它们一定和一个公共点有连边。
枚举公共点i,再用O(n)的时间算出与这个点相连的点中权值最大的两个点,求出联合权值。最后求出这些联合权值的最大值。
所有联合权值的和,就可以利用乘法分配律的思想。假设每一个和点i有关的联合权值的值为 x*i+y*i+z*i ,那么这个值也就是(x+y+z)*i,所以就可以利用一些小技巧处理x+y+z,就能在O(n^2)的时间内求出了。
注意:
1、最大的两个值的求法不能写错,当当前值在最大的两个值之间时,不能忘记更新次大值。
2、题中只说了和要取模,最大值不能取模。
代码:
#include<bits/stdc++.h>
using namespace std;
#define maxn 200000
#define ll long long
#define md 10007
int n;
int w[maxn+5];
vector<int> a[maxn+5];
ll ans=0,sum=0;
int main() {
scanf("%d",&n);
for(int i=1; i<n; i++) {
int x,y;
scanf("%d%d",&x,&y);
a[x].push_back(y),a[y].push_back(x);
}
for(int i=1; i<=n; i++) {
scanf("%d",&w[i]);
}
for(int i=1;i<=n;i++){
ll max1=0,max2=0;
ll cnt=0;
for(int j=0;j<a[i].size();j++){
if(w[a[i][j]]>max1){
max2=max1;
max1=w[a[i][j]];
} else if(w[a[i][j]]>max2) { //注意不能写掉
max2=w[a[i][j]];
}
sum=(sum+cnt*w[a[i][j]])%md;
cnt=(cnt+w[a[i][j]])%md;
}
ans=max(ans,max1*max2);
}
printf("%lld %lld",ans,(sum*2)%md);
return 0;
}