ZR#959

ZR#959

解法:

对于一个询问,设路径 $ (u, v) $ 经过的所有边的 $ gcd $ 为 $ g $,这可以倍增求出。
考虑 $ g $ 的所有质因子 $ p_1, p_2, cdots , p_k $ ,因为 $ g leq 10^6 $ ,所以 $ k leq 7 $ 。
则最终的路径的 $ gcd $ 为 $ 1 $,等价于对于每个 $ 1 leq i leq k $ ,存在至少一条路径上的边不是 $ p_i $ 的倍数。我们要求 $ l $ 的最小值,即等价于对于每个 $ 1 leq i leq k $ ,计算出最长的不满足条件的 $ l′ $,则最终答案即为所有 $ i $ 对应的 $ l′ $ 的最大值加一(无解的情况除外)。
考虑对于某个 $ p_i $ 而言,我们如何求出这样的 $ l′ $ 。我们考虑将所有满足 $ p_i | w $的边拿出来,并只保留这些边。则 $ l′ $ 等价于在这样得到的森林中,经过 $ (u, v) $ 的最长路径。
使用简单的树形DP即可求出某个点向子树方向以及向祖先方向延伸的最长路径,分类讨论即可对于每个 $ (u, v) $ 求出对应的 $ l′ $ 。
接下来考虑无解的情况。事实上,无解等价于刚刚求出的某个 $ l′ $ 和经过 $ (u, v) $ 的最长路径相同。经过 $ (u, v) $ 的最长路径和刚刚是同样的问题,直接对整棵树都做一遍树形 DP 即可。
接下来考虑复杂度。求出每个询问的 $ gcd $ 的复杂度为 $ O(q log_2 n log_2 w) $ ,而求出最长路的部分是与边数成线性的,而每条边至多出现 $ 7 $ 次,因此该做法的总复杂度即为 $ O(q log_2 n log_2 w + nω(w)) $ 。

CODE:

//9.10补全
#pragma GCC optimize(2)
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>

using namespace std;

#define LL long long

const int N = 2e5 + 100;
const int V = 2e6 + 100;
const int M = 18;

struct Edge {
    int to,from;
    int data;
}e[N<<1];

int head[N],cnt,tot,n,q;
int f[M + 1][N],g[M + 1][N];
int prime[V],d[V],deep[N];
bool p[V];

inline void prework() {
    d[1] = 1;
    for(int i = 2 ; i <= V - 1 ; i++) {
        if(!p[i]) prime[++tot] = i,d[i] = i;
        for(int j = 1 ; j <= tot && i * prime[j] < V ; ++j) {
            p[i * prime[j]] = true;
            d[i * prime[j]] = prime[j];
            if(!(i % prime[j])) break;
        }
    }
}
inline void add_edge(int x,int y,int z) {
    e[++cnt].from = y;
    e[cnt].data = z;
    e[cnt].to = head[x];
    head[x] = cnt;
} 
inline int gcd(int a,int b) {
    return !b ? a :gcd(b,a % b);
}
inline void dfs(int v,int fa) {
    for(int i = 1 ; i <= M ; i++) {
        f[i][v] = f[i - 1][f[i - 1][v]];
        g[i][v] = gcd(g[i - 1][v],g[i - 1][f[i - 1][v]]);
    }
    for(int i = head[v] ; i ; i = e[i].to) {
        if(e[i].from == fa) continue;
        f[0][e[i].from] = v;
        g[0][e[i].from] = e[i].data;
        deep[e[i].from] = deep[v] + 1;
        dfs(e[i].from,v);
    }
}
inline int LCA(int x,int y) {
    if(deep[x] != deep[y]) {
        if(deep[x] < deep[y]) swap(x,y);
        int dis = deep[x] - deep[y];
        for(int i = 0 ; i <= M ; i++) {
            if((1 << i) & dis) x = f[i][x];
        }
    }
    if(x == y) return x;
    for(int i = M ; i >= 0 ; i--) {
        if(f[i][x] == f[i][y]) continue;
        x = f[i][x],y = f[i][y];
    }
    return f[0][x];
}
inline int cal(int x,int y) {
    if(deep[x] < deep[y]) swap(x,y);
    int d = deep[x] - deep[y],ans = 0;
    for(int i = 0 ; i <= M ; i++) {
        if((1<<i)&d) {
            ans = gcd(g[i][x],ans);
            x = f[i][x];
        }
    }
    return ans;
}
inline int calc(int x,int y) {
    int l = LCA(x,y);
    return gcd(cal(l,x),cal(l,y));
}
inline int dp(int v,int fa,int d) {
    int ans = 0;
    for(int i = head[v] ; i ; i = e[i].to) {
        if(e[i].from == fa) continue;
        if(e[i].data % d) continue;
        ans = max(ans,dp(e[i].from,v,d)+1);
    }
    return ans;
}
inline int kth(int v,int k) {
    for(int i = 0 ; i <= M ; i++) {
        if((1 << i) & k) v = f[i][v];
    }
    return v;
}
inline int dis(int x,int y) {
    return deep[x] + deep[y] - deep[LCA(x,y)] * 2;
}
inline int read() {
    int x = 0, f = 1; 
    char ch = getchar(); 
    while(ch < '0' || ch > '9') {if (ch == '-')f = -1; ch = getchar();}
    while(ch >= '0' && ch <= '9') {x = (x << 1) + (x << 3) + (ch ^ 48); ch = getchar();}
    return x * f; 
}

int main() {
    prework();
    n = read(),q = read();
    for(int i = 1 ; i < n ; i++) {
        int u = read(),v = read(),z = read();
        add_edge(u,v,z);
        add_edge(v,u,z);
    }
    dfs(1,0);
    while(q--) {
        int x = read(),y = read();
        int l = LCA(x,y),g = calc(x,y);
        if(g == 1) {
            printf("%d 
",dis(x,y));
            continue;
        }
        int fx = f[0][x],fy = f[0][y];
        if(l == x) fx = kth(y,deep[y] - deep[x] - 1);
        else if(l == y) fy = kth(x,deep[x] - deep[y] - 1); // 限制不走相同的子树
        int mxdis = dp(x,fx,1) + dp(y,fy,1);
        int ans = 0;
        while(g != 1) {
            int p = d[g];
            ans = max(ans,dp(x,fx,p) + dp(y,fy,p));
            if(ans == mxdis) break;
            while(!(g % p)) g /= p;
        }
        printf("%d
",ans == mxdis ? -1 : ans + dis(x,y) + 1);
    }
    //system("pause");
    return 0;
}
原文地址:https://www.cnblogs.com/Repulser/p/11494359.html