版权声明:转载者乖乖♂站好 https://blog.csdn.net/Eric1561759334/article/details/82624820
题目
https://www.luogu.org/problemnew/show/P1351#sub
思路
其实,题目是给了你一棵无根树
枚举每一个点,取其任意两个点,然后进行组合,然后两两相乘,得到最大值与他们的和。将所有值统计一下,然后注意因为和是组合出来的,所以再乘2,输出就可以了。
代码
#include<cstdio>
#include<cmath>
#include<cstring>
#include<iostream>
using namespace std;
struct E
{
int t,nexty;
E(){t=nexty=0;}
}e[1000000];
int h[300000],cnt=0,n,a,b;
long long sum=0,maxn=0,rmax,rsum,num,w[300000];
void add(int a,int b)
{
cnt++;
e[cnt].t=b,e[cnt].nexty=h[a];
h[a]=cnt;
}
int main()
{
scanf("%d",&n);
for(int i=0;i<n-1;i++)
{
scanf("%d%d",&a,&b);
add(a,b),add(b,a);
}
for(int i=1;i<=n;i++)scanf("%lld",&w[i]);
for(int i=1;i<=n;i++)
{
num=h[i];
rsum=(rmax=w[e[num].t])%10007;
num=e[num].nexty;
for(;num!=0;num=e[num].nexty)
{
sum=(sum+rsum*w[e[num].t])%10007;
maxn=max(maxn,rmax*w[e[num].t]);
rsum=(rsum+w[e[num].t])%10007;
rmax=max(rmax,w[e[num].t]);
}
}
//printf("%lld %lld",maxn,(sum*2)%10007);
cout<<maxn<<' '<<sum*2%10007;
return 0;
}