leetcode1766 互质树 题解

早起一看今天的题吓我一跳，好家伙 “互质树” 听起来就是一个非常 diao 的题，题面看起来也非常 diao，但是看完数据范围就发现这题有点拉了。

题目描述

原地址
大致意思是给你一棵树，然后每个节点都有一个值，然后问你距离每个点最近，且两个值互质的祖宗编号是多少。
点最多有 $10^5$ 个，每个点的值小于等于 $50$，大于 $0$。

做法

dfs + 链表
对于每一种值的点，我们都对其建一个链表，当我们 dfs 遇到一个新的点的时候，我们就把这个点加入到对应的链表里面，当回溯的时候，就把对应的点从链表里面删除，我们来举个例子。
$nums=[5,6,10,2,3,6,15]$
$edges=[[0,1],[0,2],[1,3],[1,4],[2,5],[2,6]]$

首先我们遍历到了 0 这个点，我们把 0 号点加入到链表里面去。接下来是 2 和 5，同样加到对应的链表里面，然后我们回溯，把 5 号点从链表里面删除，然后遍历 6 号点，把 6 号点加入到链表里面，然后回溯，又删掉 6 号和 2 号点，后面以此类推。这样就保证了这些链表里面的点，都是当前访问的这个点的祖宗节点。

然后因为节点的值都比较小，所以我们直接遍历与当前值互质的所有值，然后把他们对应的链表拿出来，最后一个放进去的就是距离他最近的。

大家可以看下我代码里面的注释。

代码

class Solution
{
public:
    int gcd(int x, int y) // gcd 用来判断互质
    {
        if (y == 0)
            return x;
        return gcd(y, x % y);
    }
    vector<int> getCoprimes(vector<int> &nums, vector<vector<int>> &edges)
    {
        const int N = nums.size();
        vector<int> rp[55], ans(N, -1); 
        vector<int> deep(N + 5, 0);
        vector<int> edge[N + 5];
        vector<int> link[55]; // 我们用 vector 来实现链表，反正就只对最后一个位置进行操作
        for (int i = 1; i <= 50; i++)
            for (int j = 1; j <= i; j++)
                if (gcd(i, j) == 1)
                    rp[i].push_back(j), rp[j].push_back(i); // 这样就能提前知道谁跟谁互质了
        for (int i = 0; i < edges.size(); i++) // 建边
            edge[edges[i][0]].push_back(edges[i][1]), edge[edges[i][1]].push_back(edges[i][0]);
        function<void(int, int)> dfs = [&](int x, int fa)
        {
            int pos = -1, pos_deep = -1; // 寻找距离 x 这个点最近且满足条件的祖宗
            for (int i = 0; i < rp[nums[x]].size(); i++) // 枚举所有和 nums[x] 互质的值
            {
                int num = rp[nums[x]][i];
                if (link[num].empty()) // 要判断链表是不是空的
                    continue;
                int id = link[num].back(); // 直接找最后一个，最后一个就是这个值里面深度最深的
                if (deep[id] > pos_deep) // 打擂台，找一个最深的
                {
                    pos = id;
                    pos_deep = deep[id];
                }
            }
            ans[x] = pos;
            link[nums[x]].push_back(x); // 把 x 加入到对应的链表里面
            for (int i = 0; i < edge[x].size(); i++)
            {
                int v = edge[x][i];
                if (v == fa) // 不能往回遍历
                    continue;
                deep[v] = deep[x] + 1; // 深度加一
                dfs(v, x); // 接着往下搜
            }
            link[nums[x]].pop_back(); // 回溯的时候把这个点从链表里面删掉
        };
        dfs(0, -1); // 调用一下 lambda 函数
        return ans;
    }
};

结果

总的来说就是代码稍微有点难度，想倒是不难想。

作者能力有限，如果有任何错误之处，还请各位指教。(～￣▽￣)～