D. Potion Brewing （Codeforces Round #778 (Div. 1 + Div. 2, based on Technocup 2022）

赛后补题.
给你一个n个点带权的树,(因为有n-1条边且连通),然后再给一条边$u,v,x,y$代表$u点权/v点权=x/y$.
你的任务是给树上的点赋值,使得总的点权和最小且满足边之间的比例关系.
思考:首先考虑树高为2的情况,寻找父子关系.首先,如果父亲节点的值确定了,那么每一个儿子的节点的值也一定确定了.
记$x$为父亲节点的比例,$y$为儿子节点的比例.$va{l}_x$为父亲节点的值$va{l}_y$为儿子的值
不难发现,如果要父亲节点都可以满足其儿子节点是一个整数.对于所有的$y$,均有:$va{l}_y=(va{l}_x\ast y)/x$,把$y/x$化为最简的形式后.不难发现,$va{l}_x\%x==0$总是能成立.这也就是说.$va{l}_x=lcm(x1,x2,x3..x_{siz}),siz为儿子的数目$
把这个结论扩大整张图来说,我们知道确定了根的赋值后,整个树的值都会确定.如何来满足每个结点都满足上述的等式呢.
$$va{l}_y=(va{l}_x\ast y)/x$$
当且仅当$va{l}_x\ast y$的所有素数因子的幂大于等于$x$的时候,才有整除成立.
反过来说,当前点的赋值$va{l}_x$与$y的$乘积的某个素数因子不够的时候,上述关系就会失效.那么当素数因子不够的时候怎么办,只能多给它赋值.
$假设缺少p^k$,这就势必会让该点到根的路径上所有的点都要乘上$p^k$.所以不如从根上直接乘上$p^k$.
但我们可以观察到对于每一个点来说,同一个素数$p$需求的幂次$k$是不同的,因为我们是看$va{l}_x\ast y$与$x$的对应素数来说缺了多少的$k$次.
对于一个素数$p$来说,对于点$i$来说对于它的需求记为$need(p,i)$,那么在根上对应素数幂次的取值就是$max(need(p,i))$
那么作法也十分显然了,对这根树进行dfs,那么对于每条边的信息$y,x$。$y的对应素数贡献是1,代表补充va{l}_x,x的贡献是-1,代表需求该素数$.
用一个数组$need(p),记录每个素数需求的次数,need(p)的更新就是dfs过程中每个点对应素数p需求的最大贡献.$
最后根的取值就是$\prod p^{need(p)}$.
再根据根的最佳取值去更新每个点的答案,统计答案即可.
注意,$cnt$数组的更新是包含回溯的,因为不同子树之间的素数因子是没有影响的.
再注意,t很大,随便memset超时警告.
Ac代码:

/*
*/
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 2e5+5;
const int N = 2e5;
const int INF = 1e9+7;
typedef pair<int,int> pii;
const int mod = 998244353;
#define int long long
int vis[maxn];
struct Edge{
	int v,x,y;
};
vector<Edge> G[maxn];
vector<int> fac[maxn];
int cnt[maxn];//prime_fac
int need[maxn];
int f_pow(int a,int b){
	int ans =1;
	while(b){
		if(b&1) ans = (ans*a)%mod;
		b>>=1;a=(a*a)%mod;
	}
	return ans%mod;
}
ll ans = 0;
void init(int n){
	for(int i=0;i<=n;i++) G[i].clear(),cnt[i]=0,need[i]=0;
	ans = 0;
}
void dfs1(int u,int fa){
	for(int i=0;i<G[u].size();i++){
		int v = G[u][i].v;int x=G[u][i].x;int y = G[u][i].y;
		if(v==fa) continue;
		for(auto p : fac[y]) cnt[p]++;
		for(auto p : fac[x]) cnt[p]--,need[p] = max(need[p],(-cnt[p]));
		dfs1(v,u);
		for(auto p : fac[y]) cnt[p]--;
		for(auto p : fac[x]) cnt[p]++; 
	}
}
void dfs2(int u,int fa,ll now){
	//当前节点的赋值是now 
	ans = (ans+now)%mod;
	now = (now%mod);
	for(int i=0;i<G[u].size();i++){
		int v = G[u][i].v;int x = G[u][i].x;int y = G[u][i].y;
		if(v==fa) continue;
		ll val = (((now*y)%mod) * (f_pow(x,mod-2)))%mod;
		dfs2(v,u,val);
	}
}
signed main(){
//	freopen("1.txt","r",stdin);
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	int n;
	//第一步,处理出每个数字i来说,最近的素数. 
	for(int i=2;i<=N;i++){
		if(!vis[i]){
			for(int j=i;j<=N;j+=i){
				vis[j] = i;
			}
		}
	}
	//第二步,预处理出所有数字i,对应的唯一分解.存在fac[i]中 
	for(int i=2;i<=N;i++){
		int tmp = i;
		while(tmp>1){
			fac[i].push_back(vis[tmp]);
			tmp/=vis[tmp];
		}
	}
	int T;cin>>T;
	while(T--){
		int n;cin>>n;
		init(n);
		for(int i=0;i<n-1;i++){
			int u,v,x,y;
			cin>>u>>v>>x>>y;
			G[u].push_back({v,x,y});
			G[v].push_back({u,y,x});
		}
		//第一次dfs,得到need数组 
		dfs1(1,0);
		ll val = 1;//根的值
		for(int i=2;i<=n;i++){
			if(need[i]!=0) val = ( (val) * f_pow(i,need[i]) ) %mod;
		}
		dfs2(1,0,val);
		cout<<ans<<"\n";
	}
return 0;}