联合权值dp

　　　　　　　　联合权值

　　　　　　　　　　　　　　　　洛谷中可找到

---

题目传送门https://www.luogu.org/problemnew/show/P1351

这题我就得了70分（TLE）  GG了

就是遍历它孩子的孩子（爷爷和孙子），然后相乘；

ps：这样会有很多重复，TLE

 #include<cstdio>
 #include<iostream>
 #define maxn 200000+10
 using namespace std;
 int max(int x,int y)
 {
     return x>y?x:y;
 }
 int n;
 struct ii{
     int u,v,nxt;
 }a[maxn*];
 int w[maxn];
 int head[maxn];
 int o;
 inline void add(int x,int y)
 {
     a[++o].u=x;
     a[o].v=y;
     a[o].nxt=head[x];
     head[x]=o;
 }
 int main()
 {
 //    freopen("link.in","r",stdin);
 //    freopen("link".out,"w",stdout);
     scanf("%d",&n);
     for(int i=,x,y;i<n;++i)
     {
         scanf("%d%d",&x,&y);
         add(x,y);
         add(y,x);
     }
     for(int i=;i<=n;++i)
         scanf("%d",&w[i]);
     int jl=;
     int zui=,he=,wkk=;
     for(int i=;i<=n;++i)
     {
         for(int j=head[i];j;j=a[j].nxt)
         {
             for(int zz=head[a[j].v];zz;zz=a[zz].nxt)
             {
                 if(a[zz].v!=i)
                 {
                     wkk=w[a[zz].v]*w[i];
                     zui=max(zui,wkk);
                     he+=wkk;
                     he%=;
                 }
             }
         }
     }
     printf("%d %d",zui,he);
     return ;
 }

---

---

正解

#include<iostream>
#include<cstdio>
#define maxn 300000+10
using namespace std;
int n;
int max(int x,int y)
{
    return x>y?x:y;
}//手写max，fast 

struct ii{
    int v,nxt;
}a[];//链式强向星存图1
int head[];
int w[];
int he,ma;//ans 

int o;//存图
void add(int x,int y)
{
    a[++o].v=y;
    a[o].nxt=head[x];
    head[x]=o;
}

int main()
{
     cin>>n;
     int x,y;
     for(int i=;i<=n-;++i)
     {
         scanf("%d%d",&x,&y);//输入
         add(x,y);add(y,x);//正反存一遍
    }
    for(int i=;i<=n;++i)
    {
        scanf("%d",&w[i]);//输入价值
    }
    int sum,zui;//sum是孩子加起来的和，zui是孩子中最大的
    for(int i=;i<=n;++i)
    {
        sum=(zui=w[a[head[i]].v])%;//初始值
        for(int j=a[head[i]].nxt;j;j=a[j].nxt)//从第二个孩子开始遍历
        {
            //这一部分是求和
            he=(he+sum*w[a[j].v])%;//乘法结合律
            sum=(sum+w[a[j].v])%;//更新孩子和
            //这一部分是求最大值
            ma=max(ma,zui*w[a[j].v]);//更新最大值
            zui=max(zui,w[a[j].v]);//更新孩子中的最大值
            //因为这是最后更新的，所以不用担心出现最大值和自己相乘
        }
    }

    printf("%d %d",ma,he*%);//别忘了he*2
     return ;
}