LC-0116

Signed-off-by: Tahsin Tunan <[email protected]>
Gyanachand1 · Dec 28, 2022 · 7084335 · 7084335
1 parent ee7a3bc
commit 7084335
Showing 1 changed file with 30 additions and 0 deletions.
diff --git a/go/0116-populating-next-right-pointers-in-each-node.go b/go/0116-populating-next-right-pointers-in-each-node.go
@@ -0,0 +1,30 @@
+/**
+ * Definition for a Node.
+ * type Node struct {
+ * Val int
+ * Left *Node
+ * Right *Node
+ * Next *Node
+ * }
+ */
+
+func connect(root *Node) *Node {
+ populate(root)
+ return root
+}
+
+func populate(node *Node) {
+ if node == nil {
+ return
+ }
+ if node.Left == nil {
+ return
+ }
+
+ node.Left.Next = node.Right
+ if node.Next != nil {
+ node.Right.Next = node.Next.Left
+ }
+ populate(node.Left)
+ populate(node.Right)
+}