Skip to content

Commit c14ea01

Browse files
author
jinbeom
committed
Kth Smallest Element in a BST Solution
1 parent 0066475 commit c14ea01

File tree

1 file changed

+30
-0
lines changed

1 file changed

+30
-0
lines changed
Lines changed: 30 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,30 @@
1+
class TreeNode:
2+
def __init__(self, val=0, left=None, right=None):
3+
self.val = val
4+
self.left = left
5+
self.right = right
6+
7+
8+
class Solution:
9+
10+
def kthSmallest(self, root: Optional[TreeNode], k: int) -> int:
11+
self.result = 0
12+
self.count = 0
13+
14+
def dfs(node):
15+
if node is None:
16+
return
17+
18+
dfs(node.left)
19+
20+
self.count += 1
21+
22+
if self.count == k:
23+
self.result = node.val
24+
return
25+
26+
dfs(node.right)
27+
28+
dfs(root)
29+
30+
return self.result

0 commit comments

Comments
 (0)