【洛谷P1351】[NOIP2014]联合权值

联合权值

题目链接

首先，直接两重循环暴力枚举得了70分

然后发现第二重循环可以记忆化一下

记忆一下每个点的子节点的权值和、最大值、

次大值（为了处理该点的父节点权值恰好为最大值）

具体看代码

 #include<iostream>
 #include<cstring>
 #include<cstdio>
 using namespace std;
 #define N 400010
 #define int long long
 int n,w[N],Head[N],tot;
 int Max,Sum,sum[N],max1[N],max2[N];
 inline int read(){
     int x=; char c=getchar();
     while(c<''||c>'') c=getchar();
     while(''<=c&&c<='') { x=(x<<)+(x<<)+c-''; c=getchar(); }
     return x;
 }
 struct NODE{
     int to,next;
 } e[N<<];
 inline void add(int x,int y){
     e[++tot].to=y;
     e[tot].next=Head[x];
     Head[x]=tot;
 }
 #undef int
 int main()
 #define int long long
 {
     n=read();
     int x,y;
     for(int i=;i<n;i++){
         x=read(); y=read();
         add(x,y); add(y,x);
     }
     for(int i=;i<=n;i++)
      w[i]=read();
     for(int i=;i<=n;i++)
      for(int j=Head[i];j;j=e[j].next){
         int son=e[j].to;
         if(!sum[son]&&!max1[son])
             for(int k=Head[son];k;k=e[k].next){
                 int gson=e[k].to;
                 if(max2[son]<w[gson]){
                     max2[son]=w[gson];
                     if(max2[son]>max1[son])
                      swap(max1[son],max2[son]);
                 }
                 sum[son]+=w[gson];
             }
         if(w[i]!=max1[son])
             Max=max(Max,max1[son]*w[i]);
         else Max=max(Max,max2[son]*w[i]);
         Sum=(Sum+(sum[son]-w[i])*w[i])%;
      }
     printf("%lld %lld",Max,Sum);
     return ;
 }