联合权值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*2];
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=1,x,y;i<n;++i)
    {
        scanf("%d%d",&x,&y);
        add(x,y);
        add(y,x);
    }
    for(int i=1;i<=n;++i)
        scanf("%d",&w[i]);
    int jl=0;
    int zui=0,he=0,wkk=0;
    for(int i=1;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%=10007;
                }
            } 
        }
    }
    printf("%d %d",zui,he);
    return 0;
}

---

---

正解

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



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