Create 1602-find-nearest-right-node-in-binary-tree.js

Author: everthis

1602-find-nearest-right-node-in-binary-tree.js

@@ -0,0 +1,28 @@
+/**
+ * Definition for a binary tree node.
+ * function TreeNode(val, left, right) {
+ *     this.val = (val===undefined ? 0 : val)
+ *     this.left = (left===undefined ? null : left)
+ *     this.right = (right===undefined ? null : right)
+ * }
+ */
+/**
+ * @param {TreeNode} root
+ * @param {TreeNode} u
+ * @return {TreeNode}
+ */
+const findNearestRightNode = function(root, u) {
+  const q = [root]
+  while(q.length) {
+    const size = q.length
+    let target = false
+    for(let i = 0; i < size; i++) {
+      const tmp = q.shift()
+      if(target) return tmp
+      if(tmp === u) target = true
+      if(tmp.left) q.push(tmp.left)
+      if(tmp.right) q.push(tmp.right)
+    }
+  }     
+  return null
+};