解析:这个题的贪心听大佬说还是很明显的,最后的答案就是所有边经过的次数边权,当m比n-1大的时候意味着多出来了m-n+1个乘积,因为贪心我们当然选取大点乘在一起比较好,直接乘到n-1这个数上面,那么每条边走过的次数该怎么算呢,size[u](n-size[u]),假设是u,v这条边,那么v的子树中的任意一点和v子树的补图的任意一点之间都经过这条边,次数刚好是这个,最后经过的边的次数排序就完事了。
#include<iostream>
#include<cstring>
#include<vector>
#include<algorithm>
using namespace std;
const int N=200010;
const int mod=1e9+7;
typedef long long ll;
int h[N],e[N],idx,ne[N];
ll sz[N];
ll p[N];
int n,m;
void add(int a,int b)
{
e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}
void dfs(int u ,int fa)
{
// sz[u] = 1;
for(int i=h[u];~i;i=ne[i])
{
int j=e[i];
if(j==fa) continue;
dfs(j,u);
sz[u]+=sz[j];
}
//sz[u]=sz[u]*(n-sz[u]);
}
int main()
{
int t;
cin >> t;
while(t--)
{
memset(h,-1,sizeof h);
idx = 0 ;
//memset(sz,0,sizeof sz);
cin >> n;
for(int i=0;i<n-1;i++)
{
int a,b;
cin >> a >> b;
add(a,b);
add(b,a);
}
cin>>m;
for(int i=1;i<=m;i++)
cin>>p[i];
sort(p+1,p+m+1);
for(int i=1;i<=n;i++)
sz[i]=1;
//cout<<n<<' '<<m<<endl;
dfs(1,-1);
for(int i=1;i<=n;i++)
sz[i] = sz[i]*(n-sz[i]);
sort(sz+1,sz+n+1);
if(m<=n-1)
{
ll res=0;
for(int i=m;i>=1;i--)
res=(res+sz[n--]%mod*p[i]%mod)%mod;
for(int i=n;i>=1;i--)
res = (res+sz[i])%mod;
cout<<res<<endl;
}
else
{
//printf("=-=\n");
ll res=0;
ll tt=1;
for(int i=m;i>=n;i--)
p[n-1]=(p[n-1]*p[i])%mod;
res = (res + p[n-1]*sz[n])%mod;
for(int i=n-2;i>=1;i--)
res=(res+sz[i+1]%mod*p[i]%mod)%mod;
cout<<res<<endl;
}
}
return 0;
}