【noip2014】联合权值
题目描述 Description
输入描述 Input Description
输出描述 Output Description
样例输入 Sample Input
样例输出 Sample Output
数据范围及提示 Data Size & Hint
不一定算是树形dp,也就是个暴力= =
一个点周围所有相邻的点之间的距离一定是2,存下来一个点周围所有点权值的最大值和次大值然后就得到了ans1
权值之和就是两两乘起来,存下来一个点周围所有点权值的和,然后运用一下结合律就能算出ans2
【代码】#include<iostream>
#include<cstring>
#include<cstdio>
#define p 10007
using namespace std;
struct hp{
long long u,v;
}e[200005];
long long n,ans1,ans2,u,v;
long long s[200005],w[200005],max1[200005],max2[200005];
//s[i]存储i点周围所有点的权值,w[i]存储i点的权值,maxn1[i]存放i点周围权值的最大值,max2[i]存储i点周围权值的次大值
int main(){
scanf("%lld",&n);
for (int i=1;i<n;++i)
scanf("%lld%lld",&e[i].u,&e[i].v);
for (int i=1;i<=n;++i)
scanf("%lld",&w[i]);
for (int i=1;i<n;++i){
u=e[i].u; v=e[i].v;
s[u]+=w[v];
s[v]+=w[u];
if (w[u]>max1[v]) {max2[v]=max1[v]; max1[v]=w[u];}
else if (w[u]>max2[v]&&w[u]<=max1[v]) max2[v]=w[u];
if (w[v]>max1[u]) {max2[u]=max1[u]; max1[u]=w[v];}
else if (w[v]>max2[u]&&w[v]<=max1[u]) max2[u]=w[v];
}
for (int i=1;i<=n;++i)
ans1=max(ans1,max1[i]*max2[i]);
for (int i=1;i<n;++i){
u=e[i].u; v=e[i].v;
ans2=(ans2+w[v]*(s[u]-w[v])%p)%p;
ans2=(ans2+w[u]*(s[v]-w[u])%p)%p;
}
printf("%lld %lld",ans1,ans2);
}