Codeforces Round #665 (Div. 2) D. Maximum Distributed Tree ###K ###K //K

题目链接：https://codeforces.ml/contest/1401/problem/D

题意：给每条边标记一个数，要求用的1尽量少，并且乘积之和为k    要求的是 1 到2 3……n , 2到 3,4…… n   n-1到n 的简单路径经过的边上的数字和

 

 

思路:  求出每条边走过的次数，然后贪心的让最多的边和最大的数相乘   如何o(n) 求出每条边被走过的次数呢

考虑一条边的上下两个部分，上部分的任意点都要走到下部分的任意点， 并且因为有大小关系，每对点只有走一次 那么cnt=size[u]*(n-size[u]) size为该点的子树节点

用点表示边， 每个点代表他连着的上面的边， 所以1没有边 即2~n n-1条边

然后讨论 m>n-1的情况 和m<=n-1的情况  前者要将多余的前几个数 乘在一起即为最大

还有就是sort 的时候 用greater<int>的话  只会对内部int类型的排序，  用ll 记得写greater<ll>

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e5+10;
const int mod=1e9+7;
#define ll long long
#define pi pair<int,int>
#define fi first
#define sc second
#define pb push_back
vector<int>E[maxn];
ll p[maxn];
ll cnt[maxn];
int n;
vector<ll>ed;

void dfs(int u,int fa)
{
    cnt[u]=1;
    for(auto &v:E[u])
    {
        if(v==fa) continue;
        dfs(v,u);
        cnt[u]+=cnt[v];
    }
    if(u!=1) ed.pb(cnt[u]*(n-cnt[u]));
}


int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);
    int t;
    cin>>t;
    while(t--)
    {
        cin>>n;
        for(int i=1;i<=n;i++) E[i].clear();
        for(int i=1;i<n;i++)
        {
            int x,y;
            cin>>x>>y;
            E[x].pb(y);
            E[y].pb(x);
        }
        ed.clear();
        int m;
        cin>>m;
        for(int i=1;i<=m;i++) cin>>p[i];
        dfs(1,0);
        vector<ll>eqq;
        sort(p+1,p+1+m,greater<ll>());
        sort(ed.begin(),ed.end(),greater<ll>());
        int pos=m-(n-1)+1;
        ll mul=1;
        for(int i=1;i<=pos;i++)
        {
            mul*=p[i],mul%=mod;
        }
        if(pos>=1) eqq.pb(mul);
        pos=max(pos,0);
        for(int i=pos+1;i<=m;i++) eqq.pb(p[i]);
        while((int)eqq.size()<n-1) eqq.pb(1);
        ll ans=0;
        for(int i=0;i<n-1;i++)
            ans=(ans+ed[i]%mod*eqq[i]%mod)%mod;
        cout<<ans<<'\n';

    }





}