洛谷 1351 联合权值

【题解】

　　每个点维护各个儿子的前后缀最大值、权值和，这样就可以统计儿子之间相乘的答案。然后每个节点再乘它的祖父的权值去更新答案即可。

#include<cstdio>
#include<cstring>
#include<algorithm>
#define LL long long
#define rg register
#define N 200010
#define Mod (10007)
using namespace std;
int n,tot,last[N],fa[N],w[N],son[N];
LL Sum,Mx;
struct edge{int to,pre;}e[N<<1];
inline int read(){
    int k=0,f=1; char c=getchar();
    while(c<'0'||c>'9')c=='-'&&(f=-1),c=getchar();
    while('0'<=c&&c<='9')k=k*10+c-'0',c=getchar();
    return k*f;
}
void dfs(int x){
    Mx=max(Mx,1ll*w[x]*w[fa[fa[x]]]);
    Sum+=2*w[x]*w[fa[fa[x]]]; Sum%=Mod;
    LL s=0,premx=-2e9; int top=0;
    for(rg int i=last[x],to;i;i=e[i].pre)if((to=e[i].to)!=fa[x]){
        fa[to]=x; s+=w[to]; son[++top]=to;
        Mx=max(Mx,premx*w[to]); premx=max(premx,1ll*w[to]);
    }
    premx=-2e9;
    for(rg int i=top,to;i;i--){
        Mx=max(Mx,premx*w[to=son[i]]); premx=max(premx,1ll*w[to]);
        Sum+=w[to]*(s-w[to]); Sum%=Mod;
    }
    for(rg int i=last[x],to;i;i=e[i].pre)if((to=e[i].to)!=fa[x]) dfs(to);
}
int main(){
    n=read();
    for(rg int i=1;i<n;i++){
        int u=read(),v=read();
        e[++tot]=(edge){u,last[v]}; last[v]=tot;
        e[++tot]=(edge){v,last[u]}; last[u]=tot;
    }
    for(rg int i=1;i<=n;i++) w[i]=read();
    dfs(1);
    printf("%lld %lld
",Mx,Sum%Mod);
}