思路
题意就不说了。
首先看这个式子,
∑
i
=
1
n
−
1
∑
j
=
i
+
1
n
f
(
i
,
j
)
∑\limits_{i=1}^{n−1}∑\limits_{j=i+1}^nf(i,j)
i=1∑n−1j=i+1∑nf(i,j)可以发现,其实就是所有的点对都连了一遍,也就是任意两个点都不重复的连了一遍。
那么假设有一条边L,L左边有
x
x
x个节点,那么L右边一定就有
n
−
x
n-x
n−x个节点,那么通过L的次数就是
(
n
−
x
)
x
(n -x)x
(n−x)x那么只需要dfs一遍就可以算出所有边的经过次数了。
然后对于给的m个质数需要分情况讨论:
- 如果 m < = n − 1 m <=n-1 m<=n−1那么就可以直接把经过次数多的边匹配大的质数,剩下不足的匹配1。
- 如果 m > n − 1 m > n - 1 m>n−1那么,可以先把大的质数合并,然后把合并出来的大质数当成一个整体匹配最大经过次数的边,然后剩下的质数再匹配剩下的边。
代码
#include<bits/stdc++.h>
#define ll long long
#define INF 0x3f3f3f3f
#define endl '\n'
#define eps 1e-8
using namespace std;
const ll maxn = 1e6 + 5;
const ll mod = 1e9 + 7;
int t,a[maxn],b[maxn];
int n,m;
int h[maxn],nex[maxn],e[maxn],cnt = 1;
ll p[maxn];
vector<ll> dd;
void add(int a, int b){
e[cnt] = b;
nex[cnt] = h[a];
h[a] = cnt++;
}
ll dfs(int x,int f){
ll sum = 1;
for(int i = h[x]; i ; i = nex[i]){
int endd = e[i];
if(endd == f)continue;
ll uu = dfs(endd,x);//得到右边的节点个数
dd.push_back(1ll * uu * (n - uu));
sum += uu;
}
return sum;
}
int main(){
ios::sync_with_stdio(false);
int u,v;
cin >> t;
while(t--){
ll ans = 0;
cnt = 1,dd.clear();
cin >> n;
for(int i = 1; i <= n; i++)h[i] = 0;
for(int i = 1; i <= n - 1; i++){
cin >> u >> v;
add(u,v),add(v,u);
}
cin >> m;
for(int i = 0; i < m; i++)cin >> p[i];
dfs(1,0);
sort(dd.begin(), dd.end(), greater<ll>());
sort(p, p + m, greater<ll>());
if(m <= n - 1){//如果质数的个数不足匹配所有的边
for(int i = 0; i < m; i++)
ans = (dd[i] * p[i] % mod + ans) % mod;
for(int i = m; i < n - 1; i++)ans = (ans + dd[i]) % mod;
}else{
ll tmp = 1;//tmp用于储存前几个大质数合并的结果
for(int i = 0,f = 0,j = 0; i < n - 1;){
if(f){
ans = (ans + p[j++] * dd[i++] % mod) % mod;
}else{
tmp = tmp * p[j++] % mod;
if(m - j == n - 2){//合并结束,把标记f打开,然后匹配剩下的边
f = 1;
ans = (ans + tmp * dd[i++] % mod) % mod;//第一个最大通过的边匹配合并后的数
}
}
}
}
cout << ans << endl;
}
return 0;
}