1401D
题意:给你一棵树n个节点,n-1条边,然后给你m个值且都为素数,m和n之间的大小未知,让你对这n-1条边赋值使得这n-1条边的乘积为m个值的乘积,现让你在用1最少的情况下(即为边的值),使得
最大,简单的说就是节点1到节点2,3,4…n所经过的路程加上节点2到3,4,5…n所经过的路经加上…n-1到n所经过的路经。
思路:简单的说就是如果n>m那么就意味这有n-1-m条边是要被赋值1的,如果n<m,就意味着最大的那条边要多乘上m-n+1条边,这个是对与边的贪心,之后就是难点了,每条边的相对的贡献是多少,也就是这一个遍历过后每条边会经过多少次,用dfs进行深搜,这种对树的深搜做过很多次了,有每个节点有多少个子节点的深搜。详细看代码注释 。
代码:
#pragma GCC optimize("Ofast","inline","-ffast-math")
#pragma GCC target("avx,sse2,sse3,sse4,mmx")
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int, int> pii;
#define rep(i, a, n) for(int i = a; i <= n; i++)
#define per(i, a, n) for(int i = n; i >= a; i--)
#define IOS std::ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
#define fopen freopen("file.in","r",stdin);freopen("file.out","w",stdout);
#define fclose fclose(stdin);fclose(stdout);
const int inf = 1e9;
const ll onf = 1e18;
const int maxn = 1e2+10;
const int mod = 1e9+7;
#define int ll
inline int read(){
int x=0,f=1;char ch=getchar();
while (!isdigit(ch)){if (ch=='-') f=-1;ch=getchar();}
while (isdigit(ch)){x=(x<<3)+(x<<1)+ch-48;ch=getchar();}
return x*f;
}
ll a[maxn], vis[maxn];
std::vector<int> g[maxn], v;
int n;
void dfs(int u, int fa){
vis[u] = 1; //对每个点进行初始化,因为每个节点都必然会经过一次
for(auto to : g[u]){
if(to==fa) continue;
dfs(to, u);
vis[u]+=vis[to]; //这个节点会经过的次数
v.push_back(1ll*vis[to]*(n-vis[to])); //每个点的贡献,或者说这个点和他的父节点直接的边的贡献
//重点,贡献为vis[to]*(n-vis[to]),可以把这vis[to]个节点看成一个点,那么就意味着其他n-vis[to]个点到这个点要经过的次数也就是
//vis[to]*(n-vis[to]),看了好多题解这个公式都是直接出的,果然我还是太菜了
}
}
inline void cf(){
int t = read();
while(t--){
n=read();
rep(i,1,n) g[i].clear();
v.clear();
rep(i,1,n-1){
int x = read(), y = read();
g[x].push_back(y), g[y].push_back(x); //简单建树
}
dfs(1,0); //先对这个树进行一次遍历
//rep(i,1,n) printf("vis[%d]=%d\n", i, vis[i]*(n-vis[i]));
sort(v.begin(), v.end());
int m = read();
rep(i,1,m){
a[i]=read();
}
sort(a+1, a+1+m);
int ans = 0;
n--; //这里n--方便之后的计算
if(m<n){ //贪心
rep(i,1,n-m){
ans = (ans+v[i-1])%mod;
}
rep(i,1,m){
ans = (ans+v[n-m-1+i]*a[i]%mod)%mod;
}
}else {
rep(i,n+1,m) a[n] = a[n]*a[i]%mod;
rep(i,1,n){
ans = (ans+v[i-1]*a[i]%mod)%mod;
}
}
printf("%lld\n", ans);
}
return ;
}
signed main(){
cf();
return 0;
}