【前缀和】【前缀MAX】洛谷 P1351 NOIP2014提高组 day1 T2 联合权值

不难发现，树中与某个点距离为2的点只可能是它的父亲的父亲、儿子的儿子 或者 兄弟，分类讨论一下即可。

只有对于兄弟我们不能暴力搞，维护一下每个节点的所有儿子的前缀和、前缀MAX就行了。

 #include<cstdio>
 #include<algorithm>
 #include<vector>
 using namespace std;
 #define N 200001
 #define MOD 10007
 int n;
 vector<int>G[N],son[N];
 typedef long long ll;
 typedef vector<int>::iterator ITER;
 int x,y,fa[N],dep[N],maxv,sumv,w[N];
 void dfs(int U)
 {
     for(ITER it=G[U].begin();it!=G[U].end();it++)
       if((*it)!=fa[U])
         {
           fa[*it]=U;
           dep[*it]=fa[U]+;
           son[U].push_back(*it);
           dfs(*it);
         }
 }
 void dfs2(int U)
 {
     if(dep[U]>=) sumv=(sumv+w[U]*w[fa[fa[U]]]%MOD)%MOD;
     ll All=,t=;
     for(ITER it=son[U].begin();it!=son[U].end();it++) All+=(ll)w[*it];
     for(ITER it=son[U].begin();it!=son[U].end();it++)
       {
           t+=(ll)w[*it];
           sumv=(sumv+w[*it]*(int)(All-t)%MOD)%MOD;
           dfs2(*it);
       }
 }
 void dfs3(int U)
 {
     if(dep[U]>=) maxv=max(maxv,w[U]*w[fa[fa[U]]]);
     int t=;
     for(int i=;i<son[U].size();i++)
       {
           if(i)
             {
                 t=max(t,w[son[U][i-]]);
                 maxv=max(maxv,w[son[U][i]]*t);
             }
           dfs3(son[U][i]);
       }
 }
 int main()
 {
     scanf("%d",&n);
     for(int i=;i<n;i++)
       {
           scanf("%d%d",&x,&y);
           G[x].push_back(y);
           G[y].push_back(x);
       }
     for(int i=;i<=n;i++) scanf("%d",&w[i]);
     dfs(); dfs2(); dfs3();
     printf("%d %d\n",maxv,(sumv<<)%MOD);
     return ;
 }