2024.4.11力扣每日一题——互质树

2024.4.11

      • 题目来源
      • 我的题解
        • [方法一 深度优先遍历+回溯+存储父节点](#方法一 深度优先遍历+回溯+存储父节点)
        • [方法二 官方深度优先遍历](#方法二 官方深度优先遍历)

题目来源

力扣每日一题;题序:1766

我的题解

方法一 深度优先遍历+回溯+存储父节点

使用一个List存储深度优先遍历过程中的父节点,然后从List的右侧开始遍历,直到与当前节点互质结束。
时间复杂度 :O( n 2 n^2 n2)
空间复杂度:O(n+h)。h是树的高度,表示递归栈的空间

java 复制代码
//会超时
public int[] getCoprimes(int[] nums, int[][] edges) {
    int n=nums.length;
    List<Integer>[] g=createGraph(n,edges);
    int[] res=new int[n];
    res[0]=-1;
    List<Integer> l=new ArrayList<>();
    l.add(0);
    dfs(g,nums,l,res,0,-1);
    return res;
}
public void dfs(List<Integer>[] g,int[] nums,List<Integer> parent,int[] res,int cur,int pre){
    for(int next:g[cur]){
        if(next!=pre){
            res[next]=check(nums,next,parent);
            parent.add(next);
            dfs(g,nums,parent,res,next,cur);
            parent.remove(parent.size()-1);
        }
    }

}
//获取互质的第一个祖先节点
public int check(int[] nums,int cur,List<Integer> parent){
    int res=-1;
    for(int i=parent.size()-1;i>=0;i--){
        if(gcd(nums[cur],nums[parent.get(i)])==1){
            res=parent.get(i);
            break;
        }
    }
    return res;
}
public List<Integer>[] createGraph(int n,int[][] edges){
    List<Integer>[] g=new ArrayList[n];
    for(int i=0;i<n;i++)
        g[i]=new ArrayList<>();
    for(int[] t:edges){
        int from =  t[0];
        int to = t[1];
        g[from].add(to);
        g[to].add(from);
    }
    return g;
}
public int gcd(int x,int y){
    if(y==0)
        return x;
    return gcd(y,x%y);
}
方法二 官方深度优先遍历

没怎么看懂,自己看官方题解

有任何问题,欢迎评论区交流,欢迎评论区提供其它解题思路(代码),也可以点个赞支持一下作者哈😄~

相关推荐
RainbowSea10 小时前
12. LangChain4j + 向量数据库操作详细说明
java·langchain·ai编程
RainbowSea10 小时前
11. LangChain4j + Tools(Function Calling)的使用详细说明
java·langchain·ai编程
考虑考虑14 小时前
Jpa使用union all
java·spring boot·后端
用户37215742613514 小时前
Java 实现 Excel 与 TXT 文本高效互转
java
浮游本尊15 小时前
Java学习第22天 - 云原生与容器化
java
聚客AI16 小时前
🙋‍♀️Transformer训练与推理全流程:从输入处理到输出生成
人工智能·算法·llm
渣哥17 小时前
原来 Java 里线程安全集合有这么多种
java
间彧17 小时前
Spring Boot集成Spring Security完整指南
java
间彧17 小时前
Spring Secutiy基本原理及工作流程
java
Java水解19 小时前
JAVA经典面试题附答案(持续更新版)
java·后端·面试