codeforces1401D dfs+贪心

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;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值