最长树链（dfs 树）

最长树链

题目描述

树链是指树里的一条路径。美团外卖的形象代言人袋鼠先生最近在研究一个特殊的最长树链问题。现在树中的每个点都有一个正整数值，他想在树中找出最长的树链，使得这条树链上所有对应点的值的最大公约数大于1。请求出这条树链的长度。
输入描述:
第1行：整数n（1 ≤ n ≤ 100000），表示点的个数。
第2~n行：每行两个整数x，y表示xy之间有边，数据保证给出的是一棵树。
第n+1行：n个整数，依次表示点1~n对应的权值（1 ≤ 权值 ≤ 1,000,000,000）。

输出描述：

满足最长路径的长度

输入

4
1 2
1 3
2 4
6 4 5 2

输出

3
unordered_map 比 map 更快，到所需内存更多

#include <cstdio>
#include <cstring>
#include <string>
#include <iostream>
#include <vector>
#include <unordered_map>
//#include <bits/stdc++.h>
using namespace std;
#define ll long long
vector<int> e[100100];
// val 的所有因数
unordered_map<int, vector<int>> g;
int val[100100];
int vis[100100];

//u：节点  p：val[u] 的约数
int dfs(int u, int p) {
    vis[u] = 1;
    int m1 = 0, m2 = 0;
    //遍历当前节点所连接的节点
    for (auto v:e[u]) {
        if (vis[v] || val[v] % p != 0) {
            continue;
        }
        //当前节点未访问且两点间最大公约数不为0
        int d = dfs(v, p);
        if (d > m1) {
            m2 = m1;
            m1 = d;
        } else if (d > m2) {
            m2 = d;
        }
    }
    return m1 + m2 + 1;
}

int solve(int u) {
    int res = 0;
    for (auto x:g[u]) {
        if (!vis[x]) {
            res = max(res, dfs(x, u));
        }
    }
    for (auto x:g[u]) {
        vis[x] = 0;
    }
    return res;
}

int main() {
    int n;
    cin >> n;
    for (int i = 1; i < n; i++) {
        int x, y;
        scanf("%d%d", &x, &y);
        e[x].push_back(y);
        e[y].push_back(x);
    }
    // val[i] 的约数中添加 i
    for (int i = 1; i <= n; i++) {
        scanf("%d", &val[i]);
        int use = val[i];
        for (int j = 2; j * j <= use; j++) {
            if (use % j == 0) {
                while (use % j == 0) {
                    use /= j;
                }
                g[j].push_back(i);
            }
        }
        if (use > 1) {
            g[use].push_back(i);
        }
    }
    int ans = 0;
    for (auto i:g) {
        ans = max(ans, solve(i.first));
    }
    cout << ans << endl;
    return 0;
}