Luogu P1351 联合权值

P1351 联合权值

题目描述

无向连通图 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 上所有可产生联合权值的有序点对中,联合权值最大的是多少?所有联合权值之和是多少?

输入输出格式

输入格式:

 

第一行包含 11 个整数 nn。

接下来 n-1n−1 行,每行包含 22 个用空格隔开的正整数 u,vu,v,表示编号为 uu 和编号为 vv 的点之间有边相连。

最后 11 行,包含 nn 个正整数,每两个正整数之间用一个空格隔开,其中第 ii 个整数表示图 GG 上编号为 ii 的点的权值为 W_iWi​。

 

输出格式:

 

输出共 11 行,包含 22 个整数,之间用一个空格隔开,依次为图 GG 上联合权值的最大值和所有联合权值之和。由于所有联合权值之和可能很大,输出它时要对1000710007取余。

 

输入输出样例

输入样例#1: 复制

5  
1 2  
2 3
3 4  
4 5  
1 5 2 3 10 

输出样例#1: 复制

20 74

说明

本例输入的图如上所示,距离为2 的有序点对有( 1,3)(1,3) 、( 2,4)(2,4) 、( 3,1)(3,1) 、( 3,5)(3,5)、( 4,2)(4,2) 、( 5,3)(5,3)。

其联合权值分别为2 、15、2 、20、15、20。其中最大的是20,总和为74。

【数据说明】

对于30%的数据,1 < n \leq 1001<n≤100;

对于60%的数据,1 < n \leq 20001<n≤2000;

对于100%的数据,1 < n \leq 200000, 0 < W_i \leq 100001<n≤200000,0<Wi​≤10000。

保证一定存在可产生联合权值的有序点对。

 

 

30p:floyd枚举出两个点的距离,再枚举两个点O(n^3)

60p:距离就2,就两种情况:一个点是另一个点的祖先,或者两个点的父亲一样,所以可以枚举两个点,O(1)暴力判断

100p:枚举中间点,只用知道所以与它直接连接的点,用链式前向心存一下均摊O(1),

       答案为w1*w1+w1*w2+……w1*wn+w2*w1+w2*w2+……

          =sumW*sumW-sum(W^W)

#include<cstdio>
#include<iostream>
const int p=10007;
using namespace std;
int read()
{
    int ret=0;
    char ch=getchar();
    while(ch<'0'||ch>'9') ch=getchar();
    while(ch>='0'&&ch<='9')
        ret=(ret<<1)+(ret<<3)+ch-'0',ch=getchar();
    return ret;
}
 
const int N=1e6+5;
int n,ans,mx,w[N],d1,d2,s1,s2;
int cnt,he[N],to[N],nxt[N];
 
inline void add(int u,int v)
{
    to[++cnt]=v;
    nxt[cnt]=he[u];
    he[u]=cnt;
}
 
int main()
{
    n=read();
    for(int i=1;i<n;i++)
    {
        int u=read(),v=read();
        add(u,v),add(v,u);
    }
    for(int i=1;i<=n;i++)
        w[i]=read();
    for(int i=1;i<=n;i++)
    {
        d1=d2=s1=s2=0;
        for(int e=he[i];e;e=nxt[e])
        {
            int v=to[e];
            if(w[v]>d1) d2=d1,d1=w[v];
                else d2=max(d2,w[v]);
            s1=(s1+w[v])%p;
            s2=(s2+w[v]*w[v])%p;
        }
        mx=max(mx,d1*d2);
        ans=(ans+s1*s1-s2)%p;
    }
    printf("%d %d\n",mx,ans);
    return 0;
} 

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值