Class Solution
-
- All Implemented Interfaces:
public final class Solution
1766 - Tree of Coprimes\.
Hard
There is a tree (i.e., a connected, undirected graph that has no cycles) consisting of
n
nodes numbered from0
ton - 1
and exactlyn - 1
edges. Each node has a value associated with it, and the root of the tree is node0
.To represent this tree, you are given an integer array
nums
and a 2D arrayedges
. Eachnums[i]
represents the <code>i<sup>th</sup></code> node's value, and each <code>edgesj = u<sub>j</sub>, v<sub>j</sub></code> represents an edge between nodes <code>u<sub>j</sub></code> and <code>v<sub>j</sub></code> in the tree.Two values
x
andy
are coprime ifgcd(x, y) == 1
wheregcd(x, y)
is the greatest common divisor ofx
andy
.An ancestor of a node
i
is any other node on the shortest path from nodei
to the root. A node is not considered an ancestor of itself.Return an array
ans
of sizen
, whereans[i]
is the closest ancestor to nodei
such thatnums[i]
andnums[ans[i]]
are coprime , or-1
if there is no such ancestor.Example 1:
Input: nums = 2,3,3,2, edges = \[\[0,1],1,2,1,3]
Output: -1,0,0,1
Explanation: In the above figure, each node's value is in parentheses.
Node 0 has no coprime ancestors.
Node 1 has only one ancestor, node 0. Their values are coprime (gcd(2,3) == 1). - Node 2 has two ancestors, nodes 1 and 0. Node 1's value is not coprime (gcd(3,3) == 3), but node 0's value is (gcd(2,3) == 1), so node 0 is the closest valid ancestor.
Node 3 has two ancestors, nodes 1 and 0. It is coprime with node 1 (gcd(3,2) == 1), so node 1 is its closest valid ancestor.
Example 2:
Input: nums = 5,6,10,2,3,6,15, edges = \[\[0,1],0,2,1,3,1,4,2,5,2,6]
Output: -1,0,-1,0,0,0,-1
Constraints:
nums.length == n
1 <= nums[i] <= 50
<code>1 <= n <= 10<sup>5</sup></code>
edges.length == n - 1
edges[j].length == 2
<code>0 <= u<sub>j</sub>, v<sub>j</sub>< n</code>
<code>u<sub>j</sub> != v<sub>j</sub></code>