「NOIP2014」「Codevs3728」 联合权值（乱搞

3728 联合权值

时间限制: 1 s

空间限制: 128000 KB

题目等级 : 黄金 Gold

 

题目描述 Description

输入描述 Input Description

输出描述 Output Description

样例输入 Sample Input

样例输出 Sample Output

数据范围及提示 Data Size & Hint

题解

这（在洛谷上）都是些什么标签？LCA？dp？（MAX问号脸

随意钦定一个点当根节点，转化成有根树处理。

对于dfs到的每个点，$sum$加上$x$和$fa[fa[x]]$的点权乘积$*2$（$(x,y)$和$(y,x)$要算两次

然后在当前这个点的所有儿子之间乘一下，加到$sum$里。

当然也顺手跟$mac$取个$max$。

然后在处理每个点的儿子之间的恩怨时，很多人用的是$O(n^2)$的两个for。

菊花图会直接让你们TLE的啊qwq，但是这数据蛮水的，没卡他们qaq

所以我写的时候怕T，做了一丢丢优化，搞到了$O(n)$。

反正也没什么用就是了。

 /*
     qwerta
     P1351 联合权值
     Accepted
     100
     代码 C++，1.41KB
     提交时间 2018-10-09 11:26:08
     耗时/内存
     170ms, 7356KB
 */
 #include<iostream>
 #include<cstdio>
 #include<cmath>
 using namespace std;
 const int MAXN=+;
 struct emm{
     int e,f;
 }a[*MAXN];
 int h[MAXN];
 int tot=;
 void con(int u,int v)
 {
     a[++tot].f=h[u];
     h[u]=tot;
     a[tot].e=v;
     a[++tot].f=h[v];
     h[v]=tot;
     a[tot].e=u;
     return;
 }
 int w[MAXN];
 int d[MAXN],fa[MAXN];
 const int mod=;
 long long sum=,ans=;
 int st[MAXN];
 void dfs(int x)
 {
     if(fa[fa[x]])
     {
         sum+=*w[x]*w[fa[fa[x]]];
         sum%=mod;
         ans=max(ans,(long long)w[x]*w[fa[fa[x]]]);
     }
     int tos=;
     long long sumn=;
     int mac1=,mac2=;
     for(int i=h[x];i;i=a[i].f)
     if(!d[a[i].e])
     {
         fa[a[i].e]=x;
         d[a[i].e]=d[x]+;
         st[++tos]=a[i].e;
         sumn+=w[a[i].e];
         if(w[a[i].e]>=mac1){mac2=mac1,mac1=w[a[i].e];}
         else if(w[a[i].e]>mac2){mac2=w[a[i].e];}
     }
     ans=max(ans,(long long)mac1*mac2);
     for(int i=;i<=tos;++i)
     {
         sum+=(sumn-w[st[i]])*w[st[i]];
         sum%=mod;
     }
     //cout<<x<<" "<<ans<<" "<<sum<<endl;
     for(int i=h[x];i;i=a[i].f)
     if(fa[a[i].e]==x)
       dfs(a[i].e);
     return;
 }
 int main()
 {
     //freopen("a.in","r",stdin);
     int n;
     scanf("%d",&n);
     for(int i=;i<n;++i)
     {
         int u,v;
         scanf("%d%d",&u,&v);
         con(u,v);
     }
     for(int i=;i<=n;++i)
       scanf("%d",&w[i]);
     int s=min(n,);
     d[s]=;
     dfs(s);
     cout<<ans<<" "<<sum%mod;
     return ;
 }