codeforces 1101D

传送门：https://codeforces.com/contest/1101/problem/D

题意：求一条最长的链，链上所有点的gcd大于1。

题解：维护每个点的每个质因子能向下延伸多少。

#include<bits/stdc++.h>
using namespace std;
#define fi first
#define se second
#define mp make_pair
typedef long long ll;
typedef pair<int,int> pii;
typedef pair<ll,ll> pll;
const int MAXN=(int)2e5+5;
const int MOD=(int)1e9+7;
struct edge{
    int to,nxt;
}ed[MAXN<<1];
int head[MAXN],cnt;
void addedge(int u,int v){
    ed[cnt]={v,head[u]};
    head[u]=cnt++;
}
vector<int>pri;
bool vis[MAXN];
vector<pii>dp[MAXN];
void pri_init(int n){
    for(int i=2;i<=n;i++){
        if(!vis[i])pri.push_back(i);
        for(auto p:pri){
            if(i*p>n)break;
            vis[i*p]=1;
            if(i%p==0)break;
        }
    }
}
int ans=0;
void Merge(vector<pii>&a, vector<pii> &b){
    auto pa=a.begin(),pb=b.begin();
    while(pa!=a.end()&&pb!=b.end()){
        if(pa->fi<pb->fi)pa++;
        else if(pa->fi>pb->fi)pb++;
        else {
            ans=max(ans,pa->se+pb->se);
            pa->se=max(pa->se,pb->se+1);
            pa++,pb++;
        }
    }
}
void dfs(int u,int pre=-1){
    for(int i=head[u];i!=-1;i=ed[i].nxt){
        int v=ed[i].to;
        if(v==pre)continue;
        dfs(v,u);
        Merge(dp[u],dp[v]);
    }
}
int main()
{
    pri_init(1000);
    int n;
    scanf("%d",&n);
    fill(head+1,head+1+n,-1);
    for(int i=1;i<=n;i++){
        int x;
        scanf("%d",&x);
        for(auto p:pri){
            if(p*p>x)break;
            if(x%p==0){
                dp[i].push_back(mp(p,1));
                while(x%p==0)x/=p;
            }
        }
        if(x!=1)dp[i].push_back(mp(x,1));
        if(!dp[i].empty())ans=1;
    }
    for(int i=1;i<n;i++){
        int u,v;
        scanf("%d%d",&u,&v);
        addedge(u,v);
        addedge(v,u);
    }
    dfs(1);
    printf("%d\n",ans);
    return 0;
}

猜你喜欢