Luogu P1351 联合权值___领接表+思维

题目大意：

无向连通图G有n个点，n-1条边，点从1到n依次编号，编号为i的点的权值为Wi每条边的长度均为1图上两点(u,v)的距离定义为u点到v点的最短距离。对于图G上的点对(u,v)，若它们的距离为2，则它们之间会产生Wu×Wv 的联合权值。
请问图G 上所有可产生联合权值的有序点对中，联合权值最大的是多少？所有联合权值之和是多少？

题解：

这题仔细想想其实很简单，
我们设
sum[i]表示点i所连的所有点的权值和
cmax[i]表示点i所连的点权值最大的点的位置
dmax[i]表示点i所连带点权值次大大点点位置
我们对所有的边建一个邻接表即可O(N)的去处理出这些东东
然后我们就可以直接枚举一个点以及与它相邻的点即可..
因为是用领接表，时间复杂度为O(N)

代码：

#include<bits/stdc++.h>
#define N 200005

using namespace std;

int cnt,n,dmax[N],cmax[N],next[2*N],to[2*N],ls[N];
long long sum[N],w[N];

void add(int u,int v){
     cnt++; to[cnt]=v; next[cnt]=ls[u]; ls[u]=cnt;
     cnt++; to[cnt]=u; next[cnt]=ls[v]; ls[v]=cnt;
}
int main()
{
     scanf("%d",&n);
     int u,v;
     for (int i=1; i<=n-1; i++)
         {
             scanf("%d%d",&u,&v);
             add(u,v); 
         } 
     for (int i=1; i<=n; i++) scanf("%d",&w[i]);
     for (int i=1; i<=n; i++)
          for (int j=ls[i]; j; j=next[j])
              {
                   sum[i]+=w[to[j]];
                   if (w[to[j]]>=w[cmax[i]]){
                       dmax[i]=cmax[i];
                       cmax[i]=to[j];
                   } else if (w[to[j]]>w[dmax[i]]) dmax[i]=to[j];
              }
     long long maxans=0,ans=0; 
     for (int i=1; i<=n; i++)
          for (int j=ls[i]; j; j=next[j])
          {
               if (cmax[to[j]]!=i) maxans=max(maxans,w[i]*w[cmax[to[j]]]);
                              else maxans=max(maxans,w[i]*w[dmax[to[j]]]);
               ans=(ans+w[i]*(sum[to[j]]-w[i]))%10007;
          }
     printf("%lld ",maxans);
     printf("%lld\n",ans);
     return 0;
}