【洛谷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=0; char c=getchar();
    while(c<'0'||c>'9') c=getchar();
    while('0'<=c&&c<='9') { x=(x<<3)+(x<<1)+c-'0'; c=getchar(); }
    return x;
}
struct NODE{
    int to,next;
} e[N<<1];
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=1;i<n;i++){
        x=read(); y=read();
        add(x,y); add(y,x);
    }
    for(int i=1;i<=n;i++)
     w[i]=read();
    for(int i=1;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])%10007;
     }
    printf("%lld %lld",Max,Sum);
    return 0;
}