树 dp div2 D - GCD Counting

D - GCD Counting

题目链接

题目大意

给一棵树，每个点的点权，然后问路径上的所有点权的gcd > 1的最长路径是多大。
这个感觉就树形dp了。。 但是菜鸡L不会写。。。
点的个数n 1e5.
点权是小于2e5的
刚开始想到了先把点权的质因子弄出来。然后dfs的时候对每个质因子之类的。。
但是呢， 总以为那样的话得开个1e5 * 2e5的数组。。可是用个map不就好了吗。于是就没有写。完了之后查题解恍然大悟。
题解：
先把每个点权的质因子求出来，如果两个节点有相同的质因子gcd就肯定>1。然后dfs的时候，路径过当前根节点枚举每个质因子统计出答案的最大值。。
代码：

#include <cstdio>
#include <algorithm>
#include <vector>
#include <map>
using namespace std;
const int maxn = 2e5+5;
int a[maxn];
std::vector<int> vv[maxn];
std::vector<int> vp[maxn];
map<int,int> mm[maxn];
int ans =0 ;
void dfs(int x,int fa)
{
   
	for (int i =0 ; i< vp[x].size(); i ++ )
	{
   
		mm[x][vp[x][i]] = 1;
		ans = max(ans,1);
	}
	for (int i = 0; i < vv[x].size(); i ++ )
	{
   
		int v = vv[x][i];
		if(v == fa)
			continue;
		dfs(v,x);
		for (int j = 0; j < vp[x].size(); j ++ )
		{
   
			int xx = vp[x][j];
			mm[x][xx] = max(mm[x][xx],mm[v][xx] + 1);
			ans = max(ans,mm[x][xx]);
			for (int k = 0; k < i; k ++ )
			{
   
				ans = max(ans,mm[vv[x][k]][xx] + mm[v][xx] + 1);
			}
		}
	}
}

int main()
{
   
	int n;
	scanf("%d",&n);
	for (int  i= 1; i <= n; i ++ )
	{
   
		scanf("%d",&a[i]);
		int x = a[i];
		for (int j = 2; j * j <= x; j++ )
		{
   
			if(x % j == 0)
			{
   
				while(x % j == 0)
					x /= j;
				vp[i].push_back(j);
			}
		}
		if(x > 1)
		{
   
			vp[i].push_back(x);
		}
	}
	for (int i = 1; i < n; i ++ )
	{
   
		int x,y;
		scanf("%d%d",&x,&y);
		vv[x].push_back(y);
		vv[y].push_back(x);
	}
	dfs(1,0);
	printf("%d\n", ans);
}