2024.4.11
题目来源
力扣每日一题；题序：1766
我的题解
方法一 深度优先遍历+回溯+存储父节点
//会超时
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);
}
方法二 官方深度优先遍历
有任何问题，欢迎评论区交流，欢迎评论区提供其它解题思路（代码），也可以点个赞支持一下作者哈😄~
