1766. 互质树

最新推荐文章于 2024-10-05 12:07:50 发布

咔咔咔的

最新推荐文章于 2024-10-05 12:07:50 发布

阅读量186

点赞数 2

分类专栏： leetcode 文章标签： c++

本文链接：https://blog.csdn.net/weixin_45256307/article/details/137901304

版权

leetcode 专栏收录该内容

591 篇文章 2 订阅

订阅专栏

本文介绍了一个编程问题，通过递归实现深度优先搜索算法，解决给定整数数组中互质节点的连接问题，利用欧几里得算法(gcd)来确定节点之间的关系。

摘要由CSDN通过智能技术生成

1766. 互质树

题目链接：1766. 互质树

代码如下：

//深度优先搜索
//参考leetcode官方题解
class Solution {
public:
    void dfs(vector<int>& nums,int x,int depth)
    {
        dep[x]=depth;
        for(int val:gcds[nums[x]])
        {
            if(tmp[val].empty())
            {
                continue;
            }

            int las=tmp[val].back();
            if(ans[x]==-1||dep[las]>dep[ans[x]])
            {
                ans[x]=las;
            }
        }
        tmp[nums[x]].push_back(x);

        for(int val:g[x])
        {
            if(dep[val]==-1)//被访问的dep不为-1
            {
                dfs(nums,val,depth+1);
            }
        }
        tmp[nums[x]].pop_back();
    }

    vector<int> getCoprimes(vector<int>& nums, vector<vector<int>>& edges) 
    {
        //初始化
        gcds.resize(mx);
        tmp.resize(mx);
        ans.resize(nums.size(),-1);
        dep.resize(nums.size(),-1);
        g.resize(nums.size());

        for(int i=1;i<mx;i++)
        {
            for(int j=1;j<mx;j++)
            {
                if(gcd(i,j)==1)
                {
                    gcds[i].push_back(j);
                }
            }
        }
        
        for(const auto& val:edges)
        {
            g[val[0]].push_back(val[1]);
            g[val[1]].push_back(val[0]);
        }

        dfs(nums,0,1);
        return ans;
    }
private:
    vector<vector<int>> gcds;
    vector<vector<int>> tmp;
    vector<vector<int>> g;
    vector<int> dep;
    vector<int> ans;
    const int mx=51;
};