最长树链 ( dfs + 质因数分解 )

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

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

我们不妨枚举最长树链 g c d gcd gcd , 那么该条树链上所有的点的权值可表示成 k ∗ g c d k*gcd kgcd 同样也可表示成 k ∗ g c d ′ ( g c d k*gcd'(gcd kgcd(gcd% g c d ′ = = 0 ) gcd'==0) gcd==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;
}
}
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值