【NOIP2014 提高组】联合权值

题目

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;

}