Codevs 联合权值

点击就送屠龙宝刀

题目描述 Description

输入描述 Input Description

输出描述 Output Description

blob.png
样例输入 Sample Input

样例输出 Sample Output


数据范围及提示 Data Size & Hint

此题目有坑！不用在意说什么(u,v)是最短距离，直接从第一个点dfs，每隔一层递加一遍并取max，同时要注意及时取模以免炸int。。（反正我是int存的）。记忆化什么的就不用说了。。这题太水直接贴代码

#include <iostream>
#include <cstdio>
#include <vector>
#include <algorithm>

using namespace std;

long long n,p,k,w[200010];
bool vis[200010];
vector <int> edge[200010];

void dfs(int x,int last1,int last2)
{
    vis[x]=1;
    long long t=0,t2=0,t3=0,t4=0;
    for (int i=0;i<edge[x].size();i++)
    {
        int y=edge[x][i];
        if (!vis[y])
        {
            vis[y]=1;
            if (last1!=0)
            {
                k=(k+w[y]*w[last1])%10007;
                p=max(p,w[y]*w[last1]);
            }
            t2+=t*w[y]; t+=w[y];
            if (w[y]>=t3)
            {
                t4=t3; t3=w[y];
            }
            if (w[y]<t3) t4=max(t4,w[y]);
            dfs(y,x,last1);
        }
    }
    p=max(t3*t4,p); k=(k+t2)%10007;
}

int main()
{
    cin>>n;
    for (int i=1;i<n;i++)
    {
        int u,v;
        cin>>u>>v;
        edge[u].push_back(v);
        edge[v].push_back(u);
    }
    for (int i=1;i<=n;i++) cin>>w[i];
    dfs(1,0,0);
    cout<<p<<' '<<k*2%10007<<endl;
    return 0;
}