“蔚来杯“2022牛客暑期多校训练营（加赛） Here is an Easy Problem of Zero-chan

原题题面:https://ac.nowcoder.com/acm/contest/38727/H

题目大意：
给定包含n个节点的有根树，根节点为1，定义
$$f(x)=\prod _{i=1}^{n}lca(x,i)$$
有q次询问，每次询问给定x,求$f(x)$后缀0数量
（$n,q$均不超过$1e5$）

分析：
遇事不决先画图

由上图，可以发现，对于点$A$,所有点与它的$lca$皆在$A$到$root$的路径中
所以可以考虑树形dp
因为$10=2\ast 5$，我们可以预处理每个节点$i$的2和5数量 ,记为$two[i],five[i]$
定义$dp[v].two$为在点v时$lca$积中2的数量
$dp[v].five$为在点v时$lca$积中5的数量
看图可知，$dp[so{n}_i]$的状态是由它上一级的祖先状态的得到的
所以对于每一个非叶子节点$u$的所有$so{n}_i$
状态转移方程为：
$$dp[so{n}_i].two=dp[u].two-size[v]\ast (two[u]-two[so{n}_i])$$
$$dp[so{n}_i].five=dp[u].five-size[v]\ast (five[u]-five[so{n}_i])$$
对于每次询问的x,$$an{s}_x=min(dp[x].two,dp[x].five)$$
时间复杂度：$O(nlogn+n+q)$,是绰绰有余的
代码：

#include<bits/stdc++.h>
#define LL long long

const int MXN=1e5+7;

using namespace std; 
int two[MXN],five[MXN];
int n;

struct Node{
	int Two,Five;
};
int siz[MXN];
Node dp[MXN];

vector<int> g[MXN];
void DFS_1(int now,int fa){
	int v; siz[now]=1;
	for(int i=0;i<g[now].size();i++){
		v=g[now][i];
		if(v==fa) continue;
		DFS_1(v,now);
		siz[now]+=siz[v];
	}
}

void DFS_2(int now,int fa){
	int v;
	for(int i=0;i<g[now].size();i++){
		v=g[now][i];
		if(v==fa) continue;
		dp[v].Two=dp[now].Two-siz[v]*two[now]+siz[v]*two[v];
		dp[v].Five=dp[now].Five-siz[v]*five[now]+siz[v]*five[v];
		DFS_2(v,now);
	}
}

int main(){
	int q,x,u,v;
	scanf("%d%d",&n,&q);
	for(int i=1;i<n;i++){
		scanf("%d%d",&u,&v);
		g[u].push_back(v);
		g[v].push_back(u);
	}
	for(int i=1;i<=n;i++){
		int t=i;
		while(t%2==0){
			two[i]++;
			t>>=1;
		}
		while(t%5==0){
			five[i]++;
			t/=5;
		}
	}
	DFS_1(1,-1);
	dp[1].Two=siz[1]*two[1];
	dp[1].Five=siz[1]*five[1];
	DFS_2(1,-1);
	while(q--){
		scanf("%d",&x);
		printf("%d\n",min(dp[x].Two,dp[x].Five));
	}
}