Merge pull request neetcode-gh#1203 from razer96/230-Kth-Smallest-Ele…

…ment-in-a-BST.go 230. Kth Smallest Element in a BST go solution.go
Jay-0331 · Sep 30, 2022 · 2f07dbd · 2f07dbd
2 parents 07eef8d + 0f3d0d7
commit 2f07dbd
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/go/230-Kth-Smallest-Element-in-a-BST.go b/go/230-Kth-Smallest-Element-in-a-BST.go
@@ -0,0 +1,22 @@
+func kthSmallest(root *TreeNode, k int) int {
+	stack := make([]*TreeNode, 0, k)
+
+	for {
+		for root != nil {
+			stack = append(stack, root)
+			root = root.Left
+		}
+
+		root = stack[len(stack) - 1]
+		stack = stack[:len(stack) - 1]
+
+		k--
+		if k == 0 {
+			return root.Val
+		}
+
+		root = root.Right
+	}
+
+	return -1
+}