P1351 联合权值
Posted ukcxrtjr
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了P1351 联合权值相关的知识,希望对你有一定的参考价值。
题面:https://www.luogu.org/problem/P1351
本题可以直接枚举每个点所相连的点,因为子-->父-->子距离刚好是2,所以最大权值就由儿子中的权值最大值和权值次大值的乘积来提供,而和可以直接统计,最后乘2即可,因为联合权值的产生是成对的.
Code:
#include<cstdio>
#include<iostream>
using namespace std;
const int N=200005,mod=10007;
int n,ans,maxans,cnt,head[N],w[N];
struct Node{
int v,nxt;
}edge[N*2];
void add(int u,int v){
++cnt;
edge[cnt].v=v;
edge[cnt].nxt=head[u];
head[u]=cnt;
}
void dfs(int u){
int sum=0,maxf=0,maxs=0;
for(int i=head[u];i;i=edge[i].nxt){
int v=edge[i].v;
if(w[v]>maxf){
maxs=maxf;
maxf=w[v];
}
else if(w[v]>maxs){
maxs=w[v];
}
ans=(ans+sum*w[v])%mod;
sum=(sum+w[v])%mod;
}
maxans=max(maxans,maxf*maxs);
}
int main(){
int u,v;
scanf("%d",&n);
for(int i=1;i<n;i++){
scanf("%d%d",&u,&v);
add(u,v);
add(v,u);
}
for(int i=1;i<=n;i++){
scanf("%d",&w[i]);
}
for(int i=1;i<=n;i++){
dfs(i);
}
printf("%d %d
",maxans,(ans*2)%mod);
}
以上是关于P1351 联合权值的主要内容,如果未能解决你的问题,请参考以下文章