最长树链 ( dfs + 质因数分解 )_最长树链题解-CSDN博客

本文链接：https://blog.csdn.net/qhq889/article/details/108782198

博客内容讲述了如何利用质因数分解优化树链剖分算法，解决寻找图中最大公约数最长链的问题。通过将节点按质因子分类并分别求最长链，最终得到答案。代码实现中包括了图的构建、质因子分解以及最长链的DFS遍历过程。

摘要由CSDN通过智能技术生成

题目链接
在这里插入图片描述
解题报告：

我们不妨枚举最长树链 $g c d$ , 那么该条树链上所有的点的权值可表示成 $k * g c d$ 同样也可表示成 $k * g c d^{'} (g c d$ % $g c d^{'} = = 0)$ ,因此我们可以用质因子对这些点分类。对属于同一类质因子的点跑最长链，不断更新最长链即可。每个点被访问的次数就是其质因子种类数。

代码展示：

#include<bits/stdc++.h>
#define LL long long
#define pii pair<int,LL>
#define mem(a,b) memset(a,b,sizeof(a))
using namespace std;
const int maxn=1e5+5;
const int inf=0x3f3f3f3f;
const LL MOD=998244353;
inline int read() {int s = 0, f = 1;char ch = getchar();while (!isdigit(ch)) {f = -1;ch = getchar();}while (isdigit(ch)) {s = (s << 1) + (s << 3) + ch - '0';ch = getchar();}return s * f;}
inline void write(int x) {if (x < 0) x = -x,putchar('-');if (x > 9) write(x / 10);putchar(x % 10 + '0');}

struct TMD{
    int to,next;
}edge[maxn<<2];
int cnt,head[maxn],a[maxn];
int prime[maxn],num,sum;
bool is[maxn],vis[maxn];
vector<int>g[maxn];
map<int,int>mp;
void add(int from,int to){
    edge[++cnt]={to,head[from]};head[from]=cnt;
}
void init(){
    is[0]=is[1]=true;
    for(int i=2;i<=1e5;i++){
        if(!is[i]) prime[num++]=i;
        for(int j=0;i<num&&prime[j]*i<=1e5;j++){
            is[i*prime[j]]=true;
            if(i%prime[j]==0) break;
        }
    }
}
void dfs(int now,int nm){
    sum++;vis[now]=true;
    for(int i=head[now];i;i=edge[i].next){
        int to=edge[i].to;
        if(!vis[to]&&a[to]%nm==0){
            dfs(to,nm);
        }
    }
}
int main() {
    init();
    int n=read(),u,v;
    for(int i=1;i<n;i++){
        u=read();v=read();
        add(u,v);add(v,u);
    }
    for(int i=1;i<=n;i++) a[i]=read();
    for(int i=1;i<=n;i++){
        int tmp=a[i];
        for(int j=0;j<num&&1LL*prime[j]*prime[j]<=tmp;j++){
            if(tmp%prime[j]==0){
                g[prime[j]].push_back(i);
                while(tmp%prime[j]==0) tmp/=prime[j];
            }
        }
        if(tmp>1) {
            if(tmp<=1e5) g[tmp].push_back(i);
            else mp[tmp]++;
        }
    }
    int ans=0;
    for(auto it:mp) ans=max(ans,it.second);
    for(int i=2;i<=1e5;i++){
        for(auto it:g[i]){
            if(!vis[it]){
                sum=0;dfs(it,i);ans=max(ans,sum);
            }
        }
        for(auto it:g[i]){
            vis[it]=false;
        }
    }
    write(ans);
    return 0;
}
}