前言:
之前那个博客有大问题...确实是数据太水了.
思路:
首先我们处理出4e4以内的质数,然后因为任何合数都可以写成的形式,显然大于4e4的数,假如经过这么一次筛选,一定是不会出现超过4e4的合数.对于每个数,我们处理出来它们的质因子,这里呢,用的是最原始的判断方法,假如有n个数是大质数,那么就高达
了.所以假如这样的话,最好还得套个米可罗宾先判断它是不是大质数.后面对于每个质因子搜其点,找到两条最长链即可了(随便树形dp一下).因为一个数的质因子个数最多不超过30个(1e9以内.)所以你的搜索上线也就
.然后就得以解决此问题.
代码:
#include <bits/stdc++.h>
using namespace std;
const int N=1e5+5,MM=1e5;
int w[N];
vector<int>g[N];
//map<int,vector<int> >s;
unordered_map<int,vector<int> >s;
int Ans = 1,Mod,A[11] = {0,2,3,5,7,11,13,17,19,23,61}; // A就是Miller Rabin 探测的底数
bool vis[N];
int Prime[N],tot = 0 ,cnt = 0,book[N];
int prime[N*10];
unordered_map<int,bool> M;//记录哪一些质因子出现过
void GetPrime() {
for(int i = 2 ; i <=4e4 ; i ++) {
if(!book[i]) Prime[++cnt] = i;
for(int j = 1 ; j <= cnt ; j ++) {
if(Prime[j] * i > 4e4) break;
book[Prime[j] * i] = 1;
if(i % Prime[j] == 0) break;
}
}
return ;
}
long long quickpower(long long x,long long y) {
long long ans = 1 , op = x;
while(y) {
if(y & 1ll) ans *= op , ans %= Mod;
op *= op , op %= Mod;
y >>= 1ll;
}
return ans;
}
int Miller_Rabin(int x) {//素数探测算法
Mod = x;
int s = 0;
if(x == 2) return 1;
if(x % 2 == 0 || x == 1) return 0;
int p = x - 1 ;
while(p % 2 == 0) p/= 2 , s ++;
for(int j = 1 ; j <= 5 ; j ++) {
long long B = quickpower(A[j] , p);
for(int i = 1 ; i <= s ; i ++) {
long long K = (B * B) % Mod;
if(K == 1ll && B != 1ll && B != Mod - 1) return 0;//如果二次探测发现这个不是质数
B = K;
}
if(B != 1ll) return 0;//利用费马小定理判断
}
return 1;//探测正常结束
}
int ans=-1;
int dfs(int u,int num)
{
vector<int>a;
if(w[u]%num!=0) return 0;
vis[u]=true;int res=1;
for(int x:g[u])
{
if(!vis[x])
{
int Cnt=dfs(x,num);
res=max(res,1+Cnt);
ans=max(ans,res);
a.push_back(Cnt);
}
}sort(a.begin(),a.end());
int x=a.size();
if(x>1) ans=max(ans,a[x-1]+a[x-2]+1);
return res;
}
int main()
{
int m=4e4,id=0;GetPrime();
int n;scanf("%d",&n);
for(int i=1;i<n;i++)
{
int u,v;scanf("%d%d",&u,&v);
g[u].push_back(v);
g[v].push_back(u);
}
for(int i=1;i<=n;i++)
{
scanf("%d",&w[i]);int x=w[i];
if(Miller_Rabin(x))
{
s[x].push_back(i);
if(M[x] != 1) M[x] = 1,prime[++id]=x;
continue;
}
for(int j=1; Prime[j] * Prime[j] <= x;j++)
{
if(x%Prime[j]==0)
{
s[Prime[j]].push_back(i);
while(x%Prime[j]==0) x/=Prime[j];
if(M[Prime[j]] == 0) M[Prime[j]] = 1 , prime[++id] = Prime[j];
}
}if(x != 1) {
s[x].push_back(i);
if(M[x] != 1) M[x] = 1 , prime[++id] = x;
}
}
for(int i=1;i<=id;i++)
{
for(int j:s[prime[i]])
if(!vis[j]) dfs(j,prime[i]);
for(int j:s[prime[i]]) vis[j]=false;
}printf("%d\n",ans);
return 0;
}
/*
4
1 2
1 3
1 4
2 2 2 2
*/
京公网安备 11010502036488号