【NOIP2014提高】联合权值

Description
  无向连通图G有n个点,n-1条边。点从1到n依次编号,编号为i的点的权值为Wi,每条边的长度均为1。图上两点(u,v)的距离定义为u点到v点的最短距离。对于图G上的点对(u,v),若它们的距离为2 ,则它们之间会产生Wu×Wv的联合权值。
  请问图G上所有可产生联合权值的有序点对中,联合权值最大的是多少?所有联合权值之和是多少?
Input
  输入文件名为link.in。
  第一行包含1个整数n 。
  接下来n-1行,每行包含2个用空格隔开的正整数u、v,表示编号为u和编号为v的点之间有边相连。
  最后1行,包含n个正整数,每两个正整数之间用一个空格隔开,其中第i个整数表示图G上编号为i的点的权值为Wi。
Output
  输出文件名为link.out 。
  输出共1行,包含2个整数,之间用一个空格隔开,依次为图G上联合权值的最大值和所有联合权值之和。由于所有联合权值之和可能很大,输出它时要对10007取余。
Sample Input
5
1 2
2 3
3 4
4 5
1 5 2 3 10
Sample Output
20 74
Hint
【数据说明】
  本例输入,距离为2的有序点对有(1,3)、(2,4)、(3,1)、(3,5)、(4,2)、(5,3)。其联合权值分别为2、15、2、20、15、20。其中最大的是20,总和为74。


真不知道去年怎么考的,这么一道水题唉
枚举中间节点

#include<cstdio>
#include<iostream>
#include<vector>
#define ll long long 
using namespace std;
ll maxn,w[200001],ans;
vector<ll> a[200001];
int main()
{
    ll n,u,v;
    cin>>n;
    for(ll i=1;i<n;i++)
    {
        cin>>u>>v;
        a[u].push_back(v);
        a[v].push_back(u);
    }
    for(ll i=1;i<=n;i++) cin>>w[i];
    ll sum1=0;
    for(ll i=1;i<=n;i++)
    {
        ll sum2=0,max1=0,max2=0;
        vector<ll>::iterator it;
        for(it=a[i].begin();it<a[i].end();it++)
        {
            if(w[*it]>max1){max2=max1;max1=w[*it];}
            else
            if(w[*it]>max2) {max2=w[*it];}
            sum1=(sum1+sum2*w[*it]+10007)%10007;
            sum2+=w[*it];
        }
        if(max1*max2>maxn) maxn=max1*max2;
    }
    cout<<maxn<<" "<<(10007+sum1*2)%10007;
} 
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值