树上边的被经过次数

题意：

题目链接

一棵树，给出n-1条边，为每条边分配一个值，满足三个条件，1：所有的边权值乘起来等于一个数k，2：边权值都大于0

3：边权值中1出现的次数最少。   这棵树的两个结点之间的路径上的边权之和为x, 求出所有的x之和的最大值。

k是一个很大的数，所以把它的所有质因子给出，质因子可能是重复的。比如 4的质因子是2， 会给出两个2。

 

思路：

求所有两点之间的路径上的边权和最大值。就是找出边的被经过次数，然后以这个次数为权值，来分配边权值。被经过次数越多，就分配更大的边权。如果质因子很少，就让一些边权为1， 如果质因子很多， 就让被经过次数最多的那个边的权值 是最大的那些质因子之积。

求结点i和它父亲相连的那条边 被经过的次数， 应该是子树的大小 * （n - 子树的大小）。  画个图就明白了，看样例也行。

#pragma warning(disable:4996)
#include<iostream>
#include<algorithm>
#include<cstdio>
#include<map>
#include<cmath>
#include<vector>
#include<cstring>
#include<string>
using namespace std;
typedef long long ll;
const ll maxn = 1e5 + 10;
const ll mod = 1e9 + 7;
ll siz[maxn], p[maxn], b[maxn];
//siz 子树的大小， p是输入的质因子，b是边被经过的次数
ll n, m;
vector<ll>edge[maxn];
void dfs(ll u, ll fa) {//求子树大小
	siz[u] = 1;
	for (auto t : edge[u])
	{
		if (t == fa)continue;
		dfs(t, u);
		siz[u] += siz[t];
	}
}
int main()
{
	ll T, i, j, u, v;
	scanf("%lld", &T);
	while (T--)
	{
		scanf("%lld", &n);
		for (i = 1; i <= n; i++)//初始化
			siz[i] = 0, p[i] = 1, edge[i].clear();
		for (i = 1; i < n; i++)
		{
			scanf("%lld%lld", &u, &v);
			edge[u].push_back(v);
			edge[v].push_back(u);
		}
		dfs(1, 0);
		scanf("%lld", &m);
		for (i = 1; i <= m; i++)scanf("%lld", &p[i]);
		if (m < n - 1)m = n - 1;
		sort(p + 1, p + 1 + m);//对p先进行排序
		for (i = n; i <= m; i++)
		{
			p[n - 1] = (p[n - 1] * p[i]) % mod;
		}//如果有用不完的质因子， 就全部给n-1
		ll ans = 0;
		for (i = 1; i < n; i++)
			b[i] = (n - siz[i + 1]) * siz[i + 1];
		//求b的时候，不能对b进行%mod，不然会错 
		sort(b + 1, b + n);
		for (i = 1; i < n; i++)
		{
			ans = (ans + (b[i] % mod * p[i] % mod) % mod) % mod;
		}
		printf("%lld\n", ans);
	}
	return 0;
}